//go:build windows

package diagnostic_test

import (
	"strings"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	diagnostic "github.com/kjake/cloudflared/diagnostic/network"
)

func TestDecode(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name         string
		text         string
		expectedHops []*diagnostic.Hop
	}{

		{
			"tracert output",
			`
Tracing route to region2.v2.argotunnel.com [198.41.200.73]
over a maximum of 5 hops:

  1    10 ms    <1 ms     1 ms  192.168.64.1
  2    27 ms    14 ms     5 ms  192.168.1.254
  3     *        *        *     Request timed out.
  4     *        *        *     Request timed out.
  5    27 ms     5 ms     5 ms  195.8.30.245

Trace complete.
`,
			[]*diagnostic.Hop{
				diagnostic.NewHop(
					uint8(1),
					"192.168.64.1",
					[]time.Duration{
						time.Duration(10000),
						time.Duration(1000),
						time.Duration(1000),
					},
				),
				diagnostic.NewHop(
					uint8(2),
					"192.168.1.254",
					[]time.Duration{
						time.Duration(27000),
						time.Duration(14000),
						time.Duration(5000),
					},
				),
				diagnostic.NewTimeoutHop(uint8(3)),
				diagnostic.NewTimeoutHop(uint8(4)),
				diagnostic.NewHop(
					uint8(5),
					"195.8.30.245",
					[]time.Duration{
						time.Duration(27000),
						time.Duration(5000),
						time.Duration(5000),
					},
				),
			},
		},
		{
			"repeated hop index parse failure",
			`1	12.874 ms	15.517 ms	15.311 ms	172.68.101.121 (172.68.101.121)  
2	12.874 ms	15.517 ms	15.311 ms	172.68.101.121 (172.68.101.121)  
someletters * * *`,
			[]*diagnostic.Hop{
				diagnostic.NewHop(
					uint8(1),
					"172.68.101.121 (172.68.101.121)",
					[]time.Duration{
						time.Duration(12874),
						time.Duration(15517),
						time.Duration(15311),
					},
				),
				diagnostic.NewHop(
					uint8(2),
					"172.68.101.121 (172.68.101.121)",
					[]time.Duration{
						time.Duration(12874),
						time.Duration(15517),
						time.Duration(15311),
					},
				),
			},
		},
		{
			"hop index parse failure",
			`1	12.874 ms	15.517 ms	15.311 ms	172.68.101.121 (172.68.101.121)  
2	12.874 ms	15.517 ms	15.311 ms	172.68.101.121 (172.68.101.121)  
someletters abc ms 0.456 ms 0.789 ms 8.8.8.8 8.8.8.9`,
			[]*diagnostic.Hop{
				diagnostic.NewHop(
					uint8(1),
					"172.68.101.121 (172.68.101.121)",
					[]time.Duration{
						time.Duration(12874),
						time.Duration(15517),
						time.Duration(15311),
					},
				),
				diagnostic.NewHop(
					uint8(2),
					"172.68.101.121 (172.68.101.121)",
					[]time.Duration{
						time.Duration(12874),
						time.Duration(15517),
						time.Duration(15311),
					},
				),
			},
		},
		{
			"missing rtt",
			`1	<12.874 ms	<15.517 ms	<15.311 ms	172.68.101.121 (172.68.101.121)  
2 * 0.456 ms 0.789 ms 8.8.8.8 8.8.8.9`,
			[]*diagnostic.Hop{
				diagnostic.NewHop(
					uint8(1),
					"172.68.101.121 (172.68.101.121)",
					[]time.Duration{
						time.Duration(12874),
						time.Duration(15517),
						time.Duration(15311),
					},
				),
				diagnostic.NewHop(
					uint8(2),
					"8.8.8.8 8.8.8.9",
					[]time.Duration{
						time.Duration(456),
						time.Duration(789),
					},
				),
			},
		},
		{
			"simple example ipv4",
			`1	12.874 ms	15.517 ms	15.311 ms	172.68.101.121 (172.68.101.121)  
2	12.874 ms	15.517 ms	15.311 ms	172.68.101.121 (172.68.101.121)  
3  * * * Request timed out.`,
			[]*diagnostic.Hop{
				diagnostic.NewHop(
					uint8(1),
					"172.68.101.121 (172.68.101.121)",
					[]time.Duration{
						time.Duration(12874),
						time.Duration(15517),
						time.Duration(15311),
					},
				),
				diagnostic.NewHop(
					uint8(2),
					"172.68.101.121 (172.68.101.121)",
					[]time.Duration{
						time.Duration(12874),
						time.Duration(15517),
						time.Duration(15311),
					},
				),
				diagnostic.NewTimeoutHop(uint8(3)),
			},
		},
		{
			"simple example ipv6",
			` 1 12.780 ms  9.118 ms  10.046 ms 2400:cb00:107:1024::ac44:6550
 2   9.945 ms  10.033 ms  11.562 ms 2a09:bac1::`,
			[]*diagnostic.Hop{
				diagnostic.NewHop(
					uint8(1),
					"2400:cb00:107:1024::ac44:6550",
					[]time.Duration{
						time.Duration(12780),
						time.Duration(9118),
						time.Duration(10046),
					},
				),
				diagnostic.NewHop(
					uint8(2),
					"2a09:bac1::",
					[]time.Duration{
						time.Duration(9945),
						time.Duration(10033),
						time.Duration(11562),
					},
				),
			},
		},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			t.Parallel()

			hops, err := diagnostic.Decode(strings.NewReader(test.text), diagnostic.DecodeLine)
			require.NoError(t, err)
			assert.Equal(t, test.expectedHops, hops)
		})
	}
}