Fixed generic unit tests

Author: GrigoriyPA

ydb/core/kqp/ut/federated_query/generic_ut/kqp_generic_provider_ut.cpp

@@ -239,6 +239,7 @@ namespace NKikimr::NKqp {
             std::vector<ui16> colData = {10, 20, 30, 40, 50};
             clientMock->ExpectReadSplits()
                 .Filtering(NYql::NConnector::NApi::TReadSplitsRequest::FILTERING_OPTIONAL)
+                .DataSourceInstance(dataSourceInstance)
                 .Split()
                     .Description("some binary description")
                     .Select()
@@ -360,6 +361,7 @@ namespace NKikimr::NKqp {
             // step 3: ReadSplits
             clientMock->ExpectReadSplits()
                 .Filtering(NYql::NConnector::NApi::TReadSplitsRequest::FILTERING_OPTIONAL)
+                .DataSourceInstance(dataSourceInstance)
                 .Split()
                     .Description("some binary description")
                     .Select()
@@ -477,6 +479,7 @@ namespace NKikimr::NKqp {
             // step 3: ReadSplits
             clientMock->ExpectReadSplits()
                 .Filtering(NYql::NConnector::NApi::TReadSplitsRequest::FILTERING_OPTIONAL)
+                .DataSourceInstance(dataSourceInstance)
                 .Split()
                     .Description("some binary description")
                     .Select()
@@ -598,6 +601,7 @@ namespace NKikimr::NKqp {
 
             f->ClientMock->ExpectReadSplits()
                 .Filtering(NYql::NConnector::NApi::TReadSplitsRequest::FILTERING_OPTIONAL)
+                .DataSourceInstance(f->DataSourceInstance)
                 .Split()
                     .Description("some binary description")
                     .Select(expectedSelect)