class VMTests(TestCase): def setUp(self): self.raw_vm = Mock() self.wrapped_vm = VM(self.raw_vm) def test_should_passthrough_unwrapped_attributes(self): self.assertEqual(self.wrapped_vm.anything, self.raw_vm.anything) def test_should_return_interface_when_one_matches(self): foo_mock = lambda: None foo_mock.name = "foo" bar_mock = lambda: None bar_mock.name = "bar" self.raw_vm.network = [foo_mock, bar_mock] bar = lambda n: n.name == "bar" actual = self.wrapped_vm.get_first_network_interface_matching(bar) self.assertEqual(actual, bar_mock) def test_should_return_first_interface_when_several_match(self): aha_mock = lambda: None aha_mock.name = "aha" foo_mock_1 = lambda: None foo_mock_1.name = "foo" bar_mock = lambda: None bar_mock.name = "bar" foo_mock_2 = lambda: None foo_mock_2.name = "foo" self.raw_vm.network = [aha_mock, foo_mock_1, bar_mock, foo_mock_2] foo = lambda n: n.name == "foo" actual = self.wrapped_vm.get_first_network_interface_matching(foo) self.assertEqual(actual, foo_mock_1)
def setUp(self): self.raw_vm = Mock() self.wrapped_vm = VM(self.raw_vm)