diff options
Diffstat (limited to 'svc/handlers_test.go')
| -rw-r--r-- | svc/handlers_test.go | 71 |
1 files changed, 63 insertions, 8 deletions
diff --git a/svc/handlers_test.go b/svc/handlers_test.go index 36c795a..fa73de5 100644 --- a/svc/handlers_test.go +++ b/svc/handlers_test.go @@ -52,17 +52,72 @@ func Test_apiFormatHandler(t *testing.T) { } }) } + +var endpointCases = []struct { + name string + req *http.Request + status int +}{ + { + name: "Regular Query: /api/plain/users", + req: httptest.NewRequest("GET", "http://localhost"+testport+"/api/plain/users", nil), + status: http.StatusOK, + }, + { + name: "Regular Query: /api/plain/mentions", + req: httptest.NewRequest("GET", "http://localhost"+testport+"/api/plain/mentions", nil), + status: http.StatusOK, + }, + { + name: "Regular Query: /api/plain/tweets", + req: httptest.NewRequest("GET", "http://localhost"+testport+"/api/plain/tweets", nil), + status: http.StatusOK, + }, + { + name: "Invalid Endpoint: /api/plain/statuses", + req: httptest.NewRequest("GET", "http://localhost"+testport+"/api/plain/statuses", nil), + status: http.StatusNotFound, + }, +} + func Test_apiEndpointHandler(t *testing.T) { initTestConf() - t.Run("apiEndpointHandler", func(t *testing.T) { - w := httptest.NewRecorder() - req := httptest.NewRequest("GET", "localhost"+testport+"/api/plain/users", nil) - apiEndpointHandler(w, req) - resp := w.Result() - if resp.StatusCode != http.StatusOK { - t.Errorf(fmt.Sprintf("%v", resp.StatusCode)) + mockRegistry() + for _, tt := range endpointCases { + t.Run(tt.name, func(t *testing.T) { + w := httptest.NewRecorder() + apiEndpointHandler(w, tt.req) + resp := w.Result() + if resp.StatusCode != tt.status { + t.Errorf(fmt.Sprintf("%v", resp.StatusCode)) + } + if tt.status == http.StatusOK { + var body []byte + buf := bytes.NewBuffer(body) + err := resp.Write(buf) + if err != nil { + t.Errorf("%v\n", err) + } + if buf == nil { + t.Errorf("Got nil\n") + } + if len(buf.Bytes()) == 0 { + t.Errorf("Got zero data\n") + } + } + }) + } +} +func Benchmark_apiEndpointHandler(b *testing.B) { + initTestConf() + mockRegistry() + w := httptest.NewRecorder() + b.ResetTimer() + for _, tt := range endpointCases { + for i := 0; i < b.N; i++ { + apiEndpointHandler(w, tt.req) } - }) + } } func Test_apiTagsBaseHandler(t *testing.T) { |