Method TestDatabaseCleanup

internal/api/middleware_test.go:783–835

()

}

func (ts *MiddlewareTestSuite) TestDatabaseCleanup() {
	testHandler := func(statusCode int) http.HandlerFunc {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			w.WriteHeader(statusCode)
			b, _ := json.Marshal(map[string]interface{}{"message": "ok"})
			w.Write([]byte(b))
		})
	}

	cases := []struct {
		desc       string
		statusCode int
		method     string
	}{
		{
			desc:       "Run cleanup successfully",
			statusCode: http.StatusOK,
			method:     http.MethodPost,
		},
		{
			desc:       "Skip cleanup if GET",
			statusCode: http.StatusOK,
			method:     http.MethodGet,
		},
		{
			desc:       "Skip cleanup if 3xx",
			statusCode: http.StatusSeeOther,
			method:     http.MethodPost,
		},
		{
			desc:       "Skip cleanup if 4xx",
			statusCode: http.StatusBadRequest,
			method:     http.MethodPost,
		},
		{
			desc:       "Skip cleanup if 5xx",
			statusCode: http.StatusInternalServerError,
			method:     http.MethodPost,
		},
	}

	mockCleanup := new(MockCleanup)
	mockCleanup.On("Clean", mock.Anything).Return(0, nil)
	for _, c := range cases {
		ts.Run("DatabaseCleanup", func() {
			req := httptest.NewRequest(c.method, "http://localhost", nil)
			w := httptest.NewRecorder()
			ts.API.databaseCleanup(mockCleanup)(testHandler(c.statusCode)).ServeHTTP(w, req)
			require.Equal(ts.T(), c.statusCode, w.Code)
		})
	}
	mockCleanup.AssertNumberOfCalls(ts.T(), "Clean", 1)
}

func (ts *MiddlewareTestSuite) TestRequireAdminCredentialsSessionCheck() {
	models.TruncateAll(ts.API.db)