1 /* 2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. 3 * 4 * Use of this source code is governed by a BSD-style license 5 * that can be found in the LICENSE file in the root of the source 6 * tree. An additional intellectual property rights grant can be found 7 * in the file PATENTS. All contributing project authors may 8 * be found in the AUTHORS file in the root of the source tree. 9 */ 10 11 // Unit tests for FilePlayer. 12 13 #include "webrtc/modules/utility/interface/file_player.h" 14 15 #include <stdio.h> 16 #include <string> 17 18 #include "gflags/gflags.h" 19 #include "testing/gtest/include/gtest/gtest.h" 20 #include "webrtc/base/md5digest.h" 21 #include "webrtc/base/stringencode.h" 22 #include "webrtc/test/testsupport/fileutils.h" 23 24 DEFINE_bool(file_player_output, false, "Generate reference files."); 25 26 namespace webrtc { 27 28 class FilePlayerTest : public ::testing::Test { 29 protected: 30 static const uint32_t kId = 0; 31 static const FileFormats kFileFormat = kFileFormatWavFile; 32 static const int kSampleRateHz = 8000; 33 34 FilePlayerTest() 35 : player_(FilePlayer::CreateFilePlayer(kId, kFileFormat)), 36 output_file_(NULL) {} 37 38 virtual void SetUp() OVERRIDE { 39 if (FLAGS_file_player_output) { 40 std::string output_file = 41 webrtc::test::OutputPath() + "file_player_unittest_out.pcm"; 42 output_file_ = fopen(output_file.c_str(), "wb"); 43 ASSERT_TRUE(output_file_ != NULL); 44 } 45 } 46 47 virtual void TearDown() OVERRIDE { 48 if (output_file_) 49 fclose(output_file_); 50 } 51 52 ~FilePlayerTest() { FilePlayer::DestroyFilePlayer(player_); } 53 54 void PlayFileAndCheck(const std::string& input_file, 55 const std::string& ref_checksum, 56 int output_length_ms) { 57 const float kScaling = 1; 58 ASSERT_EQ(0, 59 player_->StartPlayingFile( 60 input_file.c_str(), false, 0, kScaling, 0, 0, NULL)); 61 rtc::Md5Digest checksum; 62 for (int i = 0; i < output_length_ms / 10; ++i) { 63 int16_t out[10 * kSampleRateHz / 1000] = {0}; 64 int num_samples; 65 EXPECT_EQ(0, 66 player_->Get10msAudioFromFile(out, num_samples, kSampleRateHz)); 67 checksum.Update(out, num_samples * sizeof(out[0])); 68 if (FLAGS_file_player_output) { 69 ASSERT_EQ(static_cast<size_t>(num_samples), 70 fwrite(out, sizeof(out[0]), num_samples, output_file_)); 71 } 72 } 73 char checksum_result[rtc::Md5Digest::kSize]; 74 EXPECT_EQ(rtc::Md5Digest::kSize, 75 checksum.Finish(checksum_result, rtc::Md5Digest::kSize)); 76 EXPECT_EQ(ref_checksum, 77 rtc::hex_encode(checksum_result, sizeof(checksum_result))); 78 } 79 80 FilePlayer* player_; 81 FILE* output_file_; 82 }; 83 84 TEST_F(FilePlayerTest, PlayWavPcmuFile) { 85 const std::string kFileName = 86 test::ResourcePath("utility/encapsulated_pcmu_8khz", "wav"); 87 // The file is longer than this, but keeping the output shorter limits the 88 // runtime for the test. 89 const int kOutputLengthMs = 10000; 90 const std::string kRefChecksum = "c74e7fd432d439b1311e1c16815b3e9a"; 91 92 PlayFileAndCheck(kFileName, kRefChecksum, kOutputLengthMs); 93 } 94 95 TEST_F(FilePlayerTest, PlayWavPcm16File) { 96 const std::string kFileName = 97 test::ResourcePath("utility/encapsulated_pcm16b_8khz", "wav"); 98 // The file is longer than this, but keeping the output shorter limits the 99 // runtime for the test. 100 const int kOutputLengthMs = 10000; 101 const std::string kRefChecksum = "e41d7e1dac8aeae9f21e8e03cd7ecd71"; 102 103 PlayFileAndCheck(kFileName, kRefChecksum, kOutputLengthMs); 104 } 105 106 } // namespace webrtc 107
