chromeos/network/network_state_unittest.cc

   1 // Copyright (c) 2013 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_state.h"
   6
   7 #include "base/basictypes.h"
   8 #include "base/memory/scoped_ptr.h"
   9 #include "base/strings/string_number_conversions.h"
  10 #include "base/values.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12 #include "third_party/cros_system_api/dbus/service_constants.h"
  13
  14 namespace chromeos {
  15
  16 namespace {
  17
  18 class TestStringValue : public base::Value {
  19  public:
  20   explicit TestStringValue(const std::string& in_value)
  21       : base::Value(TYPE_STRING),
  22         value_(in_value) {
  23   }
  24
  25   virtual ~TestStringValue() {
  26   }
  27
  28   // Overridden from Value:
  29   virtual bool GetAsString(std::string* out_value) const OVERRIDE {
  30     if (out_value)
  31       *out_value = value_;
  32     return true;
  33   }
  34
  35   virtual TestStringValue* DeepCopy() const OVERRIDE {
  36     return new TestStringValue(value_);
  37   }
  38
  39   virtual bool Equals(const Value* other) const OVERRIDE {
  40     if (other->GetType() != GetType())
  41       return false;
  42     std::string lhs, rhs;
  43     return GetAsString(&lhs) && other->GetAsString(&rhs) && lhs == rhs;
  44   }
  45
  46  private:
  47   std::string value_;
  48 };
  49
  50 class NetworkStateTest : public testing::Test {
  51  public:
  52   NetworkStateTest() : network_state_("test_path") {
  53   }
  54
  55  protected:
  56   bool SetStringProperty(const std::string& key, const std::string& value) {
  57     if (!network_state_.PropertyChanged(key, TestStringValue(value)))
  58       return false;
  59     network_state_.InitialPropertiesReceived();
  60     return true;
  61   }
  62
  63   NetworkState network_state_;
  64
  65  private:
  66   DISALLOW_COPY_AND_ASSIGN(NetworkStateTest);
  67 };
  68
  69 }  // namespace
  70
  71 // Seting kNameProperty should set network name after call to
  72 // InitialPropertiesReceived() in SetStringProperty().
  73 TEST_F(NetworkStateTest, SsidAscii) {
  74   std::string wifi_setname = "SSID TEST";
  75   std::string wifi_setname_result = "SSID TEST";
  76   EXPECT_TRUE(SetStringProperty(flimflam::kNameProperty, wifi_setname));
  77   EXPECT_EQ(network_state_.name(), wifi_setname_result);
  78 }
  79
  80 TEST_F(NetworkStateTest, SsidAsciiWithNull) {
  81   std::string wifi_setname = "SSID TEST\x00xxx";
  82   std::string wifi_setname_result = "SSID TEST";
  83   EXPECT_TRUE(SetStringProperty(flimflam::kNameProperty, wifi_setname));
  84   EXPECT_EQ(network_state_.name(), wifi_setname_result);
  85 }
  86
  87 // UTF8 SSID
  88 TEST_F(NetworkStateTest, SsidUtf8) {
  89   std::string wifi_utf8 = "UTF-8 \u3042\u3044\u3046";
  90   std::string wifi_utf8_result = "UTF-8 \xE3\x81\x82\xE3\x81\x84\xE3\x81\x86";
  91   EXPECT_TRUE(SetStringProperty(flimflam::kNameProperty, wifi_utf8));
  92   EXPECT_EQ(network_state_.name(), wifi_utf8_result);
  93 }
  94
  95 // Truncates invalid UTF-8
  96 TEST_F(NetworkStateTest, SsidTruncateInvalid) {
  97   std::string wifi_setname2 = "SSID TEST \x01\xff!";
  98   std::string wifi_setname2_result = "SSID TEST \xEF\xBF\xBD\xEF\xBF\xBD!";
  99   EXPECT_TRUE(SetStringProperty(flimflam::kNameProperty, wifi_setname2));
 100   EXPECT_EQ(network_state_.name(), wifi_setname2_result);
 101 }
 102
 103 // latin1 SSID -> UTF8 SSID (Hex)
 104 TEST_F(NetworkStateTest, SsidLatin) {
 105   std::string wifi_latin1 = "latin-1 \xc0\xcb\xcc\xd6\xfb";
 106   std::string wifi_latin1_hex =
 107       base::HexEncode(wifi_latin1.c_str(), wifi_latin1.length());
 108   std::string wifi_latin1_result = "latin-1 \u00c0\u00cb\u00cc\u00d6\u00fb";
 109   EXPECT_TRUE(SetStringProperty(flimflam::kWifiHexSsid, wifi_latin1_hex));
 110   EXPECT_EQ(network_state_.name(), wifi_latin1_result);
 111 }
 112
 113 // Hex SSID
 114 TEST_F(NetworkStateTest, SsidHex) {
 115   std::string wifi_hex = "5468697320697320484558205353494421";
 116   std::string wifi_hex_result = "This is HEX SSID!";
 117   SetStringProperty(flimflam::kWifiHexSsid, wifi_hex);
 118   EXPECT_EQ(network_state_.name(), wifi_hex_result);
 119 }
 120
 121 }  // namespace chromeos