Exemplo n.º 1
0
 def test_create_provider_connection_error(self):
     """Test to create a provider with a connection error."""
     client = KokuHTTPClient(auth_header=Config.SOURCES_FAKE_HEADER)
     with requests_mock.mock() as m:
         m.post('http://www.koku.com/api/cost-management/v1/providers/',
                exc=requests.exceptions.RequestException)
         with self.assertRaises(KokuHTTPClientError):
             client.create_provider(self.name, self.provider_type, self.authentication, self.billing_source)
Exemplo n.º 2
0
 def test_create_provider_bad_permissions(self):
     """Test to create a provider with a bad permissions ."""
     client = KokuHTTPClient(auth_header=Config.SOURCES_FAKE_HEADER)
     with requests_mock.mock() as m:
         m.post('http://www.koku.com/api/cost-management/v1/providers/',
                status_code=401,
                json={'uuid': faker.uuid4()})
         with self.assertRaises(KokuHTTPClientNonRecoverableError):
             client.create_provider(self.name, self.provider_type, self.authentication, self.billing_source)
Exemplo n.º 3
0
def execute_koku_provider_op(msg, cost_management_type_id):
    """
    Execute the 'create' or 'destroy Koku-Provider operations.

    'create' operations:
        Koku POST /providers is executed along with updating the Sources database table with
        the Koku Provider uuid.
    'destroy' operations:
        Koku DELETE /providers is executed along with removing the Sources database entry.

    Two types of exceptions are handled for Koku HTTP operations.  Recoverable client and
    Non-Recoverable client errors.  If the error is recoverable the calling function
    (synchronize_sources) will re-queue the operation.

    Args:
        msg (Asyncio msg): Dictionary messages containing operation,
                                       provider and offset.
            example: {'operation': 'create', 'provider': SourcesModelObj, 'offset': 3}
        cost_management_type_id (Integer): Cost Management Type Identifier

    Returns:
        None

    """
    provider = msg.get('provider')
    operation = msg.get('operation')
    koku_client = KokuHTTPClient(provider.auth_header)
    sources_client = SourcesHTTPClient(provider.auth_header, provider.source_id)
    try:
        if operation == 'create':
            LOG.info(f'Creating Koku Provider for Source ID: {str(provider.source_id)}')
            koku_details = koku_client.create_provider(provider.name, provider.source_type, provider.authentication,
                                                       provider.billing_source, provider.source_uuid)
            LOG.info(f'Koku Provider UUID {koku_details.get("uuid")} assigned to Source ID {str(provider.source_id)}.')
            storage.add_provider_koku_uuid(provider.source_id, koku_details.get('uuid'))
        elif operation == 'destroy':
            if provider.koku_uuid:
                try:
                    response = koku_client.destroy_provider(provider.koku_uuid)
                    LOG.info(
                        f'Koku Provider UUID ({provider.koku_uuid}) Removal Status Code: {str(response.status_code)}')
                except KokuHTTPClientNonRecoverableError:
                    LOG.info(f'Koku Provider already removed.  Remove Source ID: {str(provider.source_id)}.')
            storage.destroy_provider_event(provider.source_id)
        elif operation == 'update':
            koku_details = koku_client.update_provider(provider.koku_uuid, provider.name, provider.source_type,
                                                       provider.authentication, provider.billing_source)
            storage.clear_update_flag(provider.source_id)
            LOG.info(f'Koku Provider UUID {koku_details.get("uuid")} with Source ID {str(provider.source_id)} updated.')
        sources_client.set_source_status(None, cost_management_type_id)

    except KokuHTTPClientError as koku_error:
        raise SourcesIntegrationError('Koku provider error: ', str(koku_error))
    except KokuHTTPClientNonRecoverableError as koku_error:
        err_msg = f'Unable to {operation} provider for Source ID: {str(provider.source_id)}. Reason: {str(koku_error)}'
        LOG.error(err_msg)
        sources_client.set_source_status(str(koku_error), cost_management_type_id)
Exemplo n.º 4
0
 def test_create_provider(self):
     """Test to create a provider."""
     client = KokuHTTPClient(auth_header=Config.SOURCES_FAKE_HEADER)
     expected_uuid = faker.uuid4()
     with requests_mock.mock() as m:
         m.post('http://www.koku.com/api/cost-management/v1/providers/',
                status_code=201,
                json={'uuid': expected_uuid})
         response = client.create_provider(self.name, self.provider_type, self.authentication, self.billing_source)
         self.assertEqual(response.get('uuid'), expected_uuid)