Add modified mailgraph.cgi. Update style.css. Complete test suite, every function should be tested.

1 files changed, 445 insertions, 0 deletions

diff --git a/handler_test.go b/handler_test.go
new file mode 100644
index 0000000..92088bf
--- /dev/null
+++ b/handler_test.go
@@ -0,0 +1,445 @@
+package main
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"net/url"
+	"strings"
+	"testing"
+)
+
+func TestIndexHandler(t *testing.T) {
+	request, err := http.NewRequest("GET", "/", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/)", err)
+	}
+	response := httptest.NewRecorder()
+
+	IndexHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code.")
+	}
+
+	request, err = http.NewRequest("GET", "/", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/)", err)
+	}
+	response = httptest.NewRecorder()
+
+	cookie := http.Cookie{Name: "lang", Value: "de"}
+	request.AddCookie(&cookie)
+
+	IndexHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code. Language: de")
+	}
+}
+
+func BenchmarkIndexHandler(b *testing.B) {
+	b.StopTimer()
+	request, _ := http.NewRequest("GET", "/", nil)
+	response := httptest.NewRecorder()
+
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		IndexHandler(response, request)
+	}
+}
+
+func TestRegisterHandler(t *testing.T) {
+	request, err := http.NewRequest("GET", "/register", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/register)", err)
+	}
+	response := httptest.NewRecorder()
+
+	RegisterHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code.")
+	}
+
+	request, err = http.NewRequest("GET", "/register", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/register)", err)
+	}
+	response = httptest.NewRecorder()
+
+	cookie := http.Cookie{Name: "lang", Value: "de"}
+	request.AddCookie(&cookie)
+
+	RegisterHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code. Language: de")
+	}
+}
+
+func BenchmarkRegisterHandler(b *testing.B) {
+	b.StopTimer()
+	request, _ := http.NewRequest("GET", "/register", nil)
+	response := httptest.NewRecorder()
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		RegisterHandler(response, request)
+	}
+}
+
+func TestCreateNewEntryHandler(t *testing.T) {
+	form := url.Values{}
+
+	form.Set("Email", "createNewEntry@localhost")
+	form.Set("ConfirmEmail", "createNewEntry@localhost")
+	form.Set("Password", "Password")
+	form.Set("ConfirmPassword", "Password")
+
+	request, err := http.NewRequest("POST", "/create", strings.NewReader(form.Encode()))
+	if err != nil {
+		t.Log("Error creating new http request. (/create)", err)
+	}
+	request.Header.Add("Content-Type", "application/x-www-form-urlencoded")
+
+	response := httptest.NewRecorder()
+
+	CreateNewEntryHandler(response, request)
+
+	if response.Code != 302 {
+		t.Fatal("Expected 302 as status code.", response.Code)
+	}
+
+	flash := Flash{}
+	session, err := store.Get(request, "_SID")
+	if err != nil {
+		t.Log(err)
+	}
+	flash.Error = session.Flashes("error")
+	flash.Success = session.Flashes("success")
+
+	if len(flash.Error) > 0 {
+		for _, v := range flash.Error {
+			t.Fatal(v)
+		}
+		t.Fatal("Got error message.")
+	}
+
+	if len(flash.Success) == 0 {
+		t.Fatal("Expected success message.")
+	}
+
+	t.Log(flash.Success)
+}
+
+func TestAboutHandler(t *testing.T) {
+	request, err := http.NewRequest("GET", "/about", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/about)", err)
+	}
+	response := httptest.NewRecorder()
+
+	AboutHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code.")
+	}
+
+	request, err = http.NewRequest("GET", "/about", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/about)", err)
+	}
+	response = httptest.NewRecorder()
+
+	cookie := http.Cookie{Name: "lang", Value: "de"}
+	request.AddCookie(&cookie)
+
+	AboutHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code. Language: de")
+	}
+}
+
+func BenchmarkAboutHandler(b *testing.B) {
+	b.StopTimer()
+	request, _ := http.NewRequest("GET", "/about", nil)
+	response := httptest.NewRecorder()
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		AboutHandler(response, request)
+	}
+}
+
+func TestServerHandler(t *testing.T) {
+	request, err := http.NewRequest("GET", "/server", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/server)", err)
+	}
+	response := httptest.NewRecorder()
+
+	ServerHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code.")
+	}
+
+	request, err = http.NewRequest("GET", "/server", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/server)", err)
+	}
+	response = httptest.NewRecorder()
+
+	cookie := http.Cookie{Name: "lang", Value: "de"}
+	request.AddCookie(&cookie)
+
+	ServerHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code. Language: de")
+	}
+}
+
+func BenchmarkServerHandler(b *testing.B) {
+	b.StopTimer()
+	request, _ := http.NewRequest("GET", "/server", nil)
+	response := httptest.NewRecorder()
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		ServerHandler(response, request)
+	}
+}
+
+func TestHowtoHandler(t *testing.T) {
+	request, err := http.NewRequest("GET", "/howto", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/howto)", err)
+	}
+	response := httptest.NewRecorder()
+
+	HowtoHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code.")
+	}
+
+	request, err = http.NewRequest("GET", "/howto", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/howto)", err)
+	}
+	response = httptest.NewRecorder()
+
+	cookie := http.Cookie{Name: "lang", Value: "de"}
+	request.AddCookie(&cookie)
+
+	HowtoHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code. Language: de")
+	}
+}
+
+func BenchmarkHowtoHandler(b *testing.B) {
+	b.StopTimer()
+	request, _ := http.NewRequest("GET", "/howto", nil)
+	response := httptest.NewRecorder()
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		HowtoHandler(response, request)
+	}
+}
+
+func TestUserHandler(t *testing.T) {
+	request, err := http.NewRequest("GET", "/user", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/user)", err)
+	}
+	response := httptest.NewRecorder()
+
+	UserHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code.")
+	}
+
+	request, err = http.NewRequest("GET", "/user", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/user)", err)
+	}
+	response = httptest.NewRecorder()
+
+	cookie := http.Cookie{Name: "lang", Value: "de"}
+	request.AddCookie(&cookie)
+
+	UserHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code. Language: de")
+	}
+}
+
+func BenchmarkUserHandler(b *testing.B) {
+	b.StopTimer()
+	request, _ := http.NewRequest("GET", "/user", nil)
+	response := httptest.NewRecorder()
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		UserHandler(response, request)
+	}
+}
+
+func TestPasswordHandler(t *testing.T) {
+	request, err := http.NewRequest("GET", "/password", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/password)", err)
+	}
+	response := httptest.NewRecorder()
+
+	PasswordHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code.")
+	}
+
+	request, err = http.NewRequest("GET", "/password", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/password)", err)
+	}
+	response = httptest.NewRecorder()
+
+	cookie := http.Cookie{Name: "lang", Value: "de"}
+	request.AddCookie(&cookie)
+
+	PasswordHandler(response, request)
+
+	if response.Code != 200 {
+		t.Log(response.Code)
+		t.Fatal("Expected 200 as status code. Language: de")
+	}
+}
+
+func BenchmarkPasswordHandler(b *testing.B) {
+	b.StopTimer()
+	request, _ := http.NewRequest("GET", "/password", nil)
+	response := httptest.NewRecorder()
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		PasswordHandler(response, request)
+	}
+}
+
+func TestChangePasswordHandler(t *testing.T) {
+	form := url.Values{}
+
+	if err := CreateNewEntry("test@localhost", Md5Hash("Password")); err != nil {
+		t.Fatal(err)
+	}
+
+	form.Set("Email", "test@localhost")
+	form.Set("Password", "Password")
+	form.Set("ConfirmPassword", "password")
+	form.Set("NewPassword", "password")
+
+	request, err := http.NewRequest("POST", "/changePassword", strings.NewReader(form.Encode()))
+	if err != nil {
+		t.Log("Error creating new http request. (/changePassword)", err)
+	}
+	request.Header.Add("Content-Type", "application/x-www-form-urlencoded")
+	response := httptest.NewRecorder()
+
+	ChangePasswordHandler(response, request)
+
+	if response.Code != 302 {
+		t.Log(response.Code)
+		t.Fatal("Expected 302 as status code.")
+	}
+
+	flash := Flash{}
+	session, err := store.Get(request, "_SID")
+	if err != nil {
+		t.Log(err)
+	}
+	flash.Error = session.Flashes("error")
+	flash.Success = session.Flashes("success")
+
+	if len(flash.Error) > 0 {
+		for _, v := range flash.Error {
+			t.Fatal(v)
+		}
+		t.Fatal("Got error message.")
+	}
+
+	if len(flash.Success) == 0 {
+		t.Fatal("Expected success message.")
+	}
+
+	t.Log(flash.Success)
+}
+
+func TestChangeLocaleHandler(t *testing.T) {
+	// Test change locale to German
+	request, err := http.NewRequest("GET", "/locale?to=de", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/locale?to=de)", err)
+	}
+	response := httptest.NewRecorder()
+
+	ChangeLocaleHandler(response, request)
+
+	if response.Code != 302 {
+		t.Log(response.Code)
+		t.Fatal("Expected 302 as status code.")
+	}
+
+	if response.HeaderMap["Set-Cookie"][0] != "lang=de" {
+		t.Fatal("Expected Cookie 'lang=de")
+	}
+
+	// Test change locale to English
+	request, err = http.NewRequest("GET", "/locale?to=en", nil)
+	if err != nil {
+		t.Log("Error creating new http request. (/locale?to=en)", err)
+	}
+	response = httptest.NewRecorder()
+
+	ChangeLocaleHandler(response, request)
+
+	if response.Code != 302 {
+		t.Log(response.Code)
+		t.Fatal("Expected 302 as status code.")
+	}
+
+	if response.HeaderMap["Set-Cookie"][0] != "lang=en" {
+		t.Fatal("Expected Cookie 'lang=en'", response.HeaderMap["Set-Cookie"][0])
+	}
+
+	// Test change locale to Garbage
+	request, err = http.NewRequest("GET", "/locale?to=foobar", nil)
+	if err != nil {
+		t.Fatal("Error creating new http request. (/locale?to=foobar)", err)
+	}
+	response = httptest.NewRecorder()
+
+	ChangeLocaleHandler(response, request)
+
+	if response.Code != 302 {
+		t.Log(response.Code)
+		t.Fatal("Expected 302 as status code.")
+	}
+
+	if response.HeaderMap["Set-Cookie"][0] != "lang=foobar" {
+		t.Fatal("Expected Cookie 'lang=foobar'", response.HeaderMap["Set-Cookie"][0])
+	}
+}