remoting/host/host_config_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 "remoting/host/host_config.h"
   6
   7 #include "base/files/file_util.h"
   8 #include "base/files/scoped_temp_dir.h"
   9 #include "base/memory/ref_counted.h"
  10 #include "base/values.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 namespace remoting {
  14
  15 namespace {
  16
  17 const char* kTestConfig =
  18 "{\n"
  19 "  \"xmpp_login\" : \"test@gmail.com\",\n"
  20 "  \"oauth_refresh_token\" : \"TEST_REFRESH_TOKEN\",\n"
  21 "  \"host_id\" : \"TEST_HOST_ID\",\n"
  22 "  \"host_name\" : \"TEST_MACHINE_NAME\",\n"
  23 "  \"private_key\" : \"TEST_PRIVATE_KEY\"\n"
  24 "}\n";
  25
  26 }  // namespace
  27
  28 class HostConfigTest : public testing::Test {
  29  protected:
  30   HostConfigTest() {}
  31
  32   static void WriteTestFile(const base::FilePath& filename) {
  33     base::WriteFile(filename, kTestConfig, std::strlen(kTestConfig));
  34   }
  35
  36   // The temporary directory used to contain the test operations.
  37   base::ScopedTempDir test_dir_;
  38
  39  private:
  40   DISALLOW_COPY_AND_ASSIGN(HostConfigTest);
  41 };
  42
  43 TEST_F(HostConfigTest, InvalidFile) {
  44   ASSERT_TRUE(test_dir_.CreateUniqueTempDir());
  45   base::FilePath non_existent_file =
  46       test_dir_.path().AppendASCII("non_existent.json");
  47   EXPECT_FALSE(HostConfigFromJsonFile(non_existent_file));
  48 }
  49
  50 TEST_F(HostConfigTest, Read) {
  51   ASSERT_TRUE(test_dir_.CreateUniqueTempDir());
  52   base::FilePath test_file = test_dir_.path().AppendASCII("read.json");
  53   WriteTestFile(test_file);
  54   scoped_ptr<base::DictionaryValue> target(HostConfigFromJsonFile(test_file));
  55   ASSERT_TRUE(target);
  56
  57   std::string value;
  58   EXPECT_TRUE(target->GetString(kXmppLoginConfigPath, &value));
  59   EXPECT_EQ("test@gmail.com", value);
  60   EXPECT_TRUE(target->GetString(kOAuthRefreshTokenConfigPath, &value));
  61   EXPECT_EQ("TEST_REFRESH_TOKEN", value);
  62   EXPECT_TRUE(target->GetString(kHostIdConfigPath, &value));
  63   EXPECT_EQ("TEST_HOST_ID", value);
  64   EXPECT_TRUE(target->GetString(kHostNameConfigPath, &value));
  65   EXPECT_EQ("TEST_MACHINE_NAME", value);
  66   EXPECT_TRUE(target->GetString(kPrivateKeyConfigPath, &value));
  67   EXPECT_EQ("TEST_PRIVATE_KEY", value);
  68
  69   EXPECT_FALSE(target->GetString("non_existent_value", &value));
  70 }
  71
  72 TEST_F(HostConfigTest, Write) {
  73   ASSERT_TRUE(test_dir_.CreateUniqueTempDir());
  74
  75   base::FilePath test_file = test_dir_.path().AppendASCII("write.json");
  76   WriteTestFile(test_file);
  77   scoped_ptr<base::DictionaryValue> target(HostConfigFromJsonFile(test_file));
  78   ASSERT_TRUE(target);
  79
  80   std::string new_refresh_token_value = "NEW_REFRESH_TOKEN";
  81   target->SetString(kOAuthRefreshTokenConfigPath, new_refresh_token_value);
  82   ASSERT_TRUE(HostConfigToJsonFile(*target, test_file));
  83
  84   // Now read the file again and check that the value has been written.
  85   scoped_ptr<base::DictionaryValue> reader(HostConfigFromJsonFile(test_file));
  86   ASSERT_TRUE(reader);
  87
  88   std::string value;
  89   EXPECT_TRUE(reader->GetString(kXmppLoginConfigPath, &value));
  90   EXPECT_EQ("test@gmail.com", value);
  91   EXPECT_TRUE(reader->GetString(kOAuthRefreshTokenConfigPath, &value));
  92   EXPECT_EQ(new_refresh_token_value, value);
  93   EXPECT_TRUE(reader->GetString(kHostIdConfigPath, &value));
  94   EXPECT_EQ("TEST_HOST_ID", value);
  95   EXPECT_TRUE(reader->GetString(kHostNameConfigPath, &value));
  96   EXPECT_EQ("TEST_MACHINE_NAME", value);
  97   EXPECT_TRUE(reader->GetString(kPrivateKeyConfigPath, &value));
  98   EXPECT_EQ("TEST_PRIVATE_KEY", value);
  99 }
 100
 101 }  // namespace remoting