def setUp(self): self.client = MagicMock() self.instance = SLImages(self.client) self.name = 'foo' self.guid = '1234' self.limit = '5555' self.marker = '4321'
class TestSLImages(unittest.TestCase): def setUp(self): self.client = MagicMock() self.instance = SLImages(self.client) self.name = 'foo' self.guid = '1234' self.limit = '5555' self.marker = '4321' def test_init(self): self.assertEquals(self.client, self.instance.client) @patch('SoftLayer.utils.query_filter') def test_get_private_images(self, queryMock): retExpected = self.client['Account'] retExpected.getPrivateBlockDeviceTemplateGroups.return_value = [] ret = self.instance.get_private_images(self.guid, self.name, self.limit, self.marker) queryMock.assert_any_call(self.name) queryMock.assert_any_call(self.guid) queryMock.assert_any_call('> %s' % self.marker) self.assertEquals(type(ret), list) @patch('SoftLayer.utils.query_filter') def test_get_public_image(self, queryMock): retExpected = self.client['Virtual_Guest_Block_Device_Template_Group'] retExpected.getPrivateBlockDeviceTemplateGroups.return_value = [] ret = self.instance.get_private_images(self.guid, self.name, self.limit, self.marker) queryMock.assert_any_call(self.name) queryMock.assert_any_call(self.guid) queryMock.assert_any_call('> %s' % self.marker) self.assertEquals(type(ret), list)