Intermediate changes

3 files changed, 70 insertions, 0 deletions

diff --git a/yt/yt/client/unittests/mock/table_reader.cpp b/yt/yt/client/unittests/mock/table_reader.cpp
new file mode 100644
index 0000000000..364381127c
--- /dev/null
+++ b/yt/yt/client/unittests/mock/table_reader.cpp
@@ -0,0 +1,27 @@
+#include "table_reader.h"
+
+#include <yt/yt/client/table_client/name_table.h>
+#include <yt/yt/client/table_client/schema.h>
+
+namespace NYT::NApi {
+
+////////////////////////////////////////////////////////////////////////////////
+
+TMockTableReader::TMockTableReader(const NTableClient::TTableSchemaPtr& schema)
+    : Schema_(schema)
+    , NameTable_(NTableClient::TNameTable::FromSchema(*Schema_))
+{}
+
+const NTableClient::TNameTablePtr& TMockTableReader::GetNameTable() const
+{
+    return NameTable_;
+}
+
+const NTableClient::TTableSchemaPtr& TMockTableReader::GetTableSchema() const
+{
+    return Schema_;
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+} // namespace NYT::NApi
diff --git a/yt/yt/client/unittests/mock/table_reader.h b/yt/yt/client/unittests/mock/table_reader.h
new file mode 100644
index 0000000000..7a8cf9825c
--- /dev/null
+++ b/yt/yt/client/unittests/mock/table_reader.h
@@ -0,0 +1,42 @@
+#pragma once
+
+#include <library/cpp/testing/gtest_extensions/gtest_extensions.h>
+
+#include <yt/yt/client/api/table_reader.h>
+
+namespace NYT::NApi {
+
+////////////////////////////////////////////////////////////////////////////////
+
+class TMockTableReader
+    : public ITableReader
+{
+public:
+    TMockTableReader(const NTableClient::TTableSchemaPtr& schema);
+
+    MOCK_METHOD(i64, GetStartRowIndex, (), (const, override));
+
+    MOCK_METHOD(i64, GetTotalRowCount, (), (const, override));
+
+    MOCK_METHOD(NChunkClient::NProto::TDataStatistics, GetDataStatistics, (), (const, override));
+
+    MOCK_METHOD(TFuture<void>, GetReadyEvent, (), (override));
+
+    MOCK_METHOD(NTableClient::IUnversionedRowBatchPtr, Read, (const NTableClient::TRowBatchReadOptions& options), (override));
+
+    MOCK_METHOD(const std::vector<TString>&, GetOmittedInaccessibleColumns, (), (const, override));
+
+    const NTableClient::TNameTablePtr& GetNameTable() const override;
+
+    const NTableClient::TTableSchemaPtr& GetTableSchema() const override;
+
+private:
+    NTableClient::TTableSchemaPtr Schema_;
+    NTableClient::TNameTablePtr NameTable_;
+};
+
+DEFINE_REFCOUNTED_TYPE(TMockTableReader);
+
+////////////////////////////////////////////////////////////////////////////////
+
+} // namespace NYT::NApi
diff --git a/yt/yt/client/unittests/mock/ya.make b/yt/yt/client/unittests/mock/ya.make
index bcc6df7a0c..9fa88a6aa5 100644
--- a/yt/yt/client/unittests/mock/ya.make
+++ b/yt/yt/client/unittests/mock/ya.make
@@ -4,6 +4,7 @@ INCLUDE(${ARCADIA_ROOT}/yt/ya_cpp.make.inc)
 
 SRCS(
     client.cpp
+    table_reader.cpp
 )
 
 PEERDIR(