class TestDeviceManager(TestCase):
instructions = (ObjectID(), )
def setUp(self):
self.manager = DeviceManager()
super(TestDeviceManager, self).setUp()
def test_custom_registry(self):
obj = object()
d = {'1.3.6.1.4.1.8072.3.2.10': obj}
registry = MagicMock(spec_set=dict)
registry.__getitem__.side_effect = lambda key: d[key]
cls = self.manager.get_class(host=self.test_host, port=self.test_port,
registry=registry)
self.assertTrue(obj is cls)