| Index: webrtc/pc/dtmfsender_unittest.cc | 
| diff --git a/webrtc/pc/dtmfsender_unittest.cc b/webrtc/pc/dtmfsender_unittest.cc | 
| index a58c1ec91f5c953f169d8b0cf58c67aa4e0b51e5..109760c3824a4a976adadfe62dc6a3d935ef9f36 100644 | 
| --- a/webrtc/pc/dtmfsender_unittest.cc | 
| +++ b/webrtc/pc/dtmfsender_unittest.cc | 
| @@ -78,13 +78,9 @@ class FakeDtmfProvider : public DtmfProviderInterface { | 
| } | 
|  | 
| // Implements DtmfProviderInterface. | 
| -  bool CanInsertDtmf(const std::string& track_label) override { | 
| -    return (can_insert_dtmf_tracks_.count(track_label) != 0); | 
| -  } | 
| +  bool CanInsertDtmf() override { return can_insert_; } | 
|  | 
| -  bool InsertDtmf(const std::string& track_label, | 
| -                  int code, | 
| -                  int duration) override { | 
| +  bool InsertDtmf(int code, int duration) override { | 
| int gap = 0; | 
| // TODO(ronghuawu): Make the timer (basically the rtc::TimeNanos) | 
| // mockable and use a fake timer in the unit tests. | 
| @@ -110,15 +106,10 @@ class FakeDtmfProvider : public DtmfProviderInterface { | 
| } | 
|  | 
| // helper functions | 
| -  void AddCanInsertDtmfTrack(const std::string& label) { | 
| -    can_insert_dtmf_tracks_.insert(label); | 
| -  } | 
| -  void RemoveCanInsertDtmfTrack(const std::string& label) { | 
| -    can_insert_dtmf_tracks_.erase(label); | 
| -  } | 
| +  void SetCanInsertDtmf(bool can_insert) { can_insert_ = can_insert; } | 
|  | 
| private: | 
| -  std::set<std::string> can_insert_dtmf_tracks_; | 
| +  bool can_insert_ = false; | 
| std::vector<DtmfInfo> dtmf_info_queue_; | 
| int64_t last_insert_dtmf_call_; | 
| sigslot::signal0<> SignalDestroyed; | 
| @@ -130,7 +121,7 @@ class DtmfSenderTest : public testing::Test { | 
| : track_(AudioTrack::Create(kTestAudioLabel, NULL)), | 
| observer_(new rtc::RefCountedObject<FakeDtmfObserver>()), | 
| provider_(new FakeDtmfProvider()) { | 
| -    provider_->AddCanInsertDtmfTrack(kTestAudioLabel); | 
| +    provider_->SetCanInsertDtmf(true); | 
| dtmf_ = DtmfSender::Create(track_, rtc::Thread::Current(), | 
| provider_.get()); | 
| dtmf_->RegisterObserver(observer_.get()); | 
| @@ -227,7 +218,7 @@ class DtmfSenderTest : public testing::Test { | 
|  | 
| TEST_F(DtmfSenderTest, CanInsertDtmf) { | 
| EXPECT_TRUE(dtmf_->CanInsertDtmf()); | 
| -  provider_->RemoveCanInsertDtmfTrack(kTestAudioLabel); | 
| +  provider_->SetCanInsertDtmf(false); | 
| EXPECT_FALSE(dtmf_->CanInsertDtmf()); | 
| } | 
|  | 
| @@ -333,7 +324,7 @@ TEST_F(DtmfSenderTest, TryInsertDtmfWhenItDoesNotWork) { | 
| std::string tones = "3,4"; | 
| int duration = 100; | 
| int inter_tone_gap = 50; | 
| -  provider_->RemoveCanInsertDtmfTrack(kTestAudioLabel); | 
| +  provider_->SetCanInsertDtmf(false); | 
| EXPECT_FALSE(dtmf_->InsertDtmf(tones, duration, inter_tone_gap)); | 
| } | 
|  | 
|  |