3 files changed, 29 insertions, 6 deletions
diff --git a/yt/yt/client/unittests/mock/table_reader.cpp b/yt/yt/client/unittests/mock/table_reader.cpp
index 364381127c0..ef3c44d0bde 100644
--- a/yt/yt/client/unittests/mock/table_reader.cpp
+++ b/yt/yt/client/unittests/mock/table_reader.cpp
@@ -7,10 +7,10 @@ namespace NYT::NApi {
 
 ////////////////////////////////////////////////////////////////////////////////
 
-TMockTableReader::TMockTableReader(const NTableClient::TTableSchemaPtr& schema)
-    : Schema_(schema)
+TMockTableReader::TMockTableReader(NTableClient::TTableSchemaPtr schema)
+    : Schema_(std::move(schema))
     , NameTable_(NTableClient::TNameTable::FromSchema(*Schema_))
-{}
+{ }
 
 const NTableClient::TNameTablePtr& TMockTableReader::GetNameTable() const
 {
diff --git a/yt/yt/client/unittests/mock/table_reader.h b/yt/yt/client/unittests/mock/table_reader.h
index 36ddda47ab9..be901f48814 100644
--- a/yt/yt/client/unittests/mock/table_reader.h
+++ b/yt/yt/client/unittests/mock/table_reader.h
@@ -12,7 +12,7 @@ class TMockTableReader
     : public ITableReader
 {
 public:
-    TMockTableReader(const NTableClient::TTableSchemaPtr& schema);
+    explicit TMockTableReader(NTableClient::TTableSchemaPtr schema);
 
     MOCK_METHOD(i64, GetStartRowIndex, (), (const, override));
 
@@ -31,8 +31,8 @@ public:
     const NTableClient::TTableSchemaPtr& GetTableSchema() const override;
 
 private:
-    NTableClient::TTableSchemaPtr Schema_;
-    NTableClient::TNameTablePtr NameTable_;
+    const NTableClient::TTableSchemaPtr Schema_;
+    const NTableClient::TNameTablePtr NameTable_;
 };
 
 DEFINE_REFCOUNTED_TYPE(TMockTableReader);
diff --git a/yt/yt/client/unittests/mock/timestamp_provider.h b/yt/yt/client/unittests/mock/timestamp_provider.h
new file mode 100644
index 00000000000..2a5f861a530
--- /dev/null
+++ b/yt/yt/client/unittests/mock/timestamp_provider.h
@@ -0,0 +1,23 @@
+#pragma once
+
+#include <yt/yt/client/transaction_client/timestamp_provider.h>
+
+namespace NYT::NTransactionClient {
+
+////////////////////////////////////////////////////////////////////////////////
+
+DECLARE_REFCOUNTED_CLASS(TMockTimestampProvider);
+
+class TMockTimestampProvider
+    : public ITimestampProvider
+{
+public:
+    MOCK_METHOD(TFuture<TTimestamp>, GenerateTimestamps, (int, NObjectClient::TCellTag), (override));
+    MOCK_METHOD(TTimestamp, GetLatestTimestamp, (NObjectClient::TCellTag), (override));
+};
+
+DEFINE_REFCOUNTED_TYPE(TMockTimestampProvider)
+
+////////////////////////////////////////////////////////////////////////////////
+
+} // namespace NYT::NTransactionClient