fix: intergrattion test

Signed-off-by: Vladislav Sukhin <[email protected]>

pkg/repository/sequence/mongo.go

@@ -98,7 +98,7 @@ func (r *MongoRepository) GetNextExecutionNumber(ctx context.Context, name strin
 	opts.SetReturnDocument(options.After)
 
 	err = r.Coll.FindOneAndUpdate(ctx, bson.M{"_id": id}, bson.M{"$inc": bson.M{"number": 1}}, opts).Decode(&executionNumber)
-	if err == nil {
+	if err != nil {
 		return 0, err
 	}
 

pkg/repository/sequence/mongo_integration_test.go

@@ -32,27 +32,39 @@ func TestNewMongoRepository_GetNextExecutionNumber_Sequential_Integration(t *tes
 
 	repo := NewMongoRepository(db)
 
-	num1, err := repo.GetNextExecutionNumber(ctx, "name", ExecutionTypeTest)
-	assert.NoError(t, err)
-	assert.Equal(t, 1, num1)
-
-	num2, err := repo.GetNextExecutionNumber(ctx, "name", ExecutionTypeTest)
-	assert.NoError(t, err)
-	assert.Equal(t, 2, num2)
-
-	num3, err := repo.GetNextExecutionNumber(ctx, "name", ExecutionTypeTestSuite)
-	assert.NoError(t, err)
-	assert.Equal(t, 1, num3)
-
-	num4, err := repo.GetNextExecutionNumber(ctx, "name", ExecutionTypeTestSuite)
-	assert.NoError(t, err)
-	assert.Equal(t, 2, num4)
-
-	num5, err := repo.GetNextExecutionNumber(ctx, "name", ExecutionTypeTestWorkflow)
-	assert.NoError(t, err)
-	assert.Equal(t, 1, num5)
+	var tests = []struct {
+		expectedValue int32
+		executionType ExecutionType
+	}{
+		{
+			1,
+			ExecutionTypeTest,
+		},
+		{
+			2,
+			ExecutionTypeTest,
+		},
+		{
+			1,
+			ExecutionTypeTestSuite,
+		},
+		{
+			2,
+			ExecutionTypeTestSuite,
+		},
+		{
+			1,
+			ExecutionTypeTestWorkflow,
+		},
+		{
+			2,
+			ExecutionTypeTestWorkflow,
+		},
+	}
 
-	num6, err := repo.GetNextExecutionNumber(ctx, "name", ExecutionTypeTestWorkflow)
-	assert.NoError(t, err)
-	assert.Equal(t, 2, num6)
+	for _, tt := range tests {
+		num, err := repo.GetNextExecutionNumber(ctx, "name", tt.executionType)
+		assert.NoError(t, err)
+		assert.Equal(t, tt.expectedValue, num)
+	}
 }