1 files changed, 6 insertions, 1 deletions
diff --git a/cast/streaming/testing/simple_message_port.h b/cast/streaming/testing/simple_message_port.h
index 2d56dd33..7d912918 100644
--- a/cast/streaming/testing/simple_message_port.h
+++ b/cast/streaming/testing/simple_message_port.h
@@ -20,7 +20,12 @@ namespace cast {
 class SimpleMessagePort : public MessagePort {
  public:
   ~SimpleMessagePort() override {}
-  void SetClient(MessagePort::Client* client) override { client_ = client; }
+  void SetClient(MessagePort::Client* client,
+                 std::string client_sender_id) override {
+    client_ = client;
+  }
+
+  void ResetClient() override { client_ = nullptr; }
 
   void ReceiveMessage(const std::string& message) {
     ReceiveMessage(kCastWebrtcNamespace, message);