class VoxelTest(unittest.TestCase): def setUp(self): Voxel.connectionCls.conn_classes = (None, VoxelMockHttp) VoxelMockHttp.type = None self.driver = Voxel(VOXEL_KEY, VOXEL_SECRET) def test_auth_failed(self): VoxelMockHttp.type = 'UNAUTHORIZED' try: self.driver.list_nodes() except Exception, e: self.assertTrue(isinstance(e, InvalidCredsError)) else:
class VoxelTest(unittest.TestCase): def setUp(self): Voxel.connectionCls.conn_classes = (None, VoxelMockHttp) VoxelMockHttp.type = None self.driver = Voxel(*VOXEL_PARAMS) def test_auth_failed(self): VoxelMockHttp.type = 'UNAUTHORIZED' try: self.driver.list_nodes() except Exception, e: self.assertTrue(isinstance(e, InvalidCredsError)) else:
def setUp(self): Voxel.connectionCls.conn_classes = (None, VoxelMockHttp) VoxelMockHttp.type = None self.driver = Voxel(*VOXEL_PARAMS)
class VoxelTest(unittest.TestCase): def setUp(self): Voxel.connectionCls.conn_classes = (None, VoxelMockHttp) VoxelMockHttp.type = None self.driver = Voxel(*VOXEL_PARAMS) def test_auth_failed(self): VoxelMockHttp.type = 'UNAUTHORIZED' try: self.driver.list_nodes() except Exception: e = sys.exc_info()[1] self.assertTrue(isinstance(e, InvalidCredsError)) else: self.fail('test should have thrown') def test_response_failure(self): VoxelMockHttp.type = 'FAILURE' try: self.driver.list_nodes() except Exception: pass else: self.fail('Invalid response, but exception was not thrown') def test_list_nodes(self): VoxelMockHttp.type = 'LIST_NODES' nodes = self.driver.list_nodes() self.assertEqual(len(nodes), 1) self.assertEqual(nodes[0].name, 'www.voxel.net') def test_list_sizes(self): sizes = self.driver.list_sizes() self.assertEqual(len(sizes), 13) def test_list_images(self): VoxelMockHttp.type = 'LIST_IMAGES' images = self.driver.list_images() self.assertEqual(len(images), 1) def test_list_locations(self): VoxelMockHttp.type = 'LIST_LOCATIONS' locations = self.driver.list_locations() self.assertEqual(len(locations), 2) self.assertEqual(locations[0].name, 'Amsterdam') def test_create_node_invalid_disk_size(self): image = NodeImage( id=1, name='Ubuntu 8.10 (intrepid)', driver=self.driver) size = NodeSize( 1, '256 slice', None, None, None, None, driver=self.driver) location = NodeLocation(id=1, name='Europe', country='England', driver=self.driver) try: self.driver.create_node(name='foo', image=image, size=size, location=location) except ValueError: pass else: self.fail('Invalid disk size provided but an exception was not' ' thrown') def test_create_node(self): VoxelMockHttp.type = 'CREATE_NODE' image = NodeImage( id=1, name='Ubuntu 8.10 (intrepid)', driver=self.driver) size = NodeSize( 1, '256 slice', 1024, 500, None, None, driver=self.driver) location = NodeLocation(id=1, name='Europe', country='England', driver=self.driver) node = self.driver.create_node(name='foo', image=image, size=size, location=location) self.assertEqual(node.id, '1234') node = self.driver.create_node(name='foo', image=image, size=size, location=location, voxel_access=True) self.assertEqual(node.id, '1234') def test_reboot_node(self): VoxelMockHttp.type = 'REBOOT_NODE' node = Node( id=72258, name=None, state=None, public_ips=None, private_ips=None, driver=self.driver) self.assertTrue(node.reboot()) def test_destroy_node(self): VoxelMockHttp.type = 'DESTROY_NODE' node = Node( id=72258, name=None, state=None, public_ips=None, private_ips=None, driver=self.driver) self.assertTrue(node.destroy())
def setUp(self): Voxel.connectionCls.conn_classes = (None, VoxelMockHttp) VoxelMockHttp.type = None self.driver = Voxel(VOXEL_KEY, VOXEL_SECRET)