Open Source Monorail

appengine/monorail/services/test/client_config_svc_test.py

+# Copyright 2016 The Chromium Authors. All rights reserved.
+# Use of this source code is govered by a BSD-style
+# license that can be found in the LICENSE file or at
+# https://developers.google.com/open-source/licenses/bsd
+
+"""Tests for the client config service."""
+
+import unittest
+
+from services import client_config_svc
+
+
+class ClientConfigServiceTest(unittest.TestCase):
+
+  def setUp(self):
+    self.client_config_svc = client_config_svc.GetClientConfigSvc()
+    self.client_email = '123456789@developer.gserviceaccount.com'
+    self.client_id = '123456789.apps.googleusercontent.com'
+
+  def testGetDisplayNames(self):
+    display_names_map = self.client_config_svc.GetDisplayNames()
+    self.assertIn(self.client_email, display_names_map)
+    self.assertEquals('johndoe@example.com',
+                      display_names_map[self.client_email])
+
+  def testGetClientIDEmails(self):
+    auth_client_ids, auth_emails = self.client_config_svc.GetClientIDEmails()
+    self.assertIn(self.client_id, auth_client_ids)
+    self.assertIn(self.client_email, auth_emails)
+
+  def testForceLoad(self):
+    # First time it will always read the config
+    self.client_config_svc.load_time = 10000
+    self.client_config_svc.GetConfigs(use_cache=True)
+    self.assertNotEquals(10000, self.client_config_svc.load_time)
+
+    # use_cache is false and it will read the config
+    self.client_config_svc.load_time = 10000
+    self.client_config_svc.GetConfigs(use_cache=False, cur_time=11000)
+    self.assertNotEquals(10000, self.client_config_svc.load_time)
+
+    # Cache expires after 3600 sec and it will read the config
+    self.client_config_svc.load_time = 10000
+    self.client_config_svc.GetConfigs(use_cache=True, cur_time=20000)
+    self.assertNotEquals(10000, self.client_config_svc.load_time)
+
+    # otherwise it should just use the cache
+    self.client_config_svc.load_time = 10000
+    self.client_config_svc.GetConfigs(use_cache=True, cur_time=11000)
+    self.assertEquals(10000, self.client_config_svc.load_time)