diff --git a/app/rest/server_test.go b/app/rest/server_test.go index 11f6b9a6..fbca9d81 100644 --- a/app/rest/server_test.go +++ b/app/rest/server_test.go @@ -23,10 +23,7 @@ var testDb = "/tmp/test-remark.db" func TestServer_Ping(t *testing.T) { srv := prep(t) assert.NotNil(t, srv) - defer func() { - srv.httpServer.Shutdown(context.Background()) - os.Remove(testDb) - }() + defer cleanup(srv) res, code := get(t, "http://127.0.0.1:8080/api/v1/ping") assert.Equal(t, "pong", res) @@ -36,10 +33,7 @@ func TestServer_Ping(t *testing.T) { func TestServer_Create(t *testing.T) { srv := prep(t) assert.NotNil(t, srv) - defer func() { - srv.httpServer.Shutdown(context.Background()) - os.Remove(testDb) - }() + defer cleanup(srv) r := strings.NewReader(`{"text": "test 123", "locator":{"url": "https://radio-t.com/blah1", "site": "radio-t"}}`) resp, err := http.Post("http://127.0.0.1:8080/api/v1/comment", "application/json", r) @@ -60,10 +54,7 @@ func TestServer_Create(t *testing.T) { func TestServer_CreateAndGet(t *testing.T) { srv := prep(t) assert.NotNil(t, srv) - defer func() { - srv.httpServer.Shutdown(context.Background()) - os.Remove(testDb) - }() + defer cleanup(srv) // create comment r := strings.NewReader(`{"text": "test 123", "locator":{"url": "https://radio-t.com/blah1", "site": "radio-t"}}`) @@ -94,11 +85,8 @@ func TestServer_CreateAndGet(t *testing.T) { func TestServer_Find(t *testing.T) { srv := prep(t) assert.NotNil(t, srv) - defer func() { - srv.httpServer.Close() - srv.httpServer.Shutdown(context.Background()) - os.Remove(testDb) - }() + defer cleanup(srv) + _, code := get(t, "http://127.0.0.1:8080/api/v1/find?site=radio-t&url=https://radio-t.com/blah1") assert.Equal(t, 400, code, "nothing in") @@ -173,3 +161,9 @@ func addComment(t *testing.T, c store.Comment) string { return crResp["id"].(string) } + +func cleanup(srv *Server) { + srv.httpServer.Close() + srv.httpServer.Shutdown(context.Background()) + os.Remove(testDb) +}