def set_mock_items(*items): db_api.__db_items = copy.copy(items) db_api.get_items.side_effect = ( tools.get_db_api_get_items(*db_api.__db_items)) db_api.get_item_by_id.side_effect = ( tools.get_db_api_get_item_by_id(*db_api.__db_items)) db_api.get_items_by_ids.side_effect = ( tools.get_db_api_get_items_by_ids(*db_api.__db_items)) db_api.get_items_ids.side_effect = ( tools.get_db_api_get_items_ids(*db_api.__db_items))
def set_mock_items(*items): db_api.__db_items = copy.copy(items) db_api.get_items.side_effect = (tools.get_db_api_get_items( *db_api.__db_items)) db_api.get_item_by_id.side_effect = ( tools.get_db_api_get_item_by_id(*db_api.__db_items)) db_api.get_items_by_ids.side_effect = ( tools.get_db_api_get_items_by_ids(*db_api.__db_items)) db_api.get_items_ids.side_effect = (tools.get_db_api_get_items_ids( *db_api.__db_items))
def set_mock_db_items(self, *items): self._db_items = copy.copy(items) self.db_api.get_items.side_effect = (tools.get_db_api_get_items( *self._db_items)) self.db_api.get_item_by_id.side_effect = ( tools.get_db_api_get_item_by_id(*self._db_items)) self.db_api.get_items_by_ids.side_effect = ( tools.get_db_api_get_items_by_ids(*self._db_items)) self.db_api.get_items_ids.side_effect = ( tools.get_db_api_get_items_ids(*self._db_items))
def set_mock_db_items(self, *items): self._db_items = copy.copy(items) self.db_api.get_items.side_effect = ( tools.get_db_api_get_items(*self._db_items)) self.db_api.get_item_by_id.side_effect = ( tools.get_db_api_get_item_by_id(*self._db_items)) self.db_api.get_items_by_ids.side_effect = ( tools.get_db_api_get_items_by_ids(*self._db_items)) self.db_api.get_items_ids.side_effect = ( tools.get_db_api_get_items_ids(*self._db_items))
def test_create_vpc_rollback(self, create_route_table, create_default_security_group, create_internet_gateway, attach_internet_gateway, create_subnet, create_route, detach_internet_gateway): self.configure(disable_ec2_classic=True) self.neutron.create_router.side_effect = (tools.get_neutron_create( 'router', fakes.ID_OS_ROUTER_DEFAULT)) self.db_api.add_item.side_effect = (tools.get_db_api_add_item( {'vpc': fakes.ID_EC2_VPC_DEFAULT})) DB_IGW_DEFAULT_DETACHED = ({ 'id': fakes.ID_EC2_IGW_DEFAULT, 'os_id': None, 'vpc_id': None }) self.db_api.get_item_by_id.side_effect = ( tools.get_db_api_get_item_by_id(fakes.DB_VPC_DEFAULT, fakes.DB_SUBNET_DEFAULT, fakes.DB_SECURITY_GROUP_DEFAULT, DB_IGW_DEFAULT_DETACHED)) create_route_table.return_value = fakes.DB_ROUTE_TABLE_DEFAULT create_internet_gateway.return_value = { 'internetGateway': fakes.EC2_IGW_DEFAULT } create_subnet.return_value = {'subnet': fakes.EC2_SUBNET_DEFAULT} create_default_security_group.return_value = ( fakes.ID_EC2_SECURITY_GROUP_DEFAULT) # exception during attaching internet gateway create_route.side_effect = Exception() vpc_api._check_and_create_default_vpc(self.context) detach_internet_gateway.assert_any_call(mock.ANY, fakes.ID_EC2_IGW_DEFAULT, fakes.ID_EC2_VPC_DEFAULT) self.db_api.delete_item.assert_any_call(mock.ANY, fakes.ID_EC2_SUBNET_DEFAULT) self.db_api.delete_item.assert_any_call(mock.ANY, fakes.ID_EC2_IGW_DEFAULT) self.neutron.delete_security_group.assert_any_call( fakes.ID_OS_SECURITY_GROUP_DEFAULT) self.db_api.delete_item.assert_any_call( mock.ANY, fakes.ID_EC2_ROUTE_TABLE_DEFAULT) self.db_api.delete_item.assert_any_call(mock.ANY, fakes.ID_EC2_VPC_DEFAULT) self.neutron.delete_router.assert_called_once_with( fakes.ID_OS_ROUTER_DEFAULT)