test(router): update unit test on feching all books

jeamon · Nov 13, 2023 · 4d04c97 · 4d04c97
1 parent 0fa22a6
commit 4d04c97
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests.router_test.go b/tests.router_test.go
@@ -6,6 +6,7 @@ import (
 	"net/http"
 	"net/http/httptest"
 	"testing"
+	"time"
 
 	"github.com/julienschmidt/httprouter"
 	"github.com/stretchr/testify/assert"
@@ -96,7 +97,8 @@ func TestSetupBookRoutes(t *testing.T) {
 	}
 
 	bs := NewBookService(zap.NewNop(), nil, NewMockClocker(), mockRepo, mockRepo, mockQueue)
-	api := NewAPIHandler(zap.NewNop(), nil, &Statistics{started: NewMockClocker().Now()}, NewMockClocker(), NewMockUIDHandler("", true), bs)
+	api := NewAPIHandler(zap.NewNop(), &Config{}, &Statistics{started: NewMockClocker().Now()}, NewMockClocker(), NewMockUIDHandler("", true), bs)
+	api.config.Server.LongRequestWriteTimeout = time.Second
 	router := httprouter.New()
 	m := &MiddlewareMap{public: (&Middlewares{}).Chain, ops: (&Middlewares{}).Chain}
 	api.SetupBookRoutes(router, m)