Import existing source code

idicon_test.go

@@ -0,0 +1,91 @@
+package main
+
+import (
+	"bytes"
+	_ "embed"
+	"github.com/gorilla/mux"
+	"image/png"
+	"net/http"
+	"net/http/httptest"
+	"reflect"
+	"testing"
+)
+
+//go:embed testdata/1a79a4d60de6718e8e5b326e338ae533_v1.png
+var v1 []byte
+
+//go:embed testdata/1a79a4d60de6718e8e5b326e338ae533_v2.png
+var v2 []byte
+
+func testRouter() *mux.Router {
+	router := mux.NewRouter()
+	router.HandleFunc("/avatar/{id}", RequestHandler)
+	return router
+}
+
+func TestCorrectContentType(t *testing.T) {
+	req, err := http.NewRequest("GET", "/avatar/example", nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	rr := httptest.NewRecorder()
+	testRouter().ServeHTTP(rr, req)
+
+	if ctype := rr.Header().Get("Content-Type"); ctype != "image/png" {
+		t.Errorf("content type header does not match: got %v want image/png", ctype)
+	}
+}
+
+func TestCorrectResponseForV1ColorScheme(t *testing.T) {
+	req, err := http.NewRequest("GET", "/avatar/1a79a4d60de6718e8e5b326e338ae533?c=v1", nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	rr := httptest.NewRecorder()
+	testRouter().ServeHTTP(rr, req)
+
+	if !reflect.DeepEqual(rr.Body.Bytes(), v1) {
+		t.Errorf("returned image does not match expected image")
+	}
+}
+
+func TestCorrectResponseForV2ColorScheme(t *testing.T) {
+	req, err := http.NewRequest("GET", "/avatar/1a79a4d60de6718e8e5b326e338ae533?c=v2", nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	rr := httptest.NewRecorder()
+	testRouter().ServeHTTP(rr, req)
+
+	if !reflect.DeepEqual(rr.Body.Bytes(), v2) {
+		t.Errorf("returned image does not match expected image")
+	}
+}
+
+func TestIgnoreCase(t *testing.T) {
+	w1 := bytes.NewBuffer([]byte{})
+	png.Encode(w1, genIdIcon("example", 80, colorV1))
+
+	w2 := bytes.NewBuffer([]byte{})
+	png.Encode(w2, genIdIcon("Example", 80, colorV1))
+
+	if bytes.Compare(w1.Bytes(), w2.Bytes()) != 0 {
+		t.Errorf("resulting images do not match")
+	}
+}
+
+func TestStringMatchesHash(t *testing.T) {
+	w1 := bytes.NewBuffer([]byte{})
+	// MD5 of lowercase 'example'
+	png.Encode(w1, genIdIcon("1a79a4d60de6718e8e5b326e338ae533", 80, colorV2))
+
+	w2 := bytes.NewBuffer([]byte{})
+	png.Encode(w2, genIdIcon("Example", 80, colorV2))
+
+	if bytes.Compare(w1.Bytes(), w2.Bytes()) != 0 {
+		t.Errorf("resulting images do not match")
+	}
+}