OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include <stdint.h> | |
6 | |
7 #include <memory> | |
8 #include <string> | |
9 #include <vector> | |
10 | |
11 #include "base/bind.h" | |
12 #include "base/macros.h" | |
13 #include "base/memory/ptr_util.h" | |
14 #include "base/message_loop/message_loop.h" | |
15 #include "base/test/simple_test_tick_clock.h" | |
16 #include "base/timer/mock_timer.h" | |
17 #include "components/audio_modem/public/modem.h" | |
18 #include "components/audio_modem/test/random_samples.h" | |
19 #include "components/audio_modem/test/stub_modem.h" | |
20 #include "components/copresence/handlers/audio/audio_directive_handler_impl.h" | |
21 #include "components/copresence/handlers/audio/tick_clock_ref_counted.h" | |
22 #include "components/copresence/proto/data.pb.h" | |
23 #include "testing/gtest/include/gtest/gtest.h" | |
24 | |
25 using audio_modem::AUDIBLE; | |
26 using audio_modem::AudioType; | |
27 using audio_modem::INAUDIBLE; | |
28 using audio_modem::StubModem; | |
29 | |
30 namespace copresence { | |
31 | |
32 namespace { | |
33 | |
34 const Directive CreateDirective(TokenInstructionType type, | |
35 bool audible, | |
36 int64_t ttl) { | |
37 Directive directive; | |
38 directive.mutable_token_instruction()->set_token_instruction_type(type); | |
39 directive.mutable_token_instruction()->set_token_id("token"); | |
40 directive.mutable_token_instruction()->set_medium(audible ? | |
41 AUDIO_AUDIBLE_DTMF : AUDIO_ULTRASOUND_PASSBAND); | |
42 directive.set_ttl_millis(ttl); | |
43 return directive; | |
44 } | |
45 | |
46 } // namespace | |
47 | |
48 class AudioDirectiveHandlerTest : public testing::Test { | |
49 public: | |
50 AudioDirectiveHandlerTest() { | |
51 modem_ptr_ = new StubModem; | |
52 timer_ptr_ = new base::MockTimer(false, false); | |
53 clock_ptr_ = new base::SimpleTestTickClock; | |
54 | |
55 directive_handler_.reset(new AudioDirectiveHandlerImpl( | |
56 base::Bind(&AudioDirectiveHandlerTest::GetDirectiveUpdates, | |
57 base::Unretained(this)), | |
58 base::WrapUnique<audio_modem::Modem>(modem_ptr_), | |
59 base::WrapUnique<base::Timer>(timer_ptr_), | |
60 make_scoped_refptr(new TickClockRefCounted(clock_ptr_)))); | |
61 directive_handler_->Initialize(nullptr, audio_modem::TokensCallback()); | |
62 } | |
63 ~AudioDirectiveHandlerTest() override {} | |
64 | |
65 protected: | |
66 const std::vector<Directive>& current_directives() { | |
67 return current_directives_; | |
68 } | |
69 | |
70 bool IsPlaying(AudioType type) { return modem_ptr_->IsPlaying(type); } | |
71 | |
72 bool IsRecording(AudioType type) { return modem_ptr_->IsRecording(type); } | |
73 | |
74 // This order is important. We want the message loop to get created before | |
75 // our the audio directive handler since the directive list ctor (invoked | |
76 // from the directive handler ctor) will post tasks. | |
77 base::MessageLoop message_loop_; | |
78 std::unique_ptr<AudioDirectiveHandler> directive_handler_; | |
79 | |
80 std::vector<Directive> current_directives_; | |
81 | |
82 // Unowned. | |
83 StubModem* modem_ptr_; | |
84 base::MockTimer* timer_ptr_; | |
85 base::SimpleTestTickClock* clock_ptr_; | |
86 | |
87 private: | |
88 void GetDirectiveUpdates(const std::vector<Directive>& current_directives) { | |
89 current_directives_ = current_directives; | |
90 } | |
91 | |
92 DISALLOW_COPY_AND_ASSIGN(AudioDirectiveHandlerTest); | |
93 }; | |
94 | |
95 TEST_F(AudioDirectiveHandlerTest, Basic) { | |
96 const int64_t kTtl = 10; | |
97 directive_handler_->AddInstruction(CreateDirective(TRANSMIT, true, kTtl), | |
98 "op_id1"); | |
99 directive_handler_->AddInstruction(CreateDirective(TRANSMIT, false, kTtl), | |
100 "op_id1"); | |
101 directive_handler_->AddInstruction(CreateDirective(TRANSMIT, false, kTtl), | |
102 "op_id2"); | |
103 directive_handler_->AddInstruction(CreateDirective(RECEIVE, false, kTtl), | |
104 "op_id1"); | |
105 directive_handler_->AddInstruction(CreateDirective(RECEIVE, true, kTtl), | |
106 "op_id2"); | |
107 directive_handler_->AddInstruction(CreateDirective(RECEIVE, false, kTtl), | |
108 "op_id3"); | |
109 | |
110 EXPECT_TRUE(IsPlaying(AUDIBLE)); | |
111 EXPECT_TRUE(IsPlaying(INAUDIBLE)); | |
112 EXPECT_TRUE(IsRecording(AUDIBLE)); | |
113 EXPECT_TRUE(IsRecording(INAUDIBLE)); | |
114 | |
115 directive_handler_->RemoveInstructions("op_id1"); | |
116 EXPECT_FALSE(IsPlaying(AUDIBLE)); | |
117 EXPECT_TRUE(IsPlaying(INAUDIBLE)); | |
118 EXPECT_TRUE(IsRecording(AUDIBLE)); | |
119 EXPECT_TRUE(IsRecording(INAUDIBLE)); | |
120 | |
121 directive_handler_->RemoveInstructions("op_id2"); | |
122 EXPECT_FALSE(IsPlaying(INAUDIBLE)); | |
123 EXPECT_FALSE(IsRecording(AUDIBLE)); | |
124 EXPECT_TRUE(IsRecording(INAUDIBLE)); | |
125 | |
126 directive_handler_->RemoveInstructions("op_id3"); | |
127 EXPECT_FALSE(IsRecording(INAUDIBLE)); | |
128 } | |
129 | |
130 TEST_F(AudioDirectiveHandlerTest, Timed) { | |
131 directive_handler_->AddInstruction(CreateDirective(TRANSMIT, true, 6), | |
132 "op_id1"); | |
133 directive_handler_->AddInstruction(CreateDirective(TRANSMIT, false, 8), | |
134 "op_id1"); | |
135 directive_handler_->AddInstruction(CreateDirective(RECEIVE, false, 4), | |
136 "op_id3"); | |
137 | |
138 EXPECT_TRUE(IsPlaying(AUDIBLE)); | |
139 EXPECT_TRUE(IsPlaying(INAUDIBLE)); | |
140 EXPECT_FALSE(IsRecording(AUDIBLE)); | |
141 EXPECT_TRUE(IsRecording(INAUDIBLE)); | |
142 | |
143 // Every time we advance and a directive expires, the timer should fire also. | |
144 clock_ptr_->Advance(base::TimeDelta::FromMilliseconds(5)); | |
145 timer_ptr_->Fire(); | |
146 | |
147 // We are now at +5ms. This instruction expires at +10ms. | |
148 directive_handler_->AddInstruction(CreateDirective(RECEIVE, true, 5), | |
149 "op_id4"); | |
150 EXPECT_TRUE(IsPlaying(AUDIBLE)); | |
151 EXPECT_TRUE(IsPlaying(INAUDIBLE)); | |
152 EXPECT_TRUE(IsRecording(AUDIBLE)); | |
153 EXPECT_FALSE(IsRecording(INAUDIBLE)); | |
154 | |
155 // Advance to +7ms. | |
156 const base::TimeDelta twoMs = base::TimeDelta::FromMilliseconds(2); | |
157 clock_ptr_->Advance(twoMs); | |
158 timer_ptr_->Fire(); | |
159 | |
160 EXPECT_FALSE(IsPlaying(AUDIBLE)); | |
161 EXPECT_TRUE(IsPlaying(INAUDIBLE)); | |
162 EXPECT_TRUE(IsRecording(AUDIBLE)); | |
163 | |
164 // Advance to +9ms. | |
165 clock_ptr_->Advance(twoMs); | |
166 timer_ptr_->Fire(); | |
167 EXPECT_FALSE(IsPlaying(INAUDIBLE)); | |
168 EXPECT_TRUE(IsRecording(AUDIBLE)); | |
169 | |
170 // Advance to +11ms. | |
171 clock_ptr_->Advance(twoMs); | |
172 timer_ptr_->Fire(); | |
173 EXPECT_FALSE(IsRecording(AUDIBLE)); | |
174 } | |
175 | |
176 // TODO(rkc): Write more tests that check more convoluted sequences of | |
177 // transmits/receives. | |
178 | |
179 } // namespace copresence | |
OLD | NEW |