Fix unit tests

ydb-platform · Dec 26, 2024 · 6112727 · 6112727
1 parent 8c58ea8
commit 6112727
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/app/server/datasource/rdbms/datasource_test.go b/app/server/datasource/rdbms/datasource_test.go
@@ -68,7 +68,7 @@ func TestReadSplit(t *testing.T) {
 		}
 
 		connection := &rdbms_utils.ConnectionMock{}
-		connection.On("From").Return("", "example_1").Twice()
+		connection.On("From").Return("", "example_1").Once()
 
 		connectionManager.On("Make", split.Select.DataSourceInstance).Return([]rdbms_utils.Connection{connection}, nil).Once()
 		connectionManager.On("Release", []rdbms_utils.Connection{connection}).Return().Once()
@@ -124,7 +124,7 @@ func TestReadSplit(t *testing.T) {
 		}
 
 		connection := &rdbms_utils.ConnectionMock{}
-		connection.On("From").Return("", "example_1").Twice()
+		connection.On("From").Return("", "example_1").Once()
 
 		connectionManager.On("Make", split.Select.DataSourceInstance).Return([]rdbms_utils.Connection{connection}, nil).Once()
 		connectionManager.On("Release", []rdbms_utils.Connection{connection}).Return().Once()

diff --git a/app/server/streaming/streamer_test.go b/app/server/streaming/streamer_test.go
@@ -155,7 +155,7 @@ func (tc testCaseStreaming) execute(t *testing.T) {
 	stream.On("Context").Return(ctx)
 
 	connection := &rdbms_utils.ConnectionMock{}
-	connection.On("From").Return("", "example_1").Twice()
+	connection.On("From").Return("", "example_1").Once()
 
 	connectionManager := &rdbms_utils.ConnectionManagerMock{}
 	connectionManager.On("Make", split.Select.DataSourceInstance).Return([]rdbms_utils.Connection{connection}, nil).Once()