コード例 #1
0
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)
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)
コード例 #3
0
 def setUp(self):
     self.raw_vm = Mock()
     self.wrapped_vm = VM(self.raw_vm)
 def setUp(self):
     self.raw_vm = Mock()
     self.wrapped_vm = VM(self.raw_vm)