OLD | NEW |
---|---|
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 "base/compiler_specific.h" | |
5 #include "jingle/notifier/base/fake_base_task.h" | 6 #include "jingle/notifier/base/fake_base_task.h" |
6 #include "jingle/notifier/base/weak_xmpp_client.h" | 7 #include "jingle/notifier/base/weak_xmpp_client.h" |
8 #include "talk/xmpp/jid.h" | |
7 #include "talk/xmpp/asyncsocket.h" | 9 #include "talk/xmpp/asyncsocket.h" |
8 #include "testing/gmock/include/gmock/gmock.h" | 10 #include "testing/gmock/include/gmock/gmock.h" |
9 | 11 |
10 namespace notifier { | 12 namespace notifier { |
11 | 13 |
12 using ::testing::_; | 14 using ::testing::_; |
13 using ::testing::Return; | 15 using ::testing::Return; |
14 | 16 |
15 class MockAsyncSocket : public buzz::AsyncSocket { | 17 class MockAsyncSocket : public buzz::AsyncSocket { |
16 public: | 18 public: |
17 virtual ~MockAsyncSocket() {} | 19 virtual ~MockAsyncSocket() {} |
18 | 20 |
19 MOCK_METHOD0(state, State()); | 21 MOCK_METHOD0(state, State()); |
20 MOCK_METHOD0(error, Error()); | 22 MOCK_METHOD0(error, Error()); |
21 MOCK_METHOD0(GetError, int()); | 23 MOCK_METHOD0(GetError, int()); |
22 MOCK_METHOD1(Connect, bool(const talk_base::SocketAddress&)); | 24 MOCK_METHOD1(Connect, bool(const talk_base::SocketAddress&)); |
23 MOCK_METHOD3(Read, bool(char*, size_t, size_t*)); | 25 MOCK_METHOD3(Read, bool(char*, size_t, size_t*)); |
24 MOCK_METHOD2(Write, bool(const char*, size_t)); | 26 MOCK_METHOD2(Write, bool(const char*, size_t)); |
25 MOCK_METHOD0(Close, bool()); | 27 MOCK_METHOD0(Close, bool()); |
26 MOCK_METHOD1(StartTls, bool(const std::string&)); | 28 MOCK_METHOD1(StartTls, bool(const std::string&)); |
27 }; | 29 }; |
28 | 30 |
31 } // namespace notifier | |
32 | |
akalin
2012/01/26 03:54:25
this should be in an anon namespace, not the globa
ghc
2012/01/26 20:16:44
Done.
| |
33 // Extends WeakXmppClient to make jid() return a full jid, as required by | |
34 // PushNotificationsSubscribeTask. | |
35 class FakeWeakXmppClient : public notifier::WeakXmppClient { | |
36 public: | |
37 explicit FakeWeakXmppClient(talk_base::TaskParent* parent) | |
38 : notifier::WeakXmppClient(parent), | |
39 jid_("test@example.com/testresource") {} | |
40 | |
41 virtual ~FakeWeakXmppClient() {} | |
42 | |
43 virtual const buzz::Jid& jid() const OVERRIDE { | |
44 return jid_; | |
45 } | |
46 | |
47 private: | |
48 buzz::Jid jid_; | |
49 }; | |
50 | |
51 namespace notifier { | |
52 | |
29 FakeBaseTask::FakeBaseTask() { | 53 FakeBaseTask::FakeBaseTask() { |
30 // Owned by |task_pump_|. | 54 // Owned by |task_pump_|. |
31 notifier::WeakXmppClient* weak_xmpp_client = | 55 FakeWeakXmppClient* weak_xmpp_client = |
32 new notifier::WeakXmppClient(&task_pump_); | 56 new FakeWeakXmppClient(&task_pump_); |
33 | 57 |
34 weak_xmpp_client->Start(); | 58 weak_xmpp_client->Start(); |
35 buzz::XmppClientSettings settings; | 59 buzz::XmppClientSettings settings; |
36 // Owned by |weak_xmpp_client|. | 60 // Owned by |weak_xmpp_client|. |
37 MockAsyncSocket* mock_async_socket = new MockAsyncSocket(); | 61 MockAsyncSocket* mock_async_socket = new MockAsyncSocket(); |
38 EXPECT_CALL(*mock_async_socket, Connect(_)).WillOnce(Return(true)); | 62 EXPECT_CALL(*mock_async_socket, Connect(_)).WillOnce(Return(true)); |
39 weak_xmpp_client->Connect(settings, "en", mock_async_socket, NULL); | 63 weak_xmpp_client->Connect(settings, "en", mock_async_socket, NULL); |
40 // Initialize the XMPP client. | 64 // Initialize the XMPP client. |
41 task_pump_.RunTasks(); | 65 task_pump_.RunTasks(); |
42 | 66 |
43 base_task_ = weak_xmpp_client->AsWeakPtr(); | 67 base_task_ = weak_xmpp_client->AsWeakPtr(); |
44 } | 68 } |
45 | 69 |
46 FakeBaseTask::~FakeBaseTask() {} | 70 FakeBaseTask::~FakeBaseTask() {} |
47 | 71 |
48 base::WeakPtr<buzz::XmppTaskParentInterface> FakeBaseTask::AsWeakPtr() { | 72 base::WeakPtr<buzz::XmppTaskParentInterface> FakeBaseTask::AsWeakPtr() { |
49 return base_task_; | 73 return base_task_; |
50 } | 74 } |
51 | 75 |
52 } // namespace notifier | 76 } // namespace notifier |
OLD | NEW |