diff --git a/lib/airports/airports_test.go b/lib/airports/airports_test.go new file mode 100644 index 0000000..f9e7322 --- /dev/null +++ b/lib/airports/airports_test.go @@ -0,0 +1,64 @@ +package airports + +import ( + "testing" + "fmt" + + "github.com/stretchr/testify/assert" +) + +func TestSearchCity(t *testing.T) { + cases := []struct { + query string + IATA string + }{ + {"Toronto", "YYZ"}, + {"Montreal", "YUL"}, + {"Vancouver", "YVR"}, + {"Marrakesh", "RAK"}, + } + + for _, c := range cases { + s := Search(c.query) + fmt.Println(s) + assert.NotZero(t, len(s)) + assert.Equal(t, c.IATA, s[0].IATA) + } +} + +func TestSearchAirportName(t *testing.T) { + cases := []struct { + query string + IATA string + }{ + {"Pearson", "YYZ"}, + {"Trudeau", "YUL"}, + {"Gatwick", "LGW"}, + {"Heathrow", "LHR"}, + // {"Tegel", "TXL"}, + } + + for _, c := range cases { + s := Search(c.query) + assert.NotZero(t, len(s)) + assert.Equal(t, c.IATA, s[0].IATA) + } +} + +func TestSearchIATA(t *testing.T) { + cases := []struct { + IATA string + }{ + {"YYZ"}, + {"YUL"}, + {"LGW"}, + {"LHR"}, + {"TXL"}, + } + + for _, c := range cases { + s := Search(c.IATA) + assert.NotZero(t, len(s)) + assert.Equal(t, c.IATA, s[0].IATA) + } +}