Index: webrtc/modules/audio_processing/test/conversational_speech/generator_unittest.cc |
diff --git a/webrtc/modules/audio_processing/test/conversational_speech/generator_unittest.cc b/webrtc/modules/audio_processing/test/conversational_speech/generator_unittest.cc |
index 221ad7c7066f3e568ea788f76947a2dfe42d4bc3..59454d9d47d26c864b5b1350c0c0ddde2b5f7d58 100644 |
--- a/webrtc/modules/audio_processing/test/conversational_speech/generator_unittest.cc |
+++ b/webrtc/modules/audio_processing/test/conversational_speech/generator_unittest.cc |
@@ -8,11 +8,14 @@ |
* be found in the AUTHORS file in the root of the source tree. |
*/ |
-#include <cstdio> |
+#include <stdio.h> |
#include <memory> |
#include "webrtc/modules/audio_processing/test/conversational_speech/config.h" |
+#include "webrtc/modules/audio_processing/test/conversational_speech/mock_wavreader_factory.h" |
+#include "webrtc/modules/audio_processing/test/conversational_speech/multiend_call.h" |
#include "webrtc/modules/audio_processing/test/conversational_speech/timing.h" |
+#include "webrtc/test/gmock.h" |
#include "webrtc/test/gtest.h" |
#include "webrtc/test/testsupport/fileutils.h" |
@@ -22,7 +25,10 @@ namespace { |
using conversational_speech::LoadTiming; |
using conversational_speech::SaveTiming; |
+using conversational_speech::MockWavReaderFactory; |
+using conversational_speech::MultiEndCall; |
using conversational_speech::Turn; |
+using conversational_speech::WavReaderAbstractFactory; |
const char* const audiotracks_path = "/path/to/audiotracks"; |
const char* const timing_filepath = "/path/to/timing_file.txt"; |
@@ -34,7 +40,7 @@ const std::vector<Turn> expected_timing = { |
{"A", "a2", 100}, |
{"B", "b2", -200}, |
{"A", "a3", 0}, |
- {"A", "a4", 0}, |
+ {"A", "a3", 0}, |
}; |
const std::size_t kNumberOfTurns = expected_timing.size(); |
@@ -50,11 +56,6 @@ TEST(ConversationalSpeechTest, Settings) { |
EXPECT_EQ(output_path, config.output_path()); |
} |
-TEST(ConversationalSpeechTest, ExpectedTimingSize) { |
- // Check the expected timing size. |
- EXPECT_EQ(kNumberOfTurns, 6u); |
-} |
- |
TEST(ConversationalSpeechTest, TimingSaveLoad) { |
// Save test timing. |
const std::string temporary_filepath = webrtc::test::TempFilename( |
@@ -75,5 +76,21 @@ TEST(ConversationalSpeechTest, TimingSaveLoad) { |
} |
} |
+TEST(ConversationalSpeechTest, MultiEndCallCreate) { |
+ auto mock_wavreader_factory = std::unique_ptr<MockWavReaderFactory>( |
+ new MockWavReaderFactory()); |
+ |
+ // There are 5 unique audio tracks to read. |
+ EXPECT_CALL(*mock_wavreader_factory, Create(testing::_)).Times(5); |
+ |
+ // Inject the mock wav reader factory. |
+ conversational_speech::MultiEndCall multiend_call( |
+ expected_timing, audiotracks_path, std::move(mock_wavreader_factory)); |
+ |
+ // Test. |
+ EXPECT_EQ(2u, multiend_call.speaker_names().size()); |
+ EXPECT_EQ(5u, multiend_call.audiotrack_readers().size()); |
+} |
+ |
} // namespace test |
} // namespace webrtc |