Index: webrtc/tools/frame_analyzer/video_quality_analysis_unittest.cc |
diff --git a/webrtc/tools/frame_analyzer/video_quality_analysis_unittest.cc b/webrtc/tools/frame_analyzer/video_quality_analysis_unittest.cc |
index 85d0b46e79f72879db7fa0f7053888d7fe19fc2a..10675f32d82bb8a87f0abd218d325af7c2f93267 100644 |
--- a/webrtc/tools/frame_analyzer/video_quality_analysis_unittest.cc |
+++ b/webrtc/tools/frame_analyzer/video_quality_analysis_unittest.cc |
@@ -132,6 +132,154 @@ TEST_F(VideoQualityAnalysisTest, PrintMaxRepeatedAndSkippedFramesNormalFile) { |
stats_filename_ref, stats_filename); |
} |
+namespace { |
+void VerifyLogOutput(const std::string& log_filename, |
+ const std::vector<std::string>& expected_out) { |
+ std::ifstream logf(log_filename); |
+ std::string line; |
+ |
+ std::size_t i; |
+ for (i = 0; i < expected_out.size() && getline(logf, line); ++i) { |
+ ASSERT_EQ(expected_out.at(i), line); |
+ } |
+ ASSERT_TRUE(i == expected_out.size()) << "Not enough input data"; |
+} |
+} // unnamed namespace |
+ |
+TEST_F(VideoQualityAnalysisTest, |
+ PrintMaxRepeatedAndSkippedFramesSkippedFrames) { |
+ std::string stats_filename_ref = OutputPath() + "stats-1.txt"; |
kjellander_webrtc
2017/02/08 10:00:29
Please utilize a temporary directory or unique fil
mandermo
2017/02/13 17:04:11
Done.
mandermo
2017/02/15 11:52:15
I am using the unittest library to get unique file
|
+ std::string stats_filename = OutputPath() + "stats-2.txt"; |
+ std::ofstream stats_file; |
+ |
+ std::string log_filename = webrtc::test::OutputPath() + "log.log"; |
+ FILE* logfile = fopen(log_filename.c_str(), "w"); |
+ ASSERT_TRUE(logfile_ != NULL); |
+ stats_file.open(stats_filename_ref.c_str()); |
+ stats_file << "frame_0001 0100\n"; |
+ stats_file << "frame_0002 0101\n"; |
+ stats_file << "frame_0002 0101\n"; |
+ stats_file << "frame_0003 0103\n"; |
+ stats_file << "frame_0004 0103\n"; |
+ stats_file << "frame_0005 0106\n"; |
+ stats_file << "frame_0006 0106\n"; |
+ stats_file << "frame_0007 0108\n"; |
+ stats_file << "frame_0008 0110\n"; |
+ stats_file << "frame_0009 0112\n"; |
+ stats_file.close(); |
+ |
+ stats_file.open(stats_filename.c_str()); |
+ stats_file << "frame_0001 0101\n"; |
+ stats_file << "frame_0002 0101\n"; |
+ stats_file << "frame_0003 0101\n"; |
+ stats_file << "frame_0004 0108\n"; |
+ stats_file << "frame_0005 0108\n"; |
+ stats_file << "frame_0006 0112\n"; |
+ stats_file.close(); |
+ |
+ PrintMaxRepeatedAndSkippedFrames(logfile, "NormalStatsFile", |
+ stats_filename_ref, stats_filename); |
+ ASSERT_EQ(0, fclose(logfile)); |
+ |
+ std::vector<std::string> expected_out = { |
+ "RESULT Max_repeated: NormalStatsFile= 2", |
+ "RESULT Max_skipped: NormalStatsFile= 2", |
+ "RESULT Total_skipped: NormalStatsFile= 3", |
+ "RESULT Decode_errors_reference: NormalStatsFile= 0", |
+ "RESULT Decode_errors_test: NormalStatsFile= 0", |
+ "RESULT No_matched: []"}; |
+ VerifyLogOutput(log_filename, expected_out); |
+} |
+ |
+TEST_F(VideoQualityAnalysisTest, |
+ PrintMaxRepeatedAndSkippedFramesDecodeErrorInRef) { |
+ std::string stats_filename_ref = OutputPath() + "stats-1.txt"; |
+ std::string stats_filename = OutputPath() + "stats-2.txt"; |
+ std::ofstream stats_file; |
+ |
+ std::string log_filename = webrtc::test::OutputPath() + "log.log"; |
+ FILE* logfile = fopen(log_filename.c_str(), "w"); |
+ ASSERT_TRUE(logfile_ != NULL); |
+ stats_file.open(stats_filename_ref.c_str()); |
+ stats_file << "frame_0001 0101\n"; |
+ stats_file << "frame_0002 0103\n"; |
+ stats_file << "frame_0002 Barcode error\n"; |
+ stats_file << "frame_0003 Barcode error\n"; |
+ stats_file << "frame_0004 0106\n"; |
+ stats_file << "frame_0005 0106\n"; |
+ stats_file << "frame_0006 0107\n"; |
+ stats_file << "frame_0007 0107\n"; |
+ stats_file << "frame_0008 0110\n"; |
+ stats_file << "frame_0009 0112\n"; |
+ stats_file.close(); |
+ |
+ stats_file.open(stats_filename.c_str()); |
+ stats_file << "frame_0001 0101\n"; |
+ stats_file << "frame_0002 0103\n"; |
+ stats_file << "frame_0003 0104\n"; |
+ stats_file << "frame_0004 0105\n"; |
+ stats_file << "frame_0005 0106\n"; |
+ stats_file << "frame_0006 0107\n"; |
+ stats_file.close(); |
+ |
+ PrintMaxRepeatedAndSkippedFrames(logfile, "NormalStatsFile", |
+ stats_filename_ref, stats_filename); |
+ ASSERT_EQ(0, fclose(logfile)); |
+ |
+ std::vector<std::string> expected_out = { |
+ "RESULT Max_repeated: NormalStatsFile= 1", |
+ "RESULT Max_skipped: NormalStatsFile= 0", |
+ "RESULT Total_skipped: NormalStatsFile= 0", |
+ "RESULT Decode_errors_reference: NormalStatsFile= 2", |
+ "RESULT Decode_errors_test: NormalStatsFile= 0", |
+ "RESULT No_matched: [104, 105]"}; |
+ VerifyLogOutput(log_filename, expected_out); |
+} |
+ |
+TEST_F(VideoQualityAnalysisTest, |
+ PrintMaxRepeatedAndSkippedFramesDecodeErrorInTest) { |
+ std::string stats_filename_ref = OutputPath() + "stats-1.txt"; |
+ std::string stats_filename = OutputPath() + "stats-2.txt"; |
+ std::ofstream stats_file; |
+ |
+ std::string log_filename = webrtc::test::OutputPath() + "log.log"; |
+ FILE* logfile = fopen(log_filename.c_str(), "w"); |
+ ASSERT_TRUE(logfile_ != NULL); |
+ stats_file.open(stats_filename_ref.c_str()); |
+ stats_file << "frame_0001 0100\n"; |
+ stats_file << "frame_0002 0100\n"; |
+ stats_file << "frame_0002 0101\n"; |
+ stats_file << "frame_0003 0103\n"; |
+ stats_file << "frame_0004 0103\n"; |
+ stats_file << "frame_0005 0106\n"; |
+ stats_file << "frame_0006 0107\n"; |
+ stats_file << "frame_0007 0107\n"; |
+ stats_file << "frame_0008 0110\n"; |
+ stats_file << "frame_0009 0112\n"; |
+ stats_file.close(); |
+ |
+ stats_file.open(stats_filename.c_str()); |
+ stats_file << "frame_0001 0101\n"; |
+ stats_file << "frame_0002 Barcode error\n"; |
+ stats_file << "frame_0003 Barcode error\n"; |
+ stats_file << "frame_0004 Barcode error\n"; |
+ stats_file << "frame_0005 0107\n"; |
+ stats_file << "frame_0006 0110\n"; |
+ stats_file.close(); |
+ |
+ PrintMaxRepeatedAndSkippedFrames(logfile, "NormalStatsFile", |
+ stats_filename_ref, stats_filename); |
+ ASSERT_EQ(0, fclose(logfile)); |
+ |
+ std::vector<std::string> expected_out = { |
+ "RESULT Max_repeated: NormalStatsFile= 1", |
+ "RESULT Max_skipped: NormalStatsFile= 0", |
+ "RESULT Total_skipped: NormalStatsFile= 0", |
+ "RESULT Decode_errors_reference: NormalStatsFile= 0", |
+ "RESULT Decode_errors_test: NormalStatsFile= 3", |
+ "RESULT No_matched: []"}; |
+ VerifyLogOutput(log_filename, expected_out); |
+} |
} // namespace test |
} // namespace webrtc |