Refactoring: CastTransport InitializeAudio/InitializeVideo.

media/cast/test/cast_benchmarks.cc

Index: media/cast/test/cast_benchmarks.cc
diff --git a/media/cast/test/cast_benchmarks.cc b/media/cast/test/cast_benchmarks.cc
index 018b287c25f91b7bf2cfa1c5084397bb1271d6bb..8eeae6237d9ccf8fc07d73aec284081a8c27a9a1 100644
--- a/media/cast/test/cast_benchmarks.cc
+++ b/media/cast/test/cast_benchmarks.cc
@@ -97,16 +97,13 @@ class CastTransportWrapper : public CastTransport {
     encoded_audio_bytes_ = encoded_audio_bytes;
   }
 
-  void InitializeAudio(const CastTransportRtpConfig& config,
-                       std::unique_ptr<RtcpObserver> rtcp_observer) final {
-    audio_ssrc_ = config.ssrc;
-    transport_->InitializeAudio(config, std::move(rtcp_observer));
-  }
-
-  void InitializeVideo(const CastTransportRtpConfig& config,
-                       std::unique_ptr<RtcpObserver> rtcp_observer) final {
-    video_ssrc_ = config.ssrc;
-    transport_->InitializeVideo(config, std::move(rtcp_observer));
+  void InitializeStream(const CastTransportRtpConfig& config,
+                        std::unique_ptr<RtcpObserver> rtcp_observer) final {
+    if (config.rtp_payload_type <= RtpPayloadType::AUDIO_LAST)
+      audio_ssrc_ = config.ssrc;
+    else
+      video_ssrc_ = config.ssrc;
+    transport_->InitializeStream(config, std::move(rtcp_observer));
   }
 
   void InsertFrame(uint32_t ssrc, const EncodedFrame& frame) final {
@@ -456,9 +453,7 @@ class TransportClient : public CastTransport::Client {
       : run_one_benchmark_(run_one_benchmark) {}
 
   void OnStatusChanged(CastTransportStatus status) final {
-    bool result = (status == TRANSPORT_AUDIO_INITIALIZED ||
-                   status == TRANSPORT_VIDEO_INITIALIZED);
-    EXPECT_TRUE(result);
+    EXPECT_EQ(TRANSPORT_STREAM_INITIALIZED, status);
   };
   void OnLoggingEventsReceived(
       std::unique_ptr<std::vector<FrameEvent>> frame_events,