test: TestHandlers_GetRequests の RequestQuery に Limit と Offset を追加

traPtitech · Dec 28, 2024 · ada6830 · ada6830
1 parent 8bc772f
commit ada6830
Showing 1 changed file with 28 additions and 7 deletions.
diff --git a/router/request_test.go b/router/request_test.go
@@ -67,7 +67,10 @@ func TestHandlers_GetRequests(t *testing.T) {
 		require.NoError(t, err)
 		h.Repository.MockRequestRepository.
 			EXPECT().
-			GetRequests(c.Request().Context(), model.RequestQuery{}).
+			GetRequests(c.Request().Context(), model.RequestQuery{
+				Limit:  100,
+				Offset: 0,
+			}).
 			Return(requests, nil)
 
 		res := []*RequestResponse{
@@ -122,7 +125,10 @@ func TestHandlers_GetRequests(t *testing.T) {
 		assert.NoError(t, err)
 		h.Repository.MockRequestRepository.
 			EXPECT().
-			GetRequests(c.Request().Context(), model.RequestQuery{}).
+			GetRequests(c.Request().Context(), model.RequestQuery{
+				Limit:  100,
+				Offset: 0,
+			}).
 			Return(requests, nil)
 
 		res := []*RequestResponse{}
@@ -167,6 +173,8 @@ func TestHandlers_GetRequests(t *testing.T) {
 			EXPECT().
 			GetRequests(c.Request().Context(), model.RequestQuery{
 				Status: &status,
+				Limit:  100,
+				Offset: 0,
 			}).
 			Return(requests, nil)
 
@@ -228,7 +236,9 @@ func TestHandlers_GetRequests(t *testing.T) {
 		h.Repository.MockRequestRepository.
 			EXPECT().
 			GetRequests(c.Request().Context(), model.RequestQuery{
-				Until: &date2,
+				Until:  &date2,
+				Limit:  100,
+				Offset: 0,
 			}).
 			Return(requests, nil)
 
@@ -290,7 +300,9 @@ func TestHandlers_GetRequests(t *testing.T) {
 		h.Repository.MockRequestRepository.
 			EXPECT().
 			GetRequests(c.Request().Context(), model.RequestQuery{
-				Since: &date2,
+				Since:  &date2,
+				Limit:  100,
+				Offset: 0,
 			}).
 			Return(requests, nil)
 
@@ -364,7 +376,9 @@ func TestHandlers_GetRequests(t *testing.T) {
 		h.Repository.MockRequestRepository.
 			EXPECT().
 			GetRequests(c.Request().Context(), model.RequestQuery{
-				Tag: &tag1.Name,
+				Tag:    &tag1.Name,
+				Limit:  100,
+				Offset: 0,
 			}).
 			Return(requests, nil)
 
@@ -420,7 +434,11 @@ func TestHandlers_GetRequests(t *testing.T) {
 		require.NoError(t, err)
 		h.Repository.MockRequestRepository.
 			EXPECT().
-			GetRequests(c.Request().Context(), model.RequestQuery{CreatedBy: &request.CreatedBy}).
+			GetRequests(c.Request().Context(), model.RequestQuery{
+				Limit:     100,
+				Offset:    0,
+				CreatedBy: &request.CreatedBy},
+			).
 			Return(modelRequests, nil)
 		err = h.Handlers.GetRequests(c)
 		if !assert.NoError(t, err) {
@@ -483,7 +501,10 @@ func TestHandlers_GetRequests(t *testing.T) {
 		resErr := errors.New("Failed to get requests.")
 		h.Repository.MockRequestRepository.
 			EXPECT().
-			GetRequests(c.Request().Context(), model.RequestQuery{}).
+			GetRequests(c.Request().Context(), model.RequestQuery{
+				Limit:  100,
+				Offset: 0,
+			}).
 			Return(nil, resErr)
 
 		err = h.Handlers.GetRequests(c)