summaryrefslogtreecommitdiffstats
path: root/remoting/host/json_host_config_unittest.cc
blob: bd1bc14260eefc3deaf9bb65c2d246ad0c29aa93 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
// Copyright (c) 2010 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/file_util.h"
#include "base/message_loop.h"
#include "base/message_loop_proxy.h"
#include "base/path_service.h"
#include "base/ref_counted.h"
#include "base/scoped_temp_dir.h"
#include "remoting/host/json_host_config.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace remoting {

namespace {
const char* kTestConfig =
"{\n"
"  \"xmpp_login\" : \"test@gmail.com\",\n"
"  \"xmpp_auth_token\" : \"TEST_AUTH_TOKEN\",\n"
"  \"host_id\" : \"TEST_HOST_ID\",\n"
"  \"host_name\" : \"TEST_MACHINE_NAME\",\n"
"  \"private_key\" : \"TEST_PRIVATE_KEY\"\n"
"}\n";

// Helper for Update test.
class TestConfigUpdater : public base::RefCountedThreadSafe<TestConfigUpdater> {
 public:
  void DoUpdate(scoped_refptr<JsonHostConfig> target,
                  std::string new_auth_token_value) {
    target->SetString(kXmppAuthTokenConfigPath, new_auth_token_value);
  }
};
}  // namespace

class JsonHostConfigTest : public testing::Test {
 protected:
  virtual void SetUp() {
    message_loop_proxy_ = base::MessageLoopProxy::CreateForCurrentThread();
  }

  static void WriteTestFile(const FilePath& filename) {
    file_util::WriteFile(filename, kTestConfig, std::strlen(kTestConfig));
  }

  // The temporary directory used to contain the test operations.
  ScopedTempDir test_dir_;
  // A message loop that we can use as the file thread message loop.
  MessageLoop message_loop_;
  scoped_refptr<base::MessageLoopProxy> message_loop_proxy_;
};

TEST_F(JsonHostConfigTest, InvalidFile) {
  ASSERT_TRUE(test_dir_.CreateUniqueTempDir());
  FilePath non_existent_file =
      test_dir_.path().AppendASCII("non_existent.json");
  scoped_refptr<JsonHostConfig> target =
      new JsonHostConfig(non_existent_file, message_loop_proxy_.get());
  EXPECT_FALSE(target->Read());
}

TEST_F(JsonHostConfigTest, Read) {
  ASSERT_TRUE(test_dir_.CreateUniqueTempDir());
  FilePath test_file = test_dir_.path().AppendASCII("read.json");
  WriteTestFile(test_file);
  scoped_refptr<JsonHostConfig> target =
      new JsonHostConfig(test_file, message_loop_proxy_.get());
  ASSERT_TRUE(target->Read());

  std::string value;
  EXPECT_TRUE(target->GetString(kXmppLoginConfigPath, &value));
  EXPECT_EQ("test@gmail.com", value);
  EXPECT_TRUE(target->GetString(kXmppAuthTokenConfigPath, &value));
  EXPECT_EQ("TEST_AUTH_TOKEN", value);
  EXPECT_TRUE(target->GetString(kHostIdConfigPath, &value));
  EXPECT_EQ("TEST_HOST_ID", value);
  EXPECT_TRUE(target->GetString(kHostNameConfigPath, &value));
  EXPECT_EQ("TEST_MACHINE_NAME", value);
  EXPECT_TRUE(target->GetString(kPrivateKeyConfigPath, &value));
  EXPECT_EQ("TEST_PRIVATE_KEY", value);

  EXPECT_FALSE(target->GetString("non_existent_value", &value));
}

TEST_F(JsonHostConfigTest, Write) {
  ASSERT_TRUE(test_dir_.CreateUniqueTempDir());

  FilePath test_file = test_dir_.path().AppendASCII("write.json");
  WriteTestFile(test_file);
  scoped_refptr<JsonHostConfig> target(
      new JsonHostConfig(test_file, message_loop_proxy_.get()));
  ASSERT_TRUE(target->Read());

  std::string new_auth_token_value = "NEW_AUTH_TOKEN";
  scoped_refptr<TestConfigUpdater> config_updater(new TestConfigUpdater());
  target->Update(
      NewRunnableMethod(config_updater.get(), &TestConfigUpdater::DoUpdate,
                        target, new_auth_token_value));

  message_loop_.RunAllPending();

  // Now read the file again and check that the value has been written.
  scoped_refptr<JsonHostConfig> reader(
      new JsonHostConfig(test_file, message_loop_proxy_.get()));

  ASSERT_TRUE(reader->Read());

  std::string value;
  EXPECT_TRUE(reader->GetString(kXmppLoginConfigPath, &value));
  EXPECT_EQ("test@gmail.com", value);
  EXPECT_TRUE(reader->GetString(kXmppAuthTokenConfigPath, &value));
  EXPECT_EQ(new_auth_token_value, value);
  EXPECT_TRUE(reader->GetString(kHostIdConfigPath, &value));
  EXPECT_EQ("TEST_HOST_ID", value);
  EXPECT_TRUE(reader->GetString(kHostNameConfigPath, &value));
  EXPECT_EQ("TEST_MACHINE_NAME", value);
  EXPECT_TRUE(reader->GetString(kPrivateKeyConfigPath, &value));
  EXPECT_EQ("TEST_PRIVATE_KEY", value);
}

}