chromeos/network/network_sms_handler_unittest.cc

   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
  20 namespace chromeos {
  21
  22 namespace {
  23
  24 class 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
  52 class 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
  98 TEST_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