Python UsageAutomation Examples

Example #1

def test_get_usage_template_ok(get_mock):
    get_mock.side_effect = [
        Response(ok=True, text='{"template_link": "..."}', status_code=200),
        BinaryResponse(ok=True, content=b'template_contents', status_code=200)]
    resource = UsageAutomation()
    assert resource.get_usage_template(Product(id='PRD-638-321-603')) == b'template_contents'
    get_mock.assert_has_calls([
        call(
            url='http://localhost:8080/api/public/v1/usage/products/PRD-638-321-603/template/',
            headers={'Content-Type': 'application/json', 'Authorization': 'ApiKey XXXX:YYYYY'},
            timeout=300),
        call('...')
    ])

Example #2

File: test_usage.py Project: maxipavlovic/connect-python-sdk

def test_get_usage_template_no_file():
    resource = UsageAutomation()
    with pytest.raises(FileRetrievalError):
        resource.get_usage_template(Product(id='PRD-638-321-603'))

Example #3

File: test_usage.py Project: maxipavlovic/connect-python-sdk

def test_create_resource():
    requests = UsageAutomation().list()
    assert isinstance(requests, list)
    assert len(requests) == 8