def init(): global _POLICY_PATH global _POLICY_CACHE if not _POLICY_PATH: _POLICY_PATH = utils.find_config(FLAGS.policy_file) utils.read_cached_file(_POLICY_PATH, _POLICY_CACHE, reload_func=_set_brain)
def init(): global _POLICY_PATH global _POLICY_CACHE if not _POLICY_PATH: _POLICY_PATH = utils.find_config(CONF.policy_file) utils.read_cached_file(_POLICY_PATH, _POLICY_CACHE, reload_func=_set_brain)
def test_read_modified_cached_file(self): self.mox.StubOutWithMock(os.path, "getmtime") self.mox.StubOutWithMock(__builtin__, 'open') os.path.getmtime(mox.IgnoreArg()).AndReturn(2) fake_contents = "lorem ipsum" fake_file = self.mox.CreateMockAnything() fake_file.read().AndReturn(fake_contents) fake_context_manager = self.mox.CreateMockAnything() fake_context_manager.__enter__().AndReturn(fake_file) fake_context_manager.__exit__(mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg()) __builtin__.open(mox.IgnoreArg()).AndReturn(fake_context_manager) self.mox.ReplayAll() cache_data = {"data": 1123, "mtime": 1} self.reload_called = False def test_reload(reloaded_data): self.assertEqual(reloaded_data, fake_contents) self.reload_called = True data = utils.read_cached_file("/this/is/a/fake", cache_data, reload_func=test_reload) self.assertEqual(data, fake_contents) self.assertTrue(self.reload_called)
def test_read_cached_file(self, mock_mtime, mock_open): fake_file = "/this/is/a/fake" cache_data = {"data": 1123, "mtime": 2} mock_open.return_value = _get_local_mock_open() data = utils.read_cached_file(fake_file, cache_data) self.assertEqual(cache_data["data"], data) mock_open.assert_called_once_with(fake_file)
def test_read_cached_file(self): self.mox.StubOutWithMock(os.path, "getmtime") os.path.getmtime(mox.IgnoreArg()).AndReturn(1) self.mox.ReplayAll() cache_data = {"data": 1123, "mtime": 1} data = utils.read_cached_file("/this/is/a/fake", cache_data) self.assertEqual(cache_data["data"], data)
def test_read_modified_cached_file(self, mock_mtime, mock_open): fake_data = 'lorem ipsum' fake_file = "/this/is/a/fake" mock_open.return_value = _get_local_mock_open(fake_data) cache_data = {"data": 'original data', "mtime": 2} mock_reload = mock.Mock() data = utils.read_cached_file(fake_file, cache_data, reload_func=mock_reload) self.assertEqual(data, fake_data) mock_reload.assert_called_once_with(fake_data) mock_open.assert_called_once_with(fake_file)