script_test.go

package galicia2

import (
	"net/http/httptest"
	"testing"
	"time"

	"github.com/mattn/go-nulltype"
	"github.com/stretchr/testify/assert"
	"github.com/whitewater-guide/gorge/core"
	"github.com/whitewater-guide/gorge/testutils"
)

func setupTestServer() *httptest.Server {
	return testutils.SetupFileServer(map[string]string{
		"/table": "table.xls",
		"":       "A015.html",
	}, nil)
}

func TestGalicia2_ListGauges(t *testing.T) {
	ts := setupTestServer()
	defer ts.Close()
	s := scriptGalicia2{
		name:           "galicia2",
		listURL:        ts.URL + "/table.xls",
		gaugeURLFormat: ts.URL + "/%s.html",
		skipCookies:    true,
	}
	actual, err := s.ListGauges()
	expected := core.Gauge{
		GaugeID: core.GaugeID{
			Script: "galicia2",
			Code:   "A015",
		},
		LevelUnit: "m",
		Location: &core.Location{
			Latitude:  42.87182,
			Longitude: -7.52761,
			Altitude:  348.65,
		},
		Name:     "Río Neira en Páramo (o)",
		URL:      ts.URL + "/A015.html",
		Timezone: "Europe/Madrid",
	}
	if assert.NoError(t, err) {
		assert.Len(t, actual, 58)
		assert.Contains(t, actual, expected)
	}
}

func TestGalicia2_Harvest(t *testing.T) {
	ts := setupTestServer()
	defer ts.Close()
	s := scriptGalicia2{
		name:           "galicia2",
		listURL:        ts.URL + "/table.xls",
		gaugeURLFormat: ts.URL + "/%s.html",
		skipCookies:    true,
	}
	actual, err := core.HarvestSlice(&s, core.StringSet{}, 0)
	expected := &core.Measurement{
		GaugeID: core.GaugeID{
			Script: "galicia2",
			Code:   "A015",
		},
		Timestamp: core.HTime{
			Time: time.Date(2020, time.January, 23, 8, 0, 0, 0, time.UTC),
		},
		Level: nulltype.NullFloat64Of(0.44),
	}
	if assert.NoError(t, err) {
		assert.Len(t, actual, 58)
		assert.Contains(t, actual, expected)
	}
}