git-sync/crawlab
1
0
Fork 0
mirror of https://github.com/crawlab-team/crawlab.git synced 2026-01-22 17:31:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

test: updated test cases

Browse Source
This commit is contained in:
Marvin Zhang
2024-10-20 17:55:57 +08:00
parent b5e888f3e1
commit cecf4dfd1d
20 changed files with 79 additions and 321 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
core/models/client/model_service_v2.go
View File

@@ -6,7 +6,7 @@ import (
"github.com/crawlab-team/crawlab/core/interfaces"
nodeconfig "github.com/crawlab-team/crawlab/core/node/config"
"github.com/crawlab-team/crawlab/db/mongo"
grpc "github.com/crawlab-team/crawlab/grpc"
"github.com/crawlab-team/crawlab/grpc"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"reflect"

92
core/models/client/model_service_v2_test.go
View File

@@ -2,10 +2,14 @@ package client_test
import (
"context"
"testing"
"time"
"github.com/crawlab-team/crawlab/core/models/models/v2"
"github.com/apex/log"
"github.com/crawlab-team/crawlab/core/grpc/server"
"github.com/crawlab-team/crawlab/core/models/client"
"github.com/crawlab-team/crawlab/core/models/models/v2"
"github.com/crawlab-team/crawlab/core/models/service"
"github.com/crawlab-team/crawlab/db/mongo"
"github.com/spf13/viper"
@@ -14,12 +18,8 @@ import (
"go.mongodb.org/mongo-driver/bson"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
"testing"
"time"
)
type TestModel models.TestModelV2
func setupTestDB() {
viper.Set("mongo.db", "testdb")
}
@@ -51,10 +51,10 @@ func TestModelServiceV2_GetById(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
m := TestModel{
m := models.TestModelV2{
Name: "Test Name",
}
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
id, err := modelSvc.InsertOne(m)
require.Nil(t, err)
m.SetId(id)
@@ -64,7 +64,7 @@ func TestModelServiceV2_GetById(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
res, err := clientSvc.GetById(m.Id)
require.Nil(t, err)
assert.Equal(t, res.Id, m.Id)
@@ -79,10 +79,10 @@ func TestModelServiceV2_GetOne(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
m := TestModel{
m := models.TestModelV2{
Name: "Test Name",
}
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
id, err := modelSvc.InsertOne(m)
require.Nil(t, err)
m.SetId(id)
@@ -92,7 +92,7 @@ func TestModelServiceV2_GetOne(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
res, err := clientSvc.GetOne(bson.M{"name": m.Name}, nil)
require.Nil(t, err)
assert.Equal(t, res.Id, m.Id)
@@ -107,10 +107,10 @@ func TestModelServiceV2_GetMany(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
m := TestModel{
m := models.TestModelV2{
Name: "Test Name",
}
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
id, err := modelSvc.InsertOne(m)
require.Nil(t, err)
m.SetId(id)
@@ -120,7 +120,7 @@ func TestModelServiceV2_GetMany(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
res, err := clientSvc.GetMany(bson.M{"name": m.Name}, nil)
require.Nil(t, err)
assert.Equal(t, len(res), 1)
@@ -136,10 +136,10 @@ func TestModelServiceV2_DeleteById(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
m := TestModel{
m := models.TestModelV2{
Name: "Test Name",
}
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
id, err := modelSvc.InsertOne(m)
require.Nil(t, err)
m.SetId(id)
@@ -149,7 +149,7 @@ func TestModelServiceV2_DeleteById(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
err = clientSvc.DeleteById(m.Id)
require.Nil(t, err)
@@ -166,10 +166,10 @@ func TestModelServiceV2_DeleteOne(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
m := TestModel{
m := models.TestModelV2{
Name: "Test Name",
}
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
id, err := modelSvc.InsertOne(m)
require.Nil(t, err)
m.SetId(id)
@@ -179,7 +179,7 @@ func TestModelServiceV2_DeleteOne(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
err = clientSvc.DeleteOne(bson.M{"name": m.Name})
require.Nil(t, err)
@@ -196,10 +196,10 @@ func TestModelServiceV2_DeleteMany(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
m := TestModel{
m := models.TestModelV2{
Name: "Test Name",
}
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
id, err := modelSvc.InsertOne(m)
require.Nil(t, err)
m.SetId(id)
@@ -209,7 +209,7 @@ func TestModelServiceV2_DeleteMany(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
err = clientSvc.DeleteMany(bson.M{"name": m.Name})
require.Nil(t, err)
@@ -226,10 +226,10 @@ func TestModelServiceV2_UpdateById(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
m := TestModel{
m := models.TestModelV2{
Name: "Test Name",
}
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
id, err := modelSvc.InsertOne(m)
require.Nil(t, err)
m.SetId(id)
@@ -239,7 +239,7 @@ func TestModelServiceV2_UpdateById(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
err = clientSvc.UpdateById(m.Id, bson.M{"$set": bson.M{"name": "New Name"}})
require.Nil(t, err)
@@ -256,10 +256,10 @@ func TestModelServiceV2_UpdateOne(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
m := TestModel{
m := models.TestModelV2{
Name: "Test Name",
}
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
id, err := modelSvc.InsertOne(m)
require.Nil(t, err)
m.SetId(id)
@@ -269,7 +269,7 @@ func TestModelServiceV2_UpdateOne(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
err = clientSvc.UpdateOne(bson.M{"name": m.Name}, bson.M{"$set": bson.M{"name": "New Name"}})
require.Nil(t, err)
@@ -286,13 +286,13 @@ func TestModelServiceV2_UpdateMany(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
m1 := TestModel{
m1 := models.TestModelV2{
Name: "Test Name",
}
m2 := TestModel{
m2 := models.TestModelV2{
Name: "Test Name",
}
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
_, err = modelSvc.InsertOne(m1)
require.Nil(t, err)
_, err = modelSvc.InsertOne(m2)
@@ -303,7 +303,7 @@ func TestModelServiceV2_UpdateMany(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
err = clientSvc.UpdateMany(bson.M{"name": "Test Name"}, bson.M{"$set": bson.M{"name": "New Name"}})
require.Nil(t, err)
@@ -320,10 +320,10 @@ func TestModelServiceV2_ReplaceById(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
m := TestModel{
m := models.TestModelV2{
Name: "Test Name",
}
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
id, err := modelSvc.InsertOne(m)
require.Nil(t, err)
m.SetId(id)
@@ -333,7 +333,7 @@ func TestModelServiceV2_ReplaceById(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
m.Name = "New Name"
err = clientSvc.ReplaceById(m.Id, m)
require.Nil(t, err)
@@ -351,10 +351,10 @@ func TestModelServiceV2_ReplaceOne(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
m := TestModel{
m := models.TestModelV2{
Name: "Test Name",
}
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
id, err := modelSvc.InsertOne(m)
require.Nil(t, err)
m.SetId(id)
@@ -364,7 +364,7 @@ func TestModelServiceV2_ReplaceOne(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
m.Name = "New Name"
err = clientSvc.ReplaceOne(bson.M{"name": "Test Name"}, m)
require.Nil(t, err)
@@ -386,8 +386,8 @@ func TestModelServiceV2_InsertOne(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
m := TestModel{
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
m := models.TestModelV2{
Name: "Test Name",
}
id, err := clientSvc.InsertOne(m)
@@ -410,8 +410,8 @@ func TestModelServiceV2_InsertMany(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
testModels := []TestModel{
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
testModels := []models.TestModelV2{
{Name: "Test Name 1"},
{Name: "Test Name 2"},
}
@@ -433,9 +433,9 @@ func TestModelServiceV2_Count(t *testing.T) {
go startSvr(svr)
defer stopSvr(svr)
modelSvc := service.NewModelServiceV2[TestModel]()
modelSvc := service.NewModelServiceV2[models.TestModelV2]()
for i := 0; i < 5; i++ {
_, err = modelSvc.InsertOne(TestModel{
_, err = modelSvc.InsertOne(models.TestModelV2{
Name: "Test Name",
})
require.Nil(t, err)
@@ -446,7 +446,7 @@ func TestModelServiceV2_Count(t *testing.T) {
require.Nil(t, err)
c.Connect()
clientSvc := client.NewModelServiceV2[TestModel]()
clientSvc := client.NewModelServiceV2[models.TestModelV2]()
count, err := clientSvc.Count(bson.M{})
require.Nil(t, err)