def given_the_env_is_set():
with patch.dict('os.environ', {
"SNS_ARN_BASE": TEST_SNS_ARN_BASE,
"STAGE": "test",
"GH_HOOK_SECRET": TEST_SECRET,
}):
yield
python类dict()的实例源码
def test_get_boolean_value_with_true():
with patch.dict('os.environ', {'test': 'True'}):
assert get_boolean_value('test', False)
def test_get_boolean_value_with_false():
with patch.dict('os.environ', {'test': 'False'}):
assert not get_boolean_value('test', True)
def test_get_boolean_value_with_default():
with patch.dict('os.environ', {'test': 'xxx'}):
assert get_boolean_value('test', 'default') == 'default'
def start(self, request, monkeypatch):
_env_list = self.env
_setup_dict = self.setup_dict
# first method for monkeypatching
def _setup(self, x):
_s = copy.deepcopy(_setup_dict)
return _s
# second method for monkeypatching
def _conf(self, x):
_e = copy.deepcopy(_env_list)
return _e
# monkeypatching methods _get_conf and _get_setup
monkeypatch.setattr(common3.Environment, "_get_conf", _conf)
monkeypatch.setattr(common3.Environment, "_get_setup", _setup)
# define environment with fake class
# mock Tkinter so we don't need Ixia installed
# I only know how to do this with mock, if monkeypatch can do this we should use that
with patch.dict('sys.modules', {'Tkinter': MagicMock()}):
env = common3.Environment(FakeOpts())
return env
# fake class for options
def test_multipletg_3(request, env):
"""Verify pytest_multipletg plugin restores env.tg dict on teardown.
"""
tg = MultipleTGClass(env)
tg.setup()
tg.teardown()
# Verify env.tg after teardown
assert list(env.tg.keys()) == [1, 2]
assert env.tg[1].ports == [(1, 1, 1), (1, 1, 2)]
assert env.tg[2].ports == [(1, 2, 1), (1, 2, 2)]
def setUp(self):
"""
Patch the DBus module
:return:
"""
self.dbus_mock = MagicMock()
self.dbus_exception_mock = MagicMock()
self.dbus_service_mock = MagicMock()
self.mainloop_mock = MagicMock()
self.gobject_mock = MagicMock()
modules = {
'dbus': self.dbus_mock,
'dbus.exceptions': self.dbus_exception_mock,
'dbus.service': self.dbus_service_mock,
'dbus.mainloop.glib': self.mainloop_mock,
'gi.repository': self.gobject_mock,
}
self.dbus_mock.Interface.return_value.GetManagedObjects.return_value = tests.obj_data.full_ubits
self.dbus_mock.Interface.return_value.Get = mock_get
self.dbus_mock.Interface.return_value.Set = mock_set
self.dbus_mock.SystemBus = MagicMock()
self.module_patcher = patch.dict('sys.modules', modules)
self.module_patcher.start()
from examples import adapter_example
self.module_under_test = adapter_example
def setUp(self):
"""
Patch the DBus module
:return:
"""
self.dbus_mock = MagicMock()
self.dbus_exception_mock = MagicMock()
self.dbus_service_mock = MagicMock()
self.mainloop_mock = MagicMock()
self.gobject_mock = MagicMock()
modules = {
'dbus': self.dbus_mock,
'dbus.exceptions': self.dbus_exception_mock,
'dbus.service': self.dbus_service_mock,
'dbus.mainloop.glib': self.mainloop_mock,
'gi.repository': self.gobject_mock,
}
self.dbus_mock.Interface.return_value.GetManagedObjects.return_value = tests.obj_data.full_ubits
self.dbus_mock.Interface.return_value.Get = mock_get
self.dbus_mock.Interface.return_value.Set = mock_set
self.dbus_mock.SystemBus = MagicMock()
self.module_patcher = patch.dict('sys.modules', modules)
self.module_patcher.start()
from bluezero import broadcaster
from bluezero import adapter
self.bz_adapter = adapter
self.module_under_test = broadcaster
self.path = '/org/bluez/hci0/dev_D4_AE_95_4C_3E_A4'
self.adapter_path = '/org/bluez/hci0'
self.dev_name = 'BBC micro:bit [zezet]'
self.address = 'D4:AE:95:4C:3E:A4'
def setUp(self):
"""
Patch the DBus module
:return:
"""
self.dbus_mock = MagicMock()
self.dbus_exception_mock = MagicMock()
self.dbus_service_mock = MagicMock()
self.mainloop_mock = MagicMock()
self.gobject_mock = MagicMock()
modules = {
'dbus': self.dbus_mock,
'dbus.exceptions': self.dbus_exception_mock,
'dbus.service': self.dbus_service_mock,
'dbus.mainloop.glib': self.mainloop_mock,
'gi.repository': self.gobject_mock,
}
self.dbus_mock.Interface.return_value.GetManagedObjects.return_value = tests.obj_data.full_ubits
self.dbus_mock.Interface.return_value.Get = mock_get
self.dbus_mock.Interface.return_value.Set = mock_set
self.dbus_mock.SystemBus = MagicMock()
self.module_patcher = patch.dict('sys.modules', modules)
self.module_patcher.start()
from bluezero import peripheral
self.module_under_test = peripheral
def setUp(self):
"""
Patch the DBus module
:return:
"""
self.dbus_mock = MagicMock()
self.dbus_exception_mock = MagicMock()
self.dbus_service_mock = MagicMock()
self.mainloop_mock = MagicMock()
self.gobject_mock = MagicMock()
modules = {
'dbus': self.dbus_mock,
'dbus.exceptions': self.dbus_exception_mock,
'dbus.service': self.dbus_service_mock,
'dbus.mainloop.glib': self.mainloop_mock,
'gi.repository': self.gobject_mock,
}
self.dbus_mock.Interface.return_value.GetManagedObjects.return_value = tests.obj_data.full_ubits
self.dbus_mock.Interface.return_value.Get = mock_get
self.dbus_mock.Interface.return_value.Set = mock_set
self.dbus_mock.Interface.return_value.ReadValue = mock_read
self.dbus_mock.SystemBus = MagicMock()
self.module_patcher = patch.dict('sys.modules', modules)
self.module_patcher.start()
from bluezero import microbit
self.module_under_test = microbit
def setUp(self):
"""
Patch the DBus module
:return:
"""
self.dbus_mock = MagicMock()
self.mainloop_mock = MagicMock()
self.gobject_mock = MagicMock()
modules = {
'dbus': self.dbus_mock,
'dbus.mainloop.glib': self.mainloop_mock,
'gi.repository': self.gobject_mock,
}
self.dbus_mock.Interface.return_value.GetManagedObjects.return_value = tests.obj_data.full_ubits
self.dbus_mock.Interface.return_value.Get = mock_get
self.dbus_mock.Interface.return_value.Set = mock_set
self.module_patcher = patch.dict('sys.modules', modules)
self.module_patcher.start()
from bluezero import device
self.module_under_test = device
self.path = '/org/bluez/hci0/dev_D4_AE_95_4C_3E_A4'
self.adapter_path = '/org/bluez/hci0'
self.dev_name = 'BBC micro:bit [zezet]'
self.adapter_addr = '00:00:00:00:5A:AD'
self.device_addr = 'D4:AE:95:4C:3E:A4'
def setUp(self):
"""
Patch the DBus module
:return:
"""
self.dbus_mock = MagicMock()
self.dbus_exception_mock = MagicMock()
self.dbus_service_mock = MagicMock()
self.mainloop_mock = MagicMock()
self.gobject_mock = MagicMock()
modules = {
'dbus': self.dbus_mock,
'dbus.exceptions': self.dbus_exception_mock,
'dbus.service': self.dbus_service_mock,
'dbus.mainloop.glib': self.mainloop_mock,
'gi.repository': self.gobject_mock,
}
self.dbus_mock.Interface.return_value.GetManagedObjects.return_value = tests.obj_data.full_ubits
self.dbus_mock.Interface.return_value.Get = mock_get
self.dbus_mock.Interface.return_value.Set = mock_set
self.dbus_mock.return_value
self.dbus_mock.SystemBus = MagicMock()
self.module_patcher = patch.dict('sys.modules', modules)
self.module_patcher.start()
from bluezero import advertisement
from bluezero import dbus_tools
self.module_under_test = advertisement
self.module_tools = dbus_tools
def setUp(self):
"""
Patch the DBus module
:return:
"""
self.dbus_mock = MagicMock()
self.dbus_exception_mock = MagicMock()
self.dbus_service_mock = MagicMock()
self.mainloop_mock = MagicMock()
self.gobject_mock = MagicMock()
modules = {
'dbus': self.dbus_mock,
'dbus.exceptions': self.dbus_exception_mock,
'dbus.service': self.dbus_service_mock,
'dbus.mainloop.glib': self.mainloop_mock,
'gi.repository': self.gobject_mock,
}
self.dbus_mock.Interface.return_value.GetManagedObjects.return_value = tests.obj_data.full_ubits
self.dbus_mock.Interface.return_value.Get = mock_get
self.dbus_mock.Interface.return_value.Set = mock_set
self.dbus_mock.SystemBus = MagicMock()
self.module_patcher = patch.dict('sys.modules', modules)
self.module_patcher.start()
from bluezero import eddystone
self.module_under_test = eddystone
def test_dict_context_manager(self):
foo = {}
with patch.dict(foo, {'a': 'b'}):
self.assertEqual(foo, {'a': 'b'})
self.assertEqual(foo, {})
with self.assertRaises(NameError):
with patch.dict(foo, {'a': 'b'}):
self.assertEqual(foo, {'a': 'b'})
raise NameError('Konrad')
self.assertEqual(foo, {})
def test_env_vars(self, _):
''' test environment variables '''
with patch.dict(os.environ, ENV_VAR='hello'):
self.assertEqual(Format('xyz {$ENV_VAR}'), 'xyz hello')
