|
| 1 | +import json |
| 2 | +import os |
| 3 | +from collections import namedtuple |
| 4 | + |
| 5 | +from mock import MagicMock, patch |
| 6 | + |
| 7 | +from connect.config import Config |
| 8 | +from connect.resource import FulfillmentResource |
| 9 | + |
| 10 | +response = namedtuple('Response', ('ok', 'content')) |
| 11 | + |
| 12 | + |
| 13 | +def _get_response_ok(): |
| 14 | + response.ok = True |
| 15 | + with open(os.path.join(os.path.dirname(__file__), 'response.json')) as file: |
| 16 | + response.content = file.read() |
| 17 | + |
| 18 | + return response |
| 19 | + |
| 20 | + |
| 21 | +@patch('requests.get', MagicMock(return_value=_get_response_ok())) |
| 22 | +def test_create_model_from_response(): |
| 23 | + Config(api_key='ApiKey XXXX:YYYYY', api_url='http://localhost:8080/api/public/v1/') |
| 24 | + |
| 25 | + requests = FulfillmentResource().list() |
| 26 | + request_obj = FulfillmentResource().get(pk='PR-000-000-000') |
| 27 | + |
| 28 | + assert requests[0].id == request_obj.id |
| 29 | + content = json.loads(response.content)[0] |
| 30 | + assert request_obj.id == content['id'] |
| 31 | + assert request_obj.contract.id == content['contract']['id'] |
| 32 | + assert request_obj.marketplace.id == content['marketplace']['id'] |
| 33 | + assert request_obj.asset.id == content['asset']['id'] |
| 34 | + assert request_obj.asset.product.id == content['asset']['product']['id'] |
| 35 | + |
| 36 | + try: |
| 37 | + ( |
| 38 | + request_obj.contract, |
| 39 | + request_obj.contract.id, |
| 40 | + request_obj.marketplace, |
| 41 | + request_obj.marketplace.id, |
| 42 | + request_obj.type, |
| 43 | + request_obj.updated, |
| 44 | + request_obj.created, |
| 45 | + request_obj.reason, |
| 46 | + request_obj.activation_key, |
| 47 | + request_obj.status, |
| 48 | + request_obj.asset.external_id, |
| 49 | + request_obj.asset.external_uid, |
| 50 | + request_obj.asset.product, |
| 51 | + request_obj.asset.product.id, |
| 52 | + request_obj.asset.connection, |
| 53 | + request_obj.asset.items, |
| 54 | + request_obj.asset.params, |
| 55 | + request_obj.asset.tiers, |
| 56 | + ) |
| 57 | + except AttributeError: |
| 58 | + assert False, 'Incorrectly initialized model ' |
0 commit comments