fix: test issues

This commit is contained in:
Marvin Zhang
2024-11-05 13:46:22 +08:00
parent 5703504eb4
commit 6025f20b7d
2 changed files with 18 additions and 14 deletions

View File

@@ -173,3 +173,7 @@ func GetGrpcServer() *GrpcServer {
})
return _server
}
func NewGrpcServer() *GrpcServer {
return newGrpcServer()
}

View File

@@ -45,7 +45,7 @@ func stopSvr(svr *server.GrpcServer) {
func TestModelService_GetById(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -72,7 +72,7 @@ func TestModelService_GetById(t *testing.T) {
func TestModelService_GetOne(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -99,7 +99,7 @@ func TestModelService_GetOne(t *testing.T) {
func TestModelService_GetMany(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -127,7 +127,7 @@ func TestModelService_GetMany(t *testing.T) {
func TestModelService_DeleteById(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -156,7 +156,7 @@ func TestModelService_DeleteById(t *testing.T) {
func TestModelService_DeleteOne(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -185,7 +185,7 @@ func TestModelService_DeleteOne(t *testing.T) {
func TestModelService_DeleteMany(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -214,7 +214,7 @@ func TestModelService_DeleteMany(t *testing.T) {
func TestModelService_UpdateById(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -243,7 +243,7 @@ func TestModelService_UpdateById(t *testing.T) {
func TestModelService_UpdateOne(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -272,7 +272,7 @@ func TestModelService_UpdateOne(t *testing.T) {
func TestModelService_UpdateMany(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -305,7 +305,7 @@ func TestModelService_UpdateMany(t *testing.T) {
func TestModelService_ReplaceById(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -335,7 +335,7 @@ func TestModelService_ReplaceById(t *testing.T) {
func TestModelService_ReplaceOne(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -365,7 +365,7 @@ func TestModelService_ReplaceOne(t *testing.T) {
func TestModelService_InsertOne(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -388,7 +388,7 @@ func TestModelService_InsertOne(t *testing.T) {
func TestModelService_InsertMany(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)
@@ -414,7 +414,7 @@ func TestModelService_InsertMany(t *testing.T) {
func TestModelService_Count(t *testing.T) {
setupTestDB()
defer teardownTestDB()
svr := server.GetGrpcServer()
svr := server.NewGrpcServer()
go startSvr(svr)
defer stopSvr(svr)