def test___str__(self): droplet = Droplet( Status.NEW, 21345, "This is a test", Size(1, "512M"), Image(34, "Ubuntu 10.02", "A linux distribution"), "1.2.3.4", Region(1, "USA"), False, ) self.assertEqual("21345: This is a test, new, 1.2.3.4", droplet.__str__())
def test_get(self): size_patcher = patch( 'docc.size.Size.get', new=MagicMock(return_value=Size(66, "512M")) ) size_patcher.start() image_patcher = patch( 'docc.image.Image.get', MagicMock( return_value=Image(25306, "Ubuntu 1", "Ubuntu Description") ) ) image_patcher.start() region_patcher = patch( 'docc.region.Region.get', MagicMock(return_value=Region(1, "USA")) ) region_patcher.start() def teardown(): size_patcher.stop() region_patcher.stop() image_patcher.stop() self.addCleanup(teardown) credentials = Credentials("abc", "def") service = Service(credentials) response = { "status": "OK", "droplet": { "id": 151220, "name": "cloud.segonds.org", "image_id": 25306, "size_id": 66, "region_id": 1, "backups_active": None, "ip_address": "208.68.38.181", "status": "active" } } mock = MagicMock(return_value=response) service.get = mock droplet = Droplet.get(service, 151220) self.assertEquals(droplet.id, 151220) self.assertEquals(droplet.name, "cloud.segonds.org") self.assertEquals(droplet.size, Size(66, "512M")) self.assertEquals(droplet.status, Status.ACTIVE)
def test_destroy(self): droplet = Droplet( Status.OFF, 21345, "This is a test", Size(1, "512M"), Image(34, "Ubuntu 10.02", "A linux distribution"), "1.2.3.4", Region(1, "USA"), False, ) credentials = Credentials("abc", "def") service = Service(credentials) response = { "status": "OK", "event_id": 1417387 } mock = MagicMock(return_value=response) service.get = mock self.assertTrue(droplet.destroy(service)) mock.assert_called_once_with( 'droplets/21345/destroy' )
def test_droplets(self): size_patcher = patch( 'docc.size.Size.get', new=MagicMock(return_value=Size(66, "512M")) ) size_patcher.start() image_patcher = patch( 'docc.image.Image.get', MagicMock( return_value=Image(25306, "Ubuntu 1", "Ubuntu Description") ) ) image_patcher.start() region_patcher = patch( 'docc.region.Region.get', MagicMock(return_value=Region(1, "USA")) ) region_patcher.start() def teardown(): size_patcher.stop() region_patcher.stop() image_patcher.stop() self.addCleanup(teardown) service_response = { "status": "OK", "droplets": [ { "id": 151220, "name": "cloud.segonds.org", "image_id": 25306, "size_id": 66, "region_id": 1, "backups_active": None, "ip_address": "208.68.38.181", "status": "new" }, { "id": 151221, "name": "cloud2.segonds.org", "image_id": 25306, "size_id": 66, "region_id": 1, "backups_active": None, "ip_address": "208.68.38.182", "status": "new" }, ] } credentials = Credentials("abc", "def") service = Service(credentials) service.get = MagicMock(return_value=service_response) droplets = Droplet.droplets(service) self.assertEquals(len(droplets), 2) droplet = droplets[0] self.assertEquals(droplet.id, 151220) self.assertEquals(droplet.name, "cloud.segonds.org") self.assertEquals(droplet.size, Size(66, "512M")) self.assertEquals(droplet.status, Status.NEW)