darksky_test.go

package darksky

import (
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"testing"
)

func TestForecastRequest_Get(t *testing.T) {
	usingTestServer(validForecastHandler, func(testURL string) {
		resp := MakeRequest(key, 41.8781, -87.6297).WithBaseURL(testURL).Get()

		if resp.Error != nil {
			t.Error(resp.Error)
			return
		}

		if resp.APICallCount != 1 {
			t.Errorf("Expected APICallCount to be %v but was %v.", 1, resp.APICallCount)
		}

		forecast := resp.Forecast

		if len(forecast.Alerts) != 3 {
			t.Error("Expected 3 Alerts.")
		}
	})
}

func TestForecastRequest_Get_InvalidArgs(t *testing.T) {
	resp := MakeRequest("", 41.0, -87.62).Get()
	if resp.Error == nil || resp.Error.Error() != KeyRequired {
		t.Error("Expected empty key to be invalid.")
	}

	resp = MakeRequest("abc", -91.0, 0).Get()
	if resp.Error == nil || resp.Error.Error() != LatitudeInvalid {
		t.Error("Expected latitude to be invalid.")
	}

	resp = MakeRequest("abc", 91.0, 0).Get()
	if resp.Error == nil || resp.Error.Error() != LatitudeInvalid {
		t.Error("Expected latitude to be invalid.")
	}

	resp = MakeRequest("abc", 0, -181.0).Get()
	if resp.Error == nil || resp.Error.Error() != LongitudeInvalid {
		t.Error("Expected longitude to be invalid.")
	}

	resp = MakeRequest("abc", 0, 181.0).Get()
	if resp.Error == nil || resp.Error.Error() != LongitudeInvalid {
		t.Error("Expected longitude to be invalid.")
	}
}

func TestForecastRequest_Get_WithError(t *testing.T) {
	usingTestServer(errorForecastHandler, func(testURL string) {
		resp := MakeRequest(key, 41.8781, -87.6297).WithBaseURL(testURL).Get()

		if resp.Error == nil {
			t.Error("Expected an HTTP Error Response to result in an error.")
		}

		if resp.Error.Error() != "A Server Error Occurred." {
			t.Error("Error() was not the expected value.")
		}
	})
}

func TestForecastRequest_URL(t *testing.T) {
	req := MakeRequest("foo", 41.1234, -81.1234)

	verifyURL := func(r *ForecastRequest, expectedURL string) {
		u, err := req.URL()
		if err != nil {
			t.Error(err)
		}

		if u != expectedURL {
			t.Errorf("Got: %v\nExpected: %v", u, expectedURL)
		}
	}

	verifyURL(req, "https://api.darksky.net/forecast/foo/41.1234,-81.1234?lang=en&units=us")

	req.WithLang(Spanish)

	verifyURL(req, "https://api.darksky.net/forecast/foo/41.1234,-81.1234?lang=es&units=us")

	req.WithUnits(SI)

	verifyURL(req, "https://api.darksky.net/forecast/foo/41.1234,-81.1234?lang=es&units=si")

	req.WithTime(12345)

	verifyURL(req, "https://api.darksky.net/forecast/foo/41.1234,-81.1234,12345?lang=es&units=si")

}

func TestDataPoint_WindDirection(t *testing.T) {
	dp := DataPoint{WindBearing: 147}

	if dp.WindDirection() != "SE" {
		t.Errorf("Expected WindBearing of %v to be SE, was %v.", dp.WindBearing, dp.WindDirection())
	}

	dp.WindBearing = 0

	if dp.WindDirection() != "N" {
		t.Errorf("Expected WindBearing of %v to be N, was %v.", dp.WindBearing, dp.WindDirection())
	}

	dp.WindBearing = 336

	if dp.WindDirection() != "NW" {
		t.Errorf("Expected WindBearing of %v to be NW, was %v.", dp.WindBearing, dp.WindDirection())
	}
}

var key string = "test_key"

var validForecastHandler http.HandlerFunc = http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) {
	jsonBytes, _ := ioutil.ReadFile("testdata/chicago_forecast.json")
	resp.Header().Add(APICallsHeader, "1")
	resp.WriteHeader(200)
	resp.Write(jsonBytes)
})

var errorForecastHandler http.HandlerFunc = http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) {
	resp.WriteHeader(500)
	resp.Write([]byte("A Server Error Occurred."))
})

func usingTestServer(handler http.HandlerFunc, runTest func(testURL string)) {
	ts := httptest.NewServer(handler)

	defer func() {
		ts.Close()
	}()

	runTest(ts.URL)
}