Index: trunk/src/chrome/browser/media/webrtc_rtp_dump_handler_unittest.cc |
=================================================================== |
--- trunk/src/chrome/browser/media/webrtc_rtp_dump_handler_unittest.cc (revision 273764) |
+++ trunk/src/chrome/browser/media/webrtc_rtp_dump_handler_unittest.cc (working copy) |
@@ -1,393 +0,0 @@ |
-// Copyright 2014 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "base/bind.h" |
-#include "base/file_util.h" |
-#include "base/files/scoped_temp_dir.h" |
-#include "base/macros.h" |
-#include "base/memory/scoped_ptr.h" |
-#include "base/run_loop.h" |
-#include "chrome/browser/media/webrtc_rtp_dump_handler.h" |
-#include "chrome/browser/media/webrtc_rtp_dump_writer.h" |
-#include "content/public/test/test_browser_thread_bundle.h" |
-#include "testing/gmock/include/gmock/gmock.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-class FakeDumpWriter : public WebRtcRtpDumpWriter { |
- public: |
- FakeDumpWriter(size_t max_dump_size, |
- const base::Closure& max_size_reached_callback, |
- bool end_dump_success) |
- : WebRtcRtpDumpWriter(base::FilePath(), |
- base::FilePath(), |
- max_dump_size, |
- base::Closure()), |
- max_dump_size_(max_dump_size), |
- current_dump_size_(0), |
- max_size_reached_callback_(max_size_reached_callback), |
- end_dump_success_(end_dump_success) {} |
- |
- virtual void WriteRtpPacket(const uint8* packet_header, |
- size_t header_length, |
- size_t packet_length, |
- bool incoming) OVERRIDE { |
- current_dump_size_ += header_length; |
- if (current_dump_size_ > max_dump_size_) |
- max_size_reached_callback_.Run(); |
- } |
- |
- virtual void EndDump(RtpDumpType type, |
- const EndDumpCallback& finished_callback) OVERRIDE { |
- bool incoming_sucess = end_dump_success_; |
- bool outgoing_success = end_dump_success_; |
- |
- if (type == RTP_DUMP_INCOMING) |
- outgoing_success = false; |
- else if (type == RTP_DUMP_OUTGOING) |
- incoming_sucess = false; |
- |
- base::MessageLoop::current()->PostTask( |
- FROM_HERE, |
- base::Bind(finished_callback, incoming_sucess, outgoing_success)); |
- } |
- |
- private: |
- size_t max_dump_size_; |
- size_t current_dump_size_; |
- base::Closure max_size_reached_callback_; |
- bool end_dump_success_; |
-}; |
- |
-class WebRtcRtpDumpHandlerTest : public testing::Test { |
- public: |
- WebRtcRtpDumpHandlerTest() |
- : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) { |
- ResetDumpHandler(base::FilePath(), true); |
- } |
- |
- void ResetDumpHandler(const base::FilePath& dir, bool end_dump_success) { |
- handler_.reset(new WebRtcRtpDumpHandler( |
- dir.empty() ? base::FilePath(FILE_PATH_LITERAL("dummy")) : dir)); |
- |
- scoped_ptr<WebRtcRtpDumpWriter> writer(new FakeDumpWriter( |
- 10, |
- base::Bind(&WebRtcRtpDumpHandler::OnMaxDumpSizeReached, |
- base::Unretained(handler_.get())), |
- end_dump_success)); |
- |
- handler_->SetDumpWriterForTesting(writer.Pass()); |
- } |
- |
- void WriteFakeDumpFiles(const base::FilePath& dir, |
- base::FilePath* incoming_dump, |
- base::FilePath* outgoing_dump) { |
- *incoming_dump = dir.AppendASCII("recv"); |
- *outgoing_dump = dir.AppendASCII("send"); |
- const char dummy[] = "dummy"; |
- EXPECT_GT(base::WriteFile(*incoming_dump, dummy, arraysize(dummy)), 0); |
- EXPECT_GT(base::WriteFile(*outgoing_dump, dummy, arraysize(dummy)), 0); |
- } |
- |
- MOCK_METHOD2(OnStopDumpFinished, |
- void(bool success, const std::string& error)); |
- |
- MOCK_METHOD0(OnStopOngoingDumpsFinished, void(void)); |
- |
- protected: |
- content::TestBrowserThreadBundle thread_bundle_; |
- scoped_ptr<WebRtcRtpDumpHandler> handler_; |
-}; |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, StateTransition) { |
- std::string error; |
- |
- RtpDumpType types[3]; |
- types[0] = RTP_DUMP_INCOMING; |
- types[1] = RTP_DUMP_OUTGOING; |
- types[2] = RTP_DUMP_BOTH; |
- |
- for (size_t i = 0; i < arraysize(types); ++i) { |
- DVLOG(2) << "Verifying state transition: type = " << types[i]; |
- |
- // Only StartDump is allowed in STATE_NONE. |
- EXPECT_CALL(*this, OnStopDumpFinished(false, testing::_)); |
- handler_->StopDump(types[i], |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- |
- WebRtcRtpDumpHandler::ReleasedDumps empty_dumps(handler_->ReleaseDumps()); |
- EXPECT_TRUE(empty_dumps.incoming_dump_path.empty()); |
- EXPECT_TRUE(empty_dumps.outgoing_dump_path.empty()); |
- EXPECT_TRUE(handler_->StartDump(types[i], &error)); |
- base::RunLoop().RunUntilIdle(); |
- |
- // Only StopDump is allowed in STATE_STARTED. |
- EXPECT_FALSE(handler_->StartDump(types[i], &error)); |
- EXPECT_FALSE(handler_->ReadyToRelease()); |
- |
- EXPECT_CALL(*this, OnStopDumpFinished(true, testing::_)); |
- handler_->StopDump(types[i], |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- base::RunLoop().RunUntilIdle(); |
- |
- // Only ReleaseDump is allowed in STATE_STOPPED. |
- EXPECT_FALSE(handler_->StartDump(types[i], &error)); |
- |
- EXPECT_CALL(*this, OnStopDumpFinished(false, testing::_)); |
- handler_->StopDump(types[i], |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- EXPECT_TRUE(handler_->ReadyToRelease()); |
- |
- WebRtcRtpDumpHandler::ReleasedDumps dumps(handler_->ReleaseDumps()); |
- if (types[i] == RTP_DUMP_INCOMING || types[i] == RTP_DUMP_BOTH) |
- EXPECT_FALSE(dumps.incoming_dump_path.empty()); |
- |
- if (types[i] == RTP_DUMP_OUTGOING || types[i] == RTP_DUMP_BOTH) |
- EXPECT_FALSE(dumps.outgoing_dump_path.empty()); |
- |
- base::RunLoop().RunUntilIdle(); |
- ResetDumpHandler(base::FilePath(), true); |
- } |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, StoppedWhenMaxSizeReached) { |
- std::string error; |
- |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_INCOMING, &error)); |
- |
- std::vector<uint8> buffer(100, 0); |
- handler_->OnRtpPacket(&buffer[0], buffer.size(), buffer.size(), true); |
- base::RunLoop().RunUntilIdle(); |
- |
- // Dumping should have been stopped, so ready to release. |
- WebRtcRtpDumpHandler::ReleasedDumps dumps = handler_->ReleaseDumps(); |
- EXPECT_FALSE(dumps.incoming_dump_path.empty()); |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, PacketIgnoredIfDumpingNotStarted) { |
- std::vector<uint8> buffer(100, 0); |
- handler_->OnRtpPacket(&buffer[0], buffer.size(), buffer.size(), true); |
- handler_->OnRtpPacket(&buffer[0], buffer.size(), buffer.size(), false); |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, PacketIgnoredIfDumpingStopped) { |
- std::string error; |
- |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_INCOMING, &error)); |
- |
- EXPECT_CALL(*this, OnStopDumpFinished(true, testing::_)); |
- handler_->StopDump(RTP_DUMP_INCOMING, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- |
- std::vector<uint8> buffer(100, 0); |
- handler_->OnRtpPacket(&buffer[0], buffer.size(), buffer.size(), true); |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, CannotStartMoreThanFiveDumps) { |
- std::string error; |
- |
- handler_.reset(); |
- |
- scoped_ptr<WebRtcRtpDumpHandler> handlers[6]; |
- |
- for (size_t i = 0; i < arraysize(handlers); ++i) { |
- handlers[i].reset(new WebRtcRtpDumpHandler(base::FilePath())); |
- |
- if (i < arraysize(handlers) - 1) { |
- EXPECT_TRUE(handlers[i]->StartDump(RTP_DUMP_INCOMING, &error)); |
- } else { |
- EXPECT_FALSE(handlers[i]->StartDump(RTP_DUMP_INCOMING, &error)); |
- } |
- } |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, StartStopIncomingThenStartStopOutgoing) { |
- std::string error; |
- |
- EXPECT_CALL(*this, OnStopDumpFinished(true, testing::_)).Times(2); |
- |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_INCOMING, &error)); |
- handler_->StopDump(RTP_DUMP_INCOMING, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_OUTGOING, &error)); |
- handler_->StopDump(RTP_DUMP_OUTGOING, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, StartIncomingStartOutgoingThenStopBoth) { |
- std::string error; |
- |
- EXPECT_CALL(*this, OnStopDumpFinished(true, testing::_)); |
- |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_INCOMING, &error)); |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_OUTGOING, &error)); |
- |
- handler_->StopDump(RTP_DUMP_INCOMING, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, StartBothThenStopIncomingStopOutgoing) { |
- std::string error; |
- |
- EXPECT_CALL(*this, OnStopDumpFinished(true, testing::_)).Times(2); |
- |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_BOTH, &error)); |
- |
- handler_->StopDump(RTP_DUMP_INCOMING, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- handler_->StopDump(RTP_DUMP_OUTGOING, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, DumpsCleanedUpIfNotReleased) { |
- base::ScopedTempDir temp_dir; |
- ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
- ResetDumpHandler(temp_dir.path(), true); |
- |
- base::FilePath incoming_dump, outgoing_dump; |
- WriteFakeDumpFiles(temp_dir.path(), &incoming_dump, &outgoing_dump); |
- |
- std::string error; |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_BOTH, &error)); |
- |
- EXPECT_CALL(*this, OnStopDumpFinished(true, testing::_)); |
- handler_->StopDump(RTP_DUMP_BOTH, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- base::RunLoop().RunUntilIdle(); |
- |
- handler_.reset(); |
- base::RunLoop().RunUntilIdle(); |
- |
- EXPECT_FALSE(base::PathExists(incoming_dump)); |
- EXPECT_FALSE(base::PathExists(outgoing_dump)); |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, DumpDeletedIfEndDumpFailed) { |
- base::ScopedTempDir temp_dir; |
- ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
- |
- // Make the writer return failure on EndStream. |
- ResetDumpHandler(temp_dir.path(), false); |
- |
- base::FilePath incoming_dump, outgoing_dump; |
- WriteFakeDumpFiles(temp_dir.path(), &incoming_dump, &outgoing_dump); |
- |
- std::string error; |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_BOTH, &error)); |
- EXPECT_CALL(*this, OnStopDumpFinished(true, testing::_)).Times(2); |
- |
- handler_->StopDump(RTP_DUMP_INCOMING, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- base::RunLoop().RunUntilIdle(); |
- |
- EXPECT_FALSE(base::PathExists(incoming_dump)); |
- EXPECT_TRUE(base::PathExists(outgoing_dump)); |
- |
- handler_->StopDump(RTP_DUMP_OUTGOING, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- base::RunLoop().RunUntilIdle(); |
- EXPECT_FALSE(base::PathExists(outgoing_dump)); |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, StopOngoingDumpsWhileStoppingDumps) { |
- std::string error; |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_BOTH, &error)); |
- |
- testing::InSequence s; |
- EXPECT_CALL(*this, OnStopDumpFinished(true, testing::_)); |
- EXPECT_CALL(*this, OnStopOngoingDumpsFinished()); |
- |
- handler_->StopDump(RTP_DUMP_BOTH, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- |
- handler_->StopOngoingDumps( |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopOngoingDumpsFinished, |
- base::Unretained(this))); |
- |
- base::RunLoop().RunUntilIdle(); |
- |
- WebRtcRtpDumpHandler::ReleasedDumps dumps(handler_->ReleaseDumps()); |
- EXPECT_FALSE(dumps.incoming_dump_path.empty()); |
- EXPECT_FALSE(dumps.outgoing_dump_path.empty()); |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, StopOngoingDumpsWhileDumping) { |
- std::string error; |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_BOTH, &error)); |
- |
- EXPECT_CALL(*this, OnStopOngoingDumpsFinished()); |
- |
- handler_->StopOngoingDumps( |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopOngoingDumpsFinished, |
- base::Unretained(this))); |
- |
- base::RunLoop().RunUntilIdle(); |
- |
- WebRtcRtpDumpHandler::ReleasedDumps dumps(handler_->ReleaseDumps()); |
- EXPECT_FALSE(dumps.incoming_dump_path.empty()); |
- EXPECT_FALSE(dumps.outgoing_dump_path.empty()); |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, StopOngoingDumpsWhenAlreadyStopped) { |
- std::string error; |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_BOTH, &error)); |
- |
- { |
- EXPECT_CALL(*this, OnStopDumpFinished(true, testing::_)); |
- |
- handler_->StopDump(RTP_DUMP_BOTH, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- base::RunLoop().RunUntilIdle(); |
- } |
- |
- EXPECT_CALL(*this, OnStopOngoingDumpsFinished()); |
- handler_->StopOngoingDumps( |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopOngoingDumpsFinished, |
- base::Unretained(this))); |
-} |
- |
-TEST_F(WebRtcRtpDumpHandlerTest, StopOngoingDumpsWhileStoppingOneDump) { |
- std::string error; |
- EXPECT_TRUE(handler_->StartDump(RTP_DUMP_BOTH, &error)); |
- |
- testing::InSequence s; |
- EXPECT_CALL(*this, OnStopDumpFinished(true, testing::_)); |
- EXPECT_CALL(*this, OnStopOngoingDumpsFinished()); |
- |
- handler_->StopDump(RTP_DUMP_INCOMING, |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopDumpFinished, |
- base::Unretained(this))); |
- |
- handler_->StopOngoingDumps( |
- base::Bind(&WebRtcRtpDumpHandlerTest::OnStopOngoingDumpsFinished, |
- base::Unretained(this))); |
- |
- base::RunLoop().RunUntilIdle(); |
- |
- WebRtcRtpDumpHandler::ReleasedDumps dumps(handler_->ReleaseDumps()); |
- EXPECT_FALSE(dumps.incoming_dump_path.empty()); |
- EXPECT_FALSE(dumps.outgoing_dump_path.empty()); |
-} |