Index: webrtc/api/dtmfsender_unittest.cc |
diff --git a/webrtc/api/dtmfsender_unittest.cc b/webrtc/api/dtmfsender_unittest.cc |
deleted file mode 100644 |
index 54f3a7b96d6f6a3ec4355bd86b0eb77eb4ddd136..0000000000000000000000000000000000000000 |
--- a/webrtc/api/dtmfsender_unittest.cc |
+++ /dev/null |
@@ -1,350 +0,0 @@ |
-/* |
- * Copyright 2012 The WebRTC project authors. All Rights Reserved. |
- * |
- * Use of this source code is governed by a BSD-style license |
- * that can be found in the LICENSE file in the root of the source |
- * tree. An additional intellectual property rights grant can be found |
- * in the file PATENTS. All contributing project authors may |
- * be found in the AUTHORS file in the root of the source tree. |
- */ |
- |
-#include "webrtc/api/dtmfsender.h" |
- |
-#include <memory> |
-#include <set> |
-#include <string> |
-#include <vector> |
- |
-#include "webrtc/api/audiotrack.h" |
-#include "webrtc/base/fakeclock.h" |
-#include "webrtc/base/gunit.h" |
-#include "webrtc/base/logging.h" |
-#include "webrtc/base/timeutils.h" |
- |
-using webrtc::AudioTrackInterface; |
-using webrtc::AudioTrack; |
-using webrtc::DtmfProviderInterface; |
-using webrtc::DtmfSender; |
-using webrtc::DtmfSenderObserverInterface; |
- |
-static const char kTestAudioLabel[] = "test_audio_track"; |
-// TODO(deadbeef): Even though this test now uses a fake clock, it has a |
-// generous 3-second timeout for every test case. The timeout could be tuned |
-// to each test based on the tones sent, instead. |
-static const int kMaxWaitMs = 3000; |
- |
-class FakeDtmfObserver : public DtmfSenderObserverInterface { |
- public: |
- FakeDtmfObserver() : completed_(false) {} |
- |
- // Implements DtmfSenderObserverInterface. |
- void OnToneChange(const std::string& tone) override { |
- LOG(LS_VERBOSE) << "FakeDtmfObserver::OnToneChange '" << tone << "'."; |
- tones_.push_back(tone); |
- if (tone.empty()) { |
- completed_ = true; |
- } |
- } |
- |
- // getters |
- const std::vector<std::string>& tones() const { |
- return tones_; |
- } |
- bool completed() const { |
- return completed_; |
- } |
- |
- private: |
- std::vector<std::string> tones_; |
- bool completed_; |
-}; |
- |
-class FakeDtmfProvider : public DtmfProviderInterface { |
- public: |
- struct DtmfInfo { |
- DtmfInfo(int code, int duration, int gap) |
- : code(code), |
- duration(duration), |
- gap(gap) {} |
- int code; |
- int duration; |
- int gap; |
- }; |
- |
- FakeDtmfProvider() : last_insert_dtmf_call_(0) {} |
- |
- ~FakeDtmfProvider() { |
- SignalDestroyed(); |
- } |
- |
- // Implements DtmfProviderInterface. |
- bool CanInsertDtmf(const std::string& track_label) override { |
- return (can_insert_dtmf_tracks_.count(track_label) != 0); |
- } |
- |
- bool InsertDtmf(const std::string& track_label, |
- 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. |
- if (last_insert_dtmf_call_ > 0) { |
- gap = static_cast<int>(rtc::TimeMillis() - last_insert_dtmf_call_); |
- } |
- last_insert_dtmf_call_ = rtc::TimeMillis(); |
- |
- LOG(LS_VERBOSE) << "FakeDtmfProvider::InsertDtmf code=" << code |
- << " duration=" << duration |
- << " gap=" << gap << "."; |
- dtmf_info_queue_.push_back(DtmfInfo(code, duration, gap)); |
- return true; |
- } |
- |
- sigslot::signal0<>* GetOnDestroyedSignal() override { |
- return &SignalDestroyed; |
- } |
- |
- // getter and setter |
- const std::vector<DtmfInfo>& dtmf_info_queue() const { |
- return dtmf_info_queue_; |
- } |
- |
- // 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); |
- } |
- |
- private: |
- std::set<std::string> can_insert_dtmf_tracks_; |
- std::vector<DtmfInfo> dtmf_info_queue_; |
- int64_t last_insert_dtmf_call_; |
- sigslot::signal0<> SignalDestroyed; |
-}; |
- |
-class DtmfSenderTest : public testing::Test { |
- protected: |
- DtmfSenderTest() |
- : track_(AudioTrack::Create(kTestAudioLabel, NULL)), |
- observer_(new rtc::RefCountedObject<FakeDtmfObserver>()), |
- provider_(new FakeDtmfProvider()) { |
- provider_->AddCanInsertDtmfTrack(kTestAudioLabel); |
- dtmf_ = DtmfSender::Create(track_, rtc::Thread::Current(), |
- provider_.get()); |
- dtmf_->RegisterObserver(observer_.get()); |
- } |
- |
- ~DtmfSenderTest() { |
- if (dtmf_.get()) { |
- dtmf_->UnregisterObserver(); |
- } |
- } |
- |
- // Constructs a list of DtmfInfo from |tones|, |duration| and |
- // |inter_tone_gap|. |
- void GetDtmfInfoFromString(const std::string& tones, int duration, |
- int inter_tone_gap, |
- std::vector<FakeDtmfProvider::DtmfInfo>* dtmfs) { |
- // Init extra_delay as -inter_tone_gap - duration to ensure the first |
- // DtmfInfo's gap field will be 0. |
- int extra_delay = -1 * (inter_tone_gap + duration); |
- |
- std::string::const_iterator it = tones.begin(); |
- for (; it != tones.end(); ++it) { |
- char tone = *it; |
- int code = 0; |
- webrtc::GetDtmfCode(tone, &code); |
- if (tone == ',') { |
- extra_delay = 2000; // 2 seconds |
- } else { |
- dtmfs->push_back(FakeDtmfProvider::DtmfInfo(code, duration, |
- duration + inter_tone_gap + extra_delay)); |
- extra_delay = 0; |
- } |
- } |
- } |
- |
- void VerifyExpectedState(AudioTrackInterface* track, |
- const std::string& tones, |
- int duration, int inter_tone_gap) { |
- EXPECT_EQ(track, dtmf_->track()); |
- EXPECT_EQ(tones, dtmf_->tones()); |
- EXPECT_EQ(duration, dtmf_->duration()); |
- EXPECT_EQ(inter_tone_gap, dtmf_->inter_tone_gap()); |
- } |
- |
- // Verify the provider got all the expected calls. |
- void VerifyOnProvider(const std::string& tones, int duration, |
- int inter_tone_gap) { |
- std::vector<FakeDtmfProvider::DtmfInfo> dtmf_queue_ref; |
- GetDtmfInfoFromString(tones, duration, inter_tone_gap, &dtmf_queue_ref); |
- VerifyOnProvider(dtmf_queue_ref); |
- } |
- |
- void VerifyOnProvider( |
- const std::vector<FakeDtmfProvider::DtmfInfo>& dtmf_queue_ref) { |
- const std::vector<FakeDtmfProvider::DtmfInfo>& dtmf_queue = |
- provider_->dtmf_info_queue(); |
- ASSERT_EQ(dtmf_queue_ref.size(), dtmf_queue.size()); |
- std::vector<FakeDtmfProvider::DtmfInfo>::const_iterator it_ref = |
- dtmf_queue_ref.begin(); |
- std::vector<FakeDtmfProvider::DtmfInfo>::const_iterator it = |
- dtmf_queue.begin(); |
- while (it_ref != dtmf_queue_ref.end() && it != dtmf_queue.end()) { |
- EXPECT_EQ(it_ref->code, it->code); |
- EXPECT_EQ(it_ref->duration, it->duration); |
- // Allow ~10ms error (can be small since we're using a fake clock). |
- EXPECT_GE(it_ref->gap, it->gap - 10); |
- EXPECT_LE(it_ref->gap, it->gap + 10); |
- ++it_ref; |
- ++it; |
- } |
- } |
- |
- // Verify the observer got all the expected callbacks. |
- void VerifyOnObserver(const std::string& tones_ref) { |
- const std::vector<std::string>& tones = observer_->tones(); |
- // The observer will get an empty string at the end. |
- EXPECT_EQ(tones_ref.size() + 1, tones.size()); |
- EXPECT_TRUE(tones.back().empty()); |
- std::string::const_iterator it_ref = tones_ref.begin(); |
- std::vector<std::string>::const_iterator it = tones.begin(); |
- while (it_ref != tones_ref.end() && it != tones.end()) { |
- EXPECT_EQ(*it_ref, it->at(0)); |
- ++it_ref; |
- ++it; |
- } |
- } |
- |
- rtc::scoped_refptr<AudioTrackInterface> track_; |
- std::unique_ptr<FakeDtmfObserver> observer_; |
- std::unique_ptr<FakeDtmfProvider> provider_; |
- rtc::scoped_refptr<DtmfSender> dtmf_; |
- rtc::ScopedFakeClock fake_clock_; |
-}; |
- |
-TEST_F(DtmfSenderTest, CanInsertDtmf) { |
- EXPECT_TRUE(dtmf_->CanInsertDtmf()); |
- provider_->RemoveCanInsertDtmfTrack(kTestAudioLabel); |
- EXPECT_FALSE(dtmf_->CanInsertDtmf()); |
-} |
- |
-TEST_F(DtmfSenderTest, InsertDtmf) { |
- std::string tones = "@1%a&*$"; |
- int duration = 100; |
- int inter_tone_gap = 50; |
- EXPECT_TRUE(dtmf_->InsertDtmf(tones, duration, inter_tone_gap)); |
- EXPECT_TRUE_SIMULATED_WAIT(observer_->completed(), kMaxWaitMs, fake_clock_); |
- |
- // The unrecognized characters should be ignored. |
- std::string known_tones = "1a*"; |
- VerifyOnProvider(known_tones, duration, inter_tone_gap); |
- VerifyOnObserver(known_tones); |
-} |
- |
-TEST_F(DtmfSenderTest, InsertDtmfTwice) { |
- std::string tones1 = "12"; |
- std::string tones2 = "ab"; |
- int duration = 100; |
- int inter_tone_gap = 50; |
- EXPECT_TRUE(dtmf_->InsertDtmf(tones1, duration, inter_tone_gap)); |
- VerifyExpectedState(track_, tones1, duration, inter_tone_gap); |
- // Wait until the first tone got sent. |
- EXPECT_TRUE_SIMULATED_WAIT(observer_->tones().size() == 1, kMaxWaitMs, |
- fake_clock_); |
- VerifyExpectedState(track_, "2", duration, inter_tone_gap); |
- // Insert with another tone buffer. |
- EXPECT_TRUE(dtmf_->InsertDtmf(tones2, duration, inter_tone_gap)); |
- VerifyExpectedState(track_, tones2, duration, inter_tone_gap); |
- // Wait until it's completed. |
- EXPECT_TRUE_SIMULATED_WAIT(observer_->completed(), kMaxWaitMs, fake_clock_); |
- |
- std::vector<FakeDtmfProvider::DtmfInfo> dtmf_queue_ref; |
- GetDtmfInfoFromString("1", duration, inter_tone_gap, &dtmf_queue_ref); |
- GetDtmfInfoFromString("ab", duration, inter_tone_gap, &dtmf_queue_ref); |
- VerifyOnProvider(dtmf_queue_ref); |
- VerifyOnObserver("1ab"); |
-} |
- |
-TEST_F(DtmfSenderTest, InsertDtmfWhileProviderIsDeleted) { |
- std::string tones = "@1%a&*$"; |
- int duration = 100; |
- int inter_tone_gap = 50; |
- EXPECT_TRUE(dtmf_->InsertDtmf(tones, duration, inter_tone_gap)); |
- // Wait until the first tone got sent. |
- EXPECT_TRUE_SIMULATED_WAIT(observer_->tones().size() == 1, kMaxWaitMs, |
- fake_clock_); |
- // Delete provider. |
- provider_.reset(); |
- // The queue should be discontinued so no more tone callbacks. |
- SIMULATED_WAIT(false, 200, fake_clock_); |
- EXPECT_EQ(1U, observer_->tones().size()); |
-} |
- |
-TEST_F(DtmfSenderTest, InsertDtmfWhileSenderIsDeleted) { |
- std::string tones = "@1%a&*$"; |
- int duration = 100; |
- int inter_tone_gap = 50; |
- EXPECT_TRUE(dtmf_->InsertDtmf(tones, duration, inter_tone_gap)); |
- // Wait until the first tone got sent. |
- EXPECT_TRUE_SIMULATED_WAIT(observer_->tones().size() == 1, kMaxWaitMs, |
- fake_clock_); |
- // Delete the sender. |
- dtmf_ = NULL; |
- // The queue should be discontinued so no more tone callbacks. |
- SIMULATED_WAIT(false, 200, fake_clock_); |
- EXPECT_EQ(1U, observer_->tones().size()); |
-} |
- |
-TEST_F(DtmfSenderTest, InsertEmptyTonesToCancelPreviousTask) { |
- std::string tones1 = "12"; |
- std::string tones2 = ""; |
- int duration = 100; |
- int inter_tone_gap = 50; |
- EXPECT_TRUE(dtmf_->InsertDtmf(tones1, duration, inter_tone_gap)); |
- // Wait until the first tone got sent. |
- EXPECT_TRUE_SIMULATED_WAIT(observer_->tones().size() == 1, kMaxWaitMs, |
- fake_clock_); |
- // Insert with another tone buffer. |
- EXPECT_TRUE(dtmf_->InsertDtmf(tones2, duration, inter_tone_gap)); |
- // Wait until it's completed. |
- EXPECT_TRUE_SIMULATED_WAIT(observer_->completed(), kMaxWaitMs, fake_clock_); |
- |
- std::vector<FakeDtmfProvider::DtmfInfo> dtmf_queue_ref; |
- GetDtmfInfoFromString("1", duration, inter_tone_gap, &dtmf_queue_ref); |
- VerifyOnProvider(dtmf_queue_ref); |
- VerifyOnObserver("1"); |
-} |
- |
-TEST_F(DtmfSenderTest, InsertDtmfWithCommaAsDelay) { |
- std::string tones = "3,4"; |
- int duration = 100; |
- int inter_tone_gap = 50; |
- EXPECT_TRUE(dtmf_->InsertDtmf(tones, duration, inter_tone_gap)); |
- EXPECT_TRUE_SIMULATED_WAIT(observer_->completed(), kMaxWaitMs, fake_clock_); |
- |
- VerifyOnProvider(tones, duration, inter_tone_gap); |
- VerifyOnObserver(tones); |
-} |
- |
-TEST_F(DtmfSenderTest, TryInsertDtmfWhenItDoesNotWork) { |
- std::string tones = "3,4"; |
- int duration = 100; |
- int inter_tone_gap = 50; |
- provider_->RemoveCanInsertDtmfTrack(kTestAudioLabel); |
- EXPECT_FALSE(dtmf_->InsertDtmf(tones, duration, inter_tone_gap)); |
-} |
- |
-TEST_F(DtmfSenderTest, InsertDtmfWithInvalidDurationOrGap) { |
- std::string tones = "3,4"; |
- int duration = 100; |
- int inter_tone_gap = 50; |
- |
- EXPECT_FALSE(dtmf_->InsertDtmf(tones, 6001, inter_tone_gap)); |
- EXPECT_FALSE(dtmf_->InsertDtmf(tones, 69, inter_tone_gap)); |
- EXPECT_FALSE(dtmf_->InsertDtmf(tones, duration, 49)); |
- |
- EXPECT_TRUE(dtmf_->InsertDtmf(tones, duration, inter_tone_gap)); |
-} |