Home | History | Annotate | Download | only in network 
      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", flimflam::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[] = "SMSClientStubImpl: 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