fix ut

pingcap · Feb 7, 2023 · 5102a16 · 5102a16
1 parent 153a3b6
commit 5102a16
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dbms/src/Flash/Coprocessor/tests/gtest_ti_remote_block_inputstream.cpp b/dbms/src/Flash/Coprocessor/tests/gtest_ti_remote_block_inputstream.cpp
@@ -249,7 +249,10 @@ struct MockReceiverContext
         grpc::CompletionQueue *,
         UnaryCallback<bool> *) const {}
 
-    static ExchangePacketReaderPtr makeReader(const Request &) { return nullptr; }
+    std::shared_ptr<Reader> makeReader(const Request &)
+    {
+        return std::make_shared<Reader>(queue);
+    }
 
     static std::tuple<MPPTunnelPtr, grpc::Status> establishMPPConnectionLocalV1(const ::mpp::EstablishMPPConnectionRequest *, const std::shared_ptr<MPPTaskManager> &)
     {