penguin/golifehk

datasources/kmb/query_test.go

raw · 1312 bytes

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")
	}
}