network_sms_handler_unittest.cc revision a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7 
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
3// found in the LICENSE file.
4
5#include "chromeos/network/network_sms_handler.h"
6
7#include <set>
8#include <string>
9
10#include "base/command_line.h"
11#include "base/memory/scoped_ptr.h"
12#include "base/message_loop/message_loop.h"
13#include "chromeos/chromeos_switches.h"
14#include "chromeos/dbus/dbus_thread_manager.h"
15#include "chromeos/dbus/shill_device_client.h"
16#include "chromeos/dbus/shill_manager_client.h"
17#include "testing/gtest/include/gtest/gtest.h"
18#include "third_party/cros_system_api/dbus/service_constants.h"
19
20namespace chromeos {
21
22namespace {
23
24class TestObserver : public NetworkSmsHandler::Observer {
25 public:
26  TestObserver() {}
27  virtual ~TestObserver() {}
28
29  virtual void MessageReceived(const base::DictionaryValue& message) OVERRIDE {
30    std::string text;
31    if (message.GetStringWithoutPathExpansion(
32            NetworkSmsHandler::kTextKey, &text)) {
33      messages_.insert(text);
34    }
35  }
36
37  void ClearMessages() {
38    messages_.clear();
39  }
40
41  int message_count() { return messages_.size(); }
42  const std::set<std::string>& messages() const {
43    return messages_;
44  }
45
46 private:
47  std::set<std::string> messages_;
48};
49
50}  // namespace
51
52class NetworkSmsHandlerTest : public testing::Test {
53 public:
54  NetworkSmsHandlerTest() {}
55  virtual ~NetworkSmsHandlerTest() {}
56
57  virtual void SetUp() OVERRIDE {
58    // Append '--sms-test-messages' to the command line to tell
59    // SMSClientStubImpl to generate a series of test SMS messages.
60    CommandLine* command_line = CommandLine::ForCurrentProcess();
61    command_line->AppendSwitch(chromeos::switches::kSmsTestMessages);
62
63    // Initialize DBusThreadManager with a stub implementation.
64    DBusThreadManager::InitializeWithStub();
65    ShillManagerClient::TestInterface* manager_test =
66        DBusThreadManager::Get()->GetShillManagerClient()->GetTestInterface();
67    ASSERT_TRUE(manager_test);
68    manager_test->AddDevice("stub_cellular_device2");
69    ShillDeviceClient::TestInterface* device_test =
70        DBusThreadManager::Get()->GetShillDeviceClient()->GetTestInterface();
71    ASSERT_TRUE(device_test);
72    device_test->AddDevice("stub_cellular_device2", shill::kTypeCellular,
73                           "/org/freedesktop/ModemManager1/stub/0");
74
75    // This relies on the stub dbus implementations for ShillManagerClient,
76    // ShillDeviceClient, GsmSMSClient, ModemMessagingClient and SMSClient.
77    // Initialize a sms handler. The stub dbus clients will not send the
78    // first test message until RequestUpdate has been called.
79    network_sms_handler_.reset(new NetworkSmsHandler());
80    network_sms_handler_->Init();
81    test_observer_.reset(new TestObserver());
82    network_sms_handler_->AddObserver(test_observer_.get());
83    network_sms_handler_->RequestUpdate(true);
84    message_loop_.RunUntilIdle();
85  }
86
87  virtual void TearDown() OVERRIDE {
88    network_sms_handler_.reset();
89    DBusThreadManager::Shutdown();
90  }
91
92 protected:
93  base::MessageLoopForUI message_loop_;
94  scoped_ptr<NetworkSmsHandler> network_sms_handler_;
95  scoped_ptr<TestObserver> test_observer_;
96};
97
98TEST_F(NetworkSmsHandlerTest, SmsHandlerDbusStub) {
99  EXPECT_EQ(test_observer_->message_count(), 0);
100
101  // Test that no messages have been received yet
102  const std::set<std::string>& messages(test_observer_->messages());
103  // Note: The following string corresponds to values in
104  // ModemMessagingClientStubImpl and SmsClientStubImpl.
105  // TODO(stevenjb): Use a TestInterface to set this up to remove dependency.
106  const char kMessage1[] = "FakeSMSClient: Test Message: /SMS/0";
107  EXPECT_EQ(messages.find(kMessage1), messages.end());
108
109  // Test for messages delivered by signals.
110  test_observer_->ClearMessages();
111  network_sms_handler_->RequestUpdate(false);
112  message_loop_.RunUntilIdle();
113  EXPECT_GE(test_observer_->message_count(), 1);
114  EXPECT_NE(messages.find(kMessage1), messages.end());
115}
116
117}  // namespace chromeos
118