OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <string> | 5 #include <string> |
6 #include <vector> | 6 #include <vector> |
7 | 7 |
8 #include "base/bind.h" | 8 #include "base/bind.h" |
9 #include "base/compiler_specific.h" | 9 #include "base/compiler_specific.h" |
10 #include "base/location.h" | 10 #include "base/location.h" |
(...skipping 28 matching lines...) Expand all Loading... |
39 explicit Listener(const std::vector<std::string>& messages) | 39 explicit Listener(const std::vector<std::string>& messages) |
40 : messages_(messages) {} | 40 : messages_(messages) {} |
41 | 41 |
42 ~Listener() override { EXPECT_TRUE(messages_.empty()); } | 42 ~Listener() override { EXPECT_TRUE(messages_.empty()); } |
43 | 43 |
44 void OnMessageReceived(const std::string& message) override { | 44 void OnMessageReceived(const std::string& message) override { |
45 ASSERT_TRUE(messages_.size()); | 45 ASSERT_TRUE(messages_.size()); |
46 EXPECT_EQ(messages_[0], message); | 46 EXPECT_EQ(messages_[0], message); |
47 messages_.erase(messages_.begin()); | 47 messages_.erase(messages_.begin()); |
48 if (messages_.empty()) | 48 if (messages_.empty()) |
49 base::MessageLoop::current()->Quit(); | 49 base::MessageLoop::current()->QuitWhenIdle(); |
50 } | 50 } |
51 | 51 |
52 void OnClose() override { EXPECT_TRUE(false); } | 52 void OnClose() override { EXPECT_TRUE(false); } |
53 | 53 |
54 private: | 54 private: |
55 std::vector<std::string> messages_; | 55 std::vector<std::string> messages_; |
56 }; | 56 }; |
57 | 57 |
58 class CloseListener : public WebSocketListener { | 58 class CloseListener : public WebSocketListener { |
59 public: | 59 public: |
(...skipping 131 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
191 SendReceive(messages); | 191 SendReceive(messages); |
192 } | 192 } |
193 | 193 |
194 TEST_F(WebSocketTest, SendReceiveMultiple) { | 194 TEST_F(WebSocketTest, SendReceiveMultiple) { |
195 std::vector<std::string> messages; | 195 std::vector<std::string> messages; |
196 messages.push_back("1"); | 196 messages.push_back("1"); |
197 messages.push_back("2"); | 197 messages.push_back("2"); |
198 messages.push_back("3"); | 198 messages.push_back("3"); |
199 SendReceive(messages); | 199 SendReceive(messages); |
200 } | 200 } |
OLD | NEW |