auth_service_test.go

package main

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

var (
	mux    *http.ServeMux
	server *httptest.Server
)

func authSetup() {
	mux = http.NewServeMux()
	server = httptest.NewServer(mux)
}

func authTeardown() {
	server.Close()
}

func TestDefaultToNullService(t *testing.T) {
	service, err := ConnectToAuthService(&Context{}, "")
	if err != nil {
		t.Fatalf("Unexpected error connection to authentication service: %v", err)
	}
	if _, ok := service.(NullAuthService); !ok {
		t.Errorf("Expected %#v to be a NullAuthService", service)
	}
}

func TestCreateRemoteService(t *testing.T) {
	authSetup()
	defer authTeardown()

	mux.HandleFunc("/style", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, "unit-test")
	})

	service, err := ConnectToAuthService(&Context{HTTPS: http.DefaultClient}, server.URL)
	if err != nil {
		t.Fatalf("Unexpected error connection to authentication service: %v", err)
	}
	if _, ok := service.(RemoteAuthService); !ok {
		t.Errorf("Expected %#v to be a RemoteAuthService", service)
	}
}

func TestSuccessfulRemoteAuth(t *testing.T) {
	authSetup()
	defer authTeardown()

	mux.HandleFunc("/style", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, "unit-test")
	})

	hit := false
	mux.HandleFunc("/validate", func(w http.ResponseWriter, r *http.Request) {
		hit = true
		if r.Method != "GET" {
			t.Errorf("Expected a GET request, but was [%s]", r.Method)
		}

		err := r.ParseForm()
		if err != nil {
			t.Errorf("Unexpected error parsing form: %v", err)
		}

		if username := r.FormValue("accountName"); username != "someuser" {
			t.Errorf("Unexpected account name: [%s]", username)
		}

		if token := r.FormValue("apiKey"); token != "1234567" {
			t.Errorf("Unexpected API key: [%s]", token)
		}

		w.WriteHeader(http.StatusNoContent)
	})

	c := &Context{HTTPS: http.DefaultClient}
	service, err := ConnectToAuthService(c, server.URL)
	if err != nil {
		t.Fatalf("Unexpected error connection to auth service: %v", err)
	}

	ok, err := service.Validate("someuser", "1234567")
	if err != nil {
		t.Fatalf("Unexpected error calling auth service: %v", err)
	}

	if !hit {
		t.Errorf("Service never called remote endpoint")
	}

	if !ok {
		t.Errorf("Service unexpectedly rejected authentication")
	}
}

func TestUnsuccessfulRemoteAuth(t *testing.T) {
	authSetup()
	defer authTeardown()

	mux.HandleFunc("/style", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, "unit-test")
	})

	hit := false
	mux.HandleFunc("/validate", func(w http.ResponseWriter, r *http.Request) {
		hit = true
		w.WriteHeader(http.StatusNotFound)
	})

	c := &Context{HTTPS: http.DefaultClient}
	service, err := ConnectToAuthService(c, server.URL)
	if err != nil {
		t.Fatalf("Unexpected error connection to auth service: %v", err)
	}

	ok, err := service.Validate("someuser", "1234567")
	if err != nil {
		t.Fatalf("Unexpected error calling auth service: %v", err)
	}

	if !hit {
		t.Errorf("Service never called remote endpoint")
	}

	if ok {
		t.Errorf("Service unexpectedly accepted authentication")
	}
}