network_sms_handler_unittest.cc revision 9ab5563a3196760eb381d102cbb2bc0f7abc6a50 
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    // Initialize DBusThreadManager with a stub implementation.
59    DBusThreadManager::InitializeWithStub();
60    ShillManagerClient::TestInterface* manager_test =
61        DBusThreadManager::Get()->GetShillManagerClient()->GetTestInterface();
62    ASSERT_TRUE(manager_test);
63    manager_test->AddDevice("stub_cellular_device2");
64    ShillDeviceClient::TestInterface* device_test =
65        DBusThreadManager::Get()->GetShillDeviceClient()->GetTestInterface();
66    ASSERT_TRUE(device_test);
67    device_test->AddDevice("stub_cellular_device2", flimflam::kTypeCellular,
68                           "/org/freedesktop/ModemManager1/stub/0");
69  }
70
71  virtual void TearDown() OVERRIDE {
72    DBusThreadManager::Shutdown();
73  }
74
75 protected:
76  base::MessageLoopForUI message_loop_;
77};
78
79TEST_F(NetworkSmsHandlerTest, SmsHandlerDbusStub) {
80  // Append '--sms-test-messages' to the command line to tell SMSClientStubImpl
81  // to generate a series of test SMS messages.
82  CommandLine* command_line = CommandLine::ForCurrentProcess();
83  command_line->AppendSwitch(chromeos::switches::kSmsTestMessages);
84
85  // This relies on the stub dbus implementations for ShillManagerClient,
86  // ShillDeviceClient, GsmSMSClient, ModemMessagingClient and SMSClient.
87  // Initialize a sms handler. The stub dbus clients will not send the
88  // first test message until RequestUpdate has been called.
89  scoped_ptr<NetworkSmsHandler> sms_handler(new NetworkSmsHandler());
90  scoped_ptr<TestObserver> test_observer(new TestObserver());
91  sms_handler->AddObserver(test_observer.get());
92  sms_handler->Init();
93  message_loop_.RunUntilIdle();
94  EXPECT_EQ(test_observer->message_count(), 0);
95
96  // Test that no messages have been received yet
97  const std::set<std::string>& messages(test_observer->messages());
98  // Note: The following string corresponds to values in
99  // ModemMessagingClientStubImpl and SmsClientStubImpl.
100  // TODO(stevenjb): Use a TestInterface to set this up to remove dependency.
101  const char kMessage1[] = "SMSClientStubImpl: Test Message: /SMS/0";
102  EXPECT_EQ(messages.find(kMessage1), messages.end());
103
104  // Test for messages delivered by signals.
105  test_observer->ClearMessages();
106  sms_handler->RequestUpdate();
107  message_loop_.RunUntilIdle();
108  EXPECT_GE(test_observer->message_count(), 1);
109  EXPECT_NE(messages.find(kMessage1), messages.end());
110}
111
112}  // namespace chromeos
113