package middleware import ( "net/http" "net/http/httptest" "strings" "testing" ) func TestNewReplacer(t *testing.T) { w := httptest.NewRecorder() recordRequest := NewResponseRecorder(w) userJSON := `{"username": "dennis"}` reader := strings.NewReader(userJSON) //Convert string to reader request, err := http.NewRequest("POST", "http://caddyserver.com", reader) //Create request with JSON body if err != nil { t.Fatalf("Request Formation Failed \n") } replaceValues := NewReplacer(request, recordRequest, "") switch v := replaceValues.(type) { case replacer: if v.replacements["{host}"] != "caddyserver.com" { t.Errorf("Expected host to be caddyserver.com") } if v.replacements["{method}"] != "POST" { t.Errorf("Expected request method to be POST") } if v.replacements["{status}"] != "200" { t.Errorf("Expected status to be 200") } default: t.Fatalf("Return Value from New Replacer expected pass type assertion into a replacer type \n") } } func TestReplace(t *testing.T) { w := httptest.NewRecorder() recordRequest := NewResponseRecorder(w) userJSON := `{"username": "dennis"}` reader := strings.NewReader(userJSON) //Convert string to reader request, err := http.NewRequest("POST", "http://caddyserver.com", reader) //Create request with JSON body if err != nil { t.Fatalf("Request Formation Failed \n") } replaceValues := NewReplacer(request, recordRequest, "") switch v := replaceValues.(type) { case replacer: if v.Replace("This host is {host}") != "This host is caddyserver.com" { t.Errorf("Expected host replacement failed") } if v.Replace("This request method is {method}") != "This request method is POST" { t.Errorf("Expected method replacement failed") } if v.Replace("The response status is {status}") != "The response status is 200" { t.Errorf("Expected status replacement failed") } default: t.Fatalf("Return Value from New Replacer expected pass type assertion into a replacer type \n") } }