def __init__(self): self.power = fake.FakePower() self.deploy = fake.FakeDeploy() self.a = fake.FakeVendorA() self.b = fake.FakeVendorB() self.mapping = { 'first_method': self.a, 'second_method': self.b, 'third_method_sync': self.b } self.vendor = utils.MixinVendorInterface(self.mapping) self.console = fake.FakeConsole() self.management = fake.FakeManagement() self.inspect = fake.FakeInspect()
def test_driver_passthru_mixin_success(self): vendor_a = fake.FakeVendorA() vendor_a.driver_vendor_passthru = mock.Mock() vendor_b = fake.FakeVendorB() vendor_b.driver_vendor_passthru = mock.Mock() driver_vendor_mapping = { 'method_a': vendor_a, 'method_b': vendor_b, } mixed_vendor = driver_utils.MixinVendorInterface({}, driver_vendor_mapping) mixed_vendor.driver_vendor_passthru('context', 'method_a', param1='p1') vendor_a.driver_vendor_passthru.assert_called_once_with('context', 'method_a', param1='p1')