def test_get_tier_config(get_mock): get_mock.return_value = _get_tier_config_response() tier_config = Directory().get_tier_config('TC-000-000-000') assert isinstance(tier_config, TierConfig) assert tier_config.id == 'TC-000-000-000' get_mock.assert_called_with( url='http://localhost:8080/api/public/v1/tier/configs/TC-000-000-000', headers={'Content-Type': 'application/json', 'Authorization': 'ApiKey XXXX:YYYYY'})
def test_get_asset(get_mock): get_mock.return_value = _get_asset_response() asset = Directory().get_asset('AS-9861-7949-8492') assert isinstance(asset, Asset) assert asset.id == 'AS-9861-7949-8492' get_mock.assert_called_with( url='http://localhost:8080/api/public/v1/assets/AS-9861-7949-8492', headers={'Content-Type': 'application/json', 'Authorization': 'ApiKey XXXX:YYYYY'})
def test_get_product(get_mock): get_mock.return_value = _get_product_response() product = Directory().get_product('CN-783-317-575') assert isinstance(product, Product) assert product.id == 'CN-783-317-575' get_mock.assert_called_with( url='http://localhost:8080/api/public/v1/products/CN-783-317-575', headers={'Content-Type': 'application/json', 'Authorization': 'ApiKey XXXX:YYYYY'})
def test_list_products(get_mock): get_mock.return_value = _get_array_response(_get_product_response()) products = Directory().list_products() assert isinstance(products, list) assert len(products) == 1 assert isinstance(products[0], Product) assert products[0].id == 'CN-783-317-575' get_mock.assert_called_with( url='http://localhost:8080/api/public/v1/products', headers={'Content-Type': 'application/json', 'Authorization': 'ApiKey XXXX:YYYYY'})
def get_tier_partner_data(self, account_id=None): """Look for domain name in tier1 configuration data. `partner_id` keeps this information for us""" filters = Query().equal('account.id', account_id) configs = list(Directory().list_tier_configs(filters)) for config in configs: for param in config.params: if param.id == 'partner_id': return param return None
def test_list_assets(get_mock): get_mock.return_value = _get_array_response(_get_asset_response()) assets = Directory().list_assets() assert isinstance(assets, list) assert len(assets) == 1 assert isinstance(assets[0], Asset) assert assets[0].id == 'AS-9861-7949-8492' get_mock.assert_called_with( url='http://localhost:8080/api/public/v1/assets?in(product.id,(CN-631-322-000))', headers={'Content-Type': 'application/json', 'Authorization': 'ApiKey XXXX:YYYYY'}, params=None)
def test_list_tier_configs(get_mock): get_mock.return_value = _get_array_response(_get_tier_config_response()) tier_configs = Directory().list_tier_configs() assert isinstance(tier_configs, list) assert len(tier_configs) == 1 assert isinstance(tier_configs[0], TierConfig) assert tier_configs[0].id == 'TC-000-000-000' get_mock.assert_called_with( url='http://localhost:8080/api/public/v1/tier/configs', headers={'Content-Type': 'application/json', 'Authorization': 'ApiKey XXXX:YYYYY'}, params={'product.id': 'CN-631-322-000'})
def test_get_asset_bad(): with pytest.raises(ServerError): Directory().get_asset('AS-9861-7949-8492')
def test_get_tier_config_bad(): with pytest.raises(ServerError): Directory().get_tier_config('TC-000-000-000')
def test_get_product_bad(): with pytest.raises(ServerError): Directory().get_product('CN-783-317-575')