tastyroot_test.go

package main

import (
	"net/http"
	"net/http/httptest"
	"tastyroot/resources"
	"testing"
)

type TestCat struct {
	Name string
}

func TestSimpleResourceHandleGET(t *testing.T) {
	testUrl := "/cat"
	resource := resources.SimpleResource{
		TestCat{"Meow"},
		testUrl,
	}

	// Successful processing
	expectedBody := "{\"Name\":\"Meow\"}"
	recorder := httptest.NewRecorder()

	req, err := http.NewRequest("GET", "http://example.com", nil)
	if err != nil {
		t.Errorf("Failed to create request.")
	}

	resource.HandleGET(recorder, req)

	if body := recorder.Body.String(); body != expectedBody {
		t.Errorf("Body (%s) did not match expectation (%s).",
			recorder.Body.String(),
			expectedBody)
	}
}

func TestSimpleResourceHandlePOST(t *testing.T) {
	testUrl := "/cat"
	resource := resources.SimpleResource{
		TestCat{"Meow"},
		testUrl,
	}

	expectedBody := "{\"ping\":\"pong\"}"
	recorder := httptest.NewRecorder()

	req, err := http.NewRequest("POST", "http://example.com", nil)
	if err != nil {
		t.Errorf("Failed to create request.")
	}

	resource.HandlePOST(recorder, req)

	if body := recorder.Body.String(); body != expectedBody {
		t.Errorf("Body (%s) did not match expectation (%s).",
			recorder.Body.String(),
			expectedBody)
	}
}

func TestSimpleResourceGetters(t *testing.T) {
	testUrl := "/cat2"
	cat := TestCat{"Meow"}
	resource := resources.SimpleResource{cat, testUrl}

	if url, _ := resource.GetBaseUrl(); url != testUrl {
		t.Errorf("GetBaseUrl didn't return expected url")
	}
}