stitcher tests

stitcher_test.go

package main

import (
	"fmt"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
)

func TestStitcher(t *testing.T) {

	t.Run("it gets both websites", func(t *testing.T) {

		site1 := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			fmt.Fprint(w, "hello")
		}))
		defer site1.Close()

		site2 := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			fmt.Fprint(w, "world")
		}))
		defer site2.Close()

		req := httptest.NewRequest(http.MethodGet, "/", nil)
		q := req.URL.Query()
		q.Add("url", site1.URL)
		q.Add("url", site2.URL)
		req.URL.RawQuery = q.Encode()
		res := httptest.NewRecorder()

		Stitcher(res, req)

		if res.Code != http.StatusOK {
			t.Fatal("not 200", res.Code)
		}

		if !strings.Contains(res.Body.String(), "hello") {
			t.Fatal("didnt get 'hello' in response", res.Body.String())
		}

		if !strings.Contains(res.Body.String(), "world") {
			t.Fatal("didnt get 'world' in response", res.Body.String())
		}
	})

	t.Run("400 when no querystring values", func(t *testing.T) {
		req := httptest.NewRequest(http.MethodGet, "/", nil)
		res := httptest.NewRecorder()

		Stitcher(res, req)

		if res.Code != http.StatusBadRequest {
			t.Fatal("not 400", res.Code)
		}
	})

}