package kmb import ( "fmt" query "github.com/tgckpg/golifehk/query" "testing" ) func TestQuerySchedule(t *testing.T) { qo := Query(query.QueryMessage{Lang: "zh-Hant", Text: "68X"}) mesg, err := qo.Message() if err != nil { t.Errorf("Unexpected Error: %s", err) } qo = Query(query.QueryMessage{Lang: "zh-Hant", Text: "K66 朗屏"}) mesg, err = qo.Message() if err == nil { t.Errorf("Expected Error: %s, got \"\" instead", mesg) } qo = Query(query.QueryMessage{Lang: "zh-Hant", Text: "大欖"}) mesg, err = qo.Message() if err != nil { t.Errorf("Unexpected Error: %s", err) } qo = Query(query.QueryMessage{Lang: "zh-Hant", Text: "261B 大欖"}) mesg, err = qo.Message() if err != nil { t.Errorf("Unexpected Error: %s", err) } qo = Query(query.QueryMessage{ Lang: "zh-Hant", Text: "", // Yuen Long Plaza // Location: &query.GeoLocation{22.444894482044997, 114.02393826485495}, // Nathan Rd // Location: &query.GeoLocation{22.308944848482525, 114.17116565400259}, // GO PARK Location: &query.GeoLocation{22.427238734660868, 114.26595846515744}, }) mesg, err = qo.Message() if err != nil { t.Errorf("Unexpected Error: %s", err) } for _, row := range qo.GetTableData() { for _, cell := range row { fmt.Printf("| %s |", cell.Name) } fmt.Print("\n") } }