Пример #1
0
 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))
Пример #2
0
 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))
Пример #3
0
 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))
Пример #4
0
 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))
Пример #5
0
    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)