feat: CoreDumpsRepository implementation and test

opencrasher · Sep 19, 2022 · 634a3ef · 634a3ef
1 parent f90fc91
commit 634a3ef
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 22 deletions.
diff --git a/server/internal/app/repositories/mongodb_repository/core_dumps_repository.go b/server/internal/app/repositories/mongodb_repository/core_dumps_repository.go
@@ -25,7 +25,7 @@ var _ repositories.CoreDumpsRepository = (*CoreDumpsRepository)(nil)
 
 func NewCoreDumpsRepository(db *mongo.Client, l *zap.Logger, timeout time.Duration) *CoreDumpsRepository {
 	mongoDB := db.Database(mongo_configs.DBname)
-	collection := mongoDB.Collection(mongo_configs.DBname)
+	collection := mongoDB.Collection(mongo_configs.CollectionName)
 
 	return &CoreDumpsRepository{
 		dbClient:   db,
@@ -61,7 +61,7 @@ func (r *CoreDumpsRepository) GetCoreDumps(parameters ...interface{}) ([]entitie
 	}
 
 	err = cur.All(ctx, &result)
-	
+
 	return result, err
 }
 

diff --git a/server/internal/app/repositories/mongodb_repository/mongo_configs/configuration.go b/server/internal/app/repositories/mongodb_repository/mongo_configs/configuration.go
@@ -1,6 +1,6 @@
 package mongo_configs
 
 var (
-	DBname = "crahser"
+	DBname         = "crashser"
 	CollectionName = "core_dumps"
-)
+)
diff --git a/server/internal/app/repositories/mongodb_repository/mongo_options/query_fields_mongo_test.go b/server/internal/app/repositories/mongodb_repository/mongo_options/query_fields_mongo_test.go
@@ -4,28 +4,27 @@ import (
 	"testing"
 	"time"
 
-	"github.com/stretchr/testify/require"
+	assert "github.com/stretchr/testify/require"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/bson/primitive"
 	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
 func TestSetApplication(t *testing.T) {
 	filter := bson.M{}
-	require.Equal(t, 0, len(filter))
+	assert.Equal(t, 0, len(filter))
 
 	options := options.Find()
 	setter := SetApplication("app")
 	setter(filter, options)
 	res := filter["appinfo.name"]
 
-	require.Equal(t, "app", res)
-	require.Equal(t, true, len(filter) > 0)
+	assert.Equal(t, "app", res)
 }
 
 func TestSetStartTimestamp(t *testing.T) {
 	filter := bson.M{}
-	require.Equal(t, 0, len(filter))
+	assert.Equal(t, 0, len(filter))
 
 	requiredDate := primitive.NewDateTimeFromTime(time.Now().AddDate(0, 0, -1))
 
@@ -34,13 +33,12 @@ func TestSetStartTimestamp(t *testing.T) {
 	setter(filter, options)
 	res := filter["timestamp"]
 
-	require.Equal(t, primitive.M(primitive.M{"$gte": requiredDate}), res)
-	require.Equal(t, true, len(filter) > 0)
+	assert.Equal(t, primitive.M(primitive.M{"$gte": requiredDate}), res)
 }
 
 func TestSetEndTimestamp(t *testing.T) {
 	filter := bson.M{}
-	require.Equal(t, 0, len(filter))
+	assert.Equal(t, 0, len(filter))
 
 	requiredDate := primitive.NewDateTimeFromTime(time.Now().AddDate(0, 0, 0))
 
@@ -49,34 +47,31 @@ func TestSetEndTimestamp(t *testing.T) {
 	setter(filter, options)
 	res := filter["timestamp"]
 
-	require.Equal(t, primitive.M(primitive.M{"$lt": requiredDate}), res)
-	require.Equal(t, true, len(filter) > 0)
+	assert.Equal(t, primitive.M(primitive.M{"$lt": requiredDate}), res)
 }
 
 func TestSetLimit(t *testing.T) {
 	filter := bson.M{}
-	require.Equal(t, 0, len(filter))
+	assert.Equal(t, 0, len(filter))
 
 	options := options.Find()
-	require.Equal(t, true, options.Limit == nil)
+	assert.Equal(t, true, options.Limit == nil)
 
 	setter := SetLimit(1)
 	setter(filter, options)
 
-	require.Equal(t, int64(1), *options.Limit)
-	require.Equal(t, false, options == nil)
+	assert.Equal(t, int64(1), *options.Limit)
 }
 
 func TestSetOffeset(t *testing.T) {
 	filter := bson.M{}
-	require.Equal(t, 0, len(filter))
+	assert.Equal(t, 0, len(filter))
 
 	options := options.Find()
-	require.Equal(t, true, options.Skip == nil)
+	assert.Equal(t, true, options.Skip == nil)
 
 	setter := SetOffset(1)
 	setter(filter, options)
 
-	require.Equal(t, int64(1), *options.Skip)
-	require.Equal(t, false, options == nil)
+	assert.Equal(t, int64(1), *options.Skip)
 }