# Copyright 2016 Canonical Ltd # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. from __future__ import absolute_import from __future__ import print_function import unittest import mock def FakeConfig(init_dict): def _config(key=None): return init_dict[key] if key else init_dict return _config class Helper(unittest.TestCase): def setUp(self): self._patches = {} self._patches_start = {} self.ch_config_patch = mock.patch('charmhelpers.core.hookenv.config') self.ch_config = self.ch_config_patch.start() self.ch_config.side_effect = lambda: {'ssl_param': None} def tearDown(self): for k, v in self._patches.items(): v.stop() setattr(self, k, None) self._patches = None self._patches_start = None self.ch_config_patch.stop() def patch(self, obj, attr, return_value=None, **kwargs): mocked = mock.patch.object(obj, attr, **kwargs) self._patches[attr] = mocked started = mocked.start() started.return_value = return_value self._patches_start[attr] = started setattr(self, attr, started) def patch_object(self, obj, attr, return_value=None, name=None, new=None): if name is None: name = attr if new is not None: mocked = mock.patch.object(obj, attr, new=new) else: mocked = mock.patch.object(obj, attr) self._patches[name] = mocked started = mocked.start() if new is None: started.return_value = return_value self._patches_start[name] = started setattr(self, name, started) class TestNeutronAristaCharm(Helper): def test_write_config(self): pass