1f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)// Copyright 2014 The Chromium Authors. All rights reserved. 2f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 3f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)// found in the LICENSE file. 4f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 5f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "remoting/host/linux/audio_pipe_reader.h" 6f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 7f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include <sys/types.h> 8f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include <sys/stat.h> 9f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include <unistd.h> 10f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 11f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "base/files/file.h" 12f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "base/files/scoped_temp_dir.h" 13f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "base/message_loop/message_loop.h" 14f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "base/run_loop.h" 15f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "base/threading/thread.h" 16f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "base/time/time.h" 17f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 18f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 19f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)namespace remoting { 20f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 21f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)class AudioPipeReaderTest : public testing::Test, 22f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) public AudioPipeReader::StreamObserver { 23f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) public: 24f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) AudioPipeReaderTest() 25f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) : stop_at_position_(-1) { 26f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 27f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 28f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) virtual void SetUp() OVERRIDE { 29f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_TRUE(test_dir_.CreateUniqueTempDir()); 30f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) pipe_path_ = test_dir_.path().AppendASCII("test_pipe"); 31f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) audio_thread_.reset(new base::Thread("TestAudioThread")); 32f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) audio_thread_->StartWithOptions( 33f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) base::Thread::Options(base::MessageLoop::TYPE_IO, 0)); 34f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) reader_ = AudioPipeReader::Create(audio_thread_->message_loop_proxy(), 35f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) pipe_path_); 36f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) reader_->AddObserver(this); 37f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 38f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 39f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) // AudioPipeReader::StreamObserver interface. 40f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) virtual void OnDataRead(scoped_refptr<base::RefCountedString> data) OVERRIDE { 41f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) read_data_ += data->data(); 42f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (stop_at_position_ > 0 && 43f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) static_cast<int>(read_data_.size()) >= stop_at_position_) { 44f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) stop_at_position_ = -1; 45f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) run_loop_->Quit(); 46f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 47f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 48f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 49f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) void CreatePipe() { 50f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_EQ(0, mkfifo(pipe_path_.value().c_str(), 0600)); 51f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) output_.reset(new base::File( 52f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) pipe_path_, base::File::FLAG_OPEN | base::File::FLAG_WRITE)); 53f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_TRUE(output_->IsValid()); 54f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 55f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 56f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) void DeletePipe() { 57f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) output_.reset(); 58f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_EQ(0, unlink(pipe_path_.value().c_str())); 59f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 60f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 61f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) void WaitForInput(int num_bytes) { 62f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) run_loop_.reset(new base::RunLoop()); 63f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) stop_at_position_ = read_data_.size() + num_bytes; 64f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) run_loop_->Run(); 65f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 66f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 67f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) void WriteAndWait(const std::string& data) { 68f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_EQ(static_cast<int>(data.size()), 69f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) output_->WriteAtCurrentPos(data.data(), data.size())); 70f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) WaitForInput(data.size()); 71f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 72f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 73f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) protected: 74f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) base::MessageLoop message_loop_; 75f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) scoped_ptr<base::RunLoop> run_loop_; 76f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) scoped_ptr<base::Thread> audio_thread_; 77f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) base::ScopedTempDir test_dir_; 78f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) base::FilePath pipe_path_; 79f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) scoped_ptr<base::File> output_; 80f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 81f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) scoped_refptr<AudioPipeReader> reader_; 82f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 83f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) std::string read_data_; 84f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) int stop_at_position_; 85f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 86f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(AudioPipeReaderTest); 87f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)}; 88f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 89f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)// Verify that the reader can detect when the pipe is created and destroyed. 90f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)TEST_F(AudioPipeReaderTest, CreateAndDestroyPipe) { 91f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreatePipe()); 92f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_NO_FATAL_FAILURE(WriteAndWait("ABCD")); 93f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_NO_FATAL_FAILURE(DeletePipe()); 94f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 95f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreatePipe()); 96f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_NO_FATAL_FAILURE(WriteAndWait("abcd")); 97f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_NO_FATAL_FAILURE(DeletePipe()); 98f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 99f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) EXPECT_EQ("ABCDabcd", read_data_); 100f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 101f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 102f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)// Verifies that the reader reads at the right speed. 103f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)TEST_F(AudioPipeReaderTest, Pacing) { 104f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) int test_data_size = AudioPipeReader::kSamplingRate * 105f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) AudioPipeReader::kChannels * 106f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) AudioPipeReader::kBytesPerSample / 2; 107f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) std::string test_data(test_data_size, '\0'); 108f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 109f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreatePipe()); 110f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 111f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) base::TimeTicks start_time = base::TimeTicks::Now(); 112f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ASSERT_NO_FATAL_FAILURE(WriteAndWait(test_data)); 113f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) base::TimeDelta time_passed = base::TimeTicks::Now() - start_time; 114f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 115f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) EXPECT_EQ(test_data, read_data_); 116f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) EXPECT_GE(time_passed, base::TimeDelta::FromMilliseconds(500)); 117f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 118f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 119f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} // namespace remoting 120