Home | History | Annotate | Download | only in service 
      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 "cloud_print/service/service_state.h"
      6 
      7 #include "base/strings/string_util.h"
      8 #include "testing/gmock/include/gmock/gmock.h"
      9 #include "testing/gtest/include/gtest/gtest.h"
     10 
     11 using ::testing::_;
     12 using ::testing::Exactly;
     13 using ::testing::Return;
     14 
     15 TEST(ServiceStateTest, Empty) {
     16   ServiceState state;
     17   EXPECT_FALSE(state.IsValid());
     18 }
     19 
     20 TEST(ServiceStateTest, ToString) {
     21   ServiceState state;
     22   EXPECT_STREQ("{\"cloud_print\": {\"enabled\": true}}",
     23                CollapseWhitespaceASCII(state.ToString(), true).c_str());
     24   state.set_email("test (at) gmail.com");
     25   state.set_proxy_id("proxy");
     26   state.set_robot_email("robot (at) gmail.com");
     27   state.set_robot_token("abc");
     28   state.set_auth_token("token1");
     29   state.set_xmpp_auth_token("token2");
     30   EXPECT_TRUE(state.IsValid());
     31   EXPECT_STREQ("{\"cloud_print\": {\"auth_token\": \"token1\",\"email\": "
     32                "\"test (at) gmail.com\",\"enabled\": true,\"proxy_id\": \"proxy\","
     33                "\"robot_email\": \"robot (at) gmail.com\",\"robot_refresh_token\": "
     34                "\"abc\",\"xmpp_auth_token\": \"token2\"}}",
     35                CollapseWhitespaceASCII(state.ToString(), true).c_str());
     36 }
     37 
     38 TEST(ServiceStateTest, FromString) {
     39   ServiceState state;
     40   // Syntax error.
     41   EXPECT_FALSE(state.FromString("<\"cloud_print\": {\"enabled\": true}}"));
     42   // No data.
     43   EXPECT_FALSE(state.FromString("{\"cloud_print\": {\"enabled\": true}}"));
     44   EXPECT_FALSE(state.FromString(
     45       "{\"cloud_print\": {\"email\": \"test (at) gmail.com\"}}"));
     46   EXPECT_STREQ("test (at) gmail.com", state.email().c_str());
     47 
     48   // Good state.
     49   EXPECT_TRUE(state.FromString(
     50       "{\"cloud_print\": {\"email\": \"test2 (at) gmail.com\",\"enabled\": true,\""
     51       "proxy_id\": \"proxy\",\"robot_email\": \"robot (at) gmail.com\",\""
     52       "robot_refresh_token\": \"abc\"}}"));
     53   EXPECT_STREQ("test2 (at) gmail.com", state.email().c_str());
     54   EXPECT_STREQ("proxy", state.proxy_id().c_str());
     55   EXPECT_STREQ("robot (at) gmail.com", state.robot_email().c_str());
     56   EXPECT_STREQ("abc", state.robot_token().c_str());
     57 }
     58 
     59 class ServiceStateMock : public ServiceState {
     60  public:
     61   ServiceStateMock() {}
     62 
     63   MOCK_METHOD3(LoginToGoogle,
     64                std::string(const std::string& service,
     65                            const std::string& email,
     66                            const std::string& password));
     67 
     68  private:
     69   DISALLOW_COPY_AND_ASSIGN(ServiceStateMock);
     70 };
     71 
     72 TEST(ServiceStateTest, Configure) {
     73   ServiceStateMock state;
     74   state.set_email("test1 (at) gmail.com");
     75   state.set_proxy_id("id1");
     76 
     77   EXPECT_CALL(state, LoginToGoogle("cloudprint", "test2 (at) gmail.com", "abc"))
     78     .Times(Exactly(1))
     79     .WillOnce(Return("auth1"));
     80   EXPECT_CALL(state, LoginToGoogle("chromiumsync", "test2 (at) gmail.com", "abc"))
     81     .Times(Exactly(1))
     82     .WillOnce(Return("auth2"));
     83 
     84   EXPECT_TRUE(state.Configure("test2 (at) gmail.com", "abc", "id2"));
     85 
     86   EXPECT_STREQ("id2", state.proxy_id().c_str());
     87   EXPECT_STREQ("auth1", state.auth_token().c_str());
     88   EXPECT_STREQ("auth2", state.xmpp_auth_token().c_str());
     89 }
     90