(+) new server error

maulana48 · Jul 15, 2023 · 547c256 · 547c256
1 parent 347b84e
commit 547c256
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 3 deletions.
diff --git a/api/account_test.go b/api/account_test.go
@@ -6,6 +6,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
+	"log"
 	"net/http"
 	"net/http/httptest"
 	"testing"
@@ -87,10 +88,15 @@ func TestGetAccountAPI(t *testing.T) {
 			ctrl := gomock.NewController(t)
 			defer ctrl.Finish()
 
+			config, err := util.LoadConfig(".")
+			if err != nil {
+				log.Fatal("cannot load config:", err)
+			}
+
 			store := mockdb.NewMockStore(ctrl)
 			tc.buildStubs(store)
 
-			server, err := NewServer(store)
+			server, err := NewServer(config, store)
 			require.NoError(t, err)
 			recorder := httptest.NewRecorder()
 

diff --git a/db/sqlc/store.go b/db/sqlc/store.go
@@ -4,6 +4,8 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
+
+	_ "github.com/golang/mock/mockgen/model"
 )
 
 type Store interface {

diff --git a/main.go b/main.go
@@ -23,7 +23,7 @@ func main() {
 	}
 
 	store := db.NewStore(conn)
-	server, err := api.NewServer(store)
+	server, err := api.NewServer(config, store)
 	if err != nil {
 		log.Fatal("cannot start server:", err)
 	}

diff --git a/test/user_test.go b/test/user_test.go
@@ -6,6 +6,7 @@ import (
 	"database/sql"
 	"encoding/json"
 	"fmt"
+	"log"
 	"net/http"
 	"net/http/httptest"
 	"reflect"
@@ -219,11 +220,15 @@ func TestCreateUserAPI(t *testing.T) {
 		t.Run(tc.name, func(t *testing.T) {
 			ctrl := gomock.NewController(t)
 			defer ctrl.Finish()
+			config, err := util.LoadConfig(".")
+			if err != nil {
+				log.Fatal("cannot load config:", err)
+			}
 
 			store := mockdb.NewMockStore(ctrl)
 			tc.buildStubs(store)
 
-			server, err := api.NewServer(store)
+			server, err := api.NewServer(config, store)
 			recorder := httptest.NewRecorder()
 
 			// Marshal body data to JSON