test_integration.py

# coding: utf-8
import coreapi
import requests
import pytest


encoded = (
    b'{"_type":"document","_meta":{"url":"http://example.org"},'
    b'"a":123,"next":{"_type":"link"}}'
)


@pytest.fixture
def document():
    codec = coreapi.codecs.CoreJSONCodec()
    return codec.decode(encoded)


class MockResponse(object):
    def __init__(self, content):
        self.content = content
        self.headers = {}
        self.url = 'http://example.org'
        self.status_code = 200

    def iter_content(self, *args, **kwargs):
        n = 2
        list_of_chunks = list(self.content[i:i + n] for i in range(0, len(self.content), n))
        return list_of_chunks

    def close(self):
        return


# Basic integration tests.

def test_load():
    codec = coreapi.codecs.CoreJSONCodec()
    assert codec.decode(encoded) == {
        "a": 123,
        "next": coreapi.Link(url='http://example.org')
    }


def test_dump(document):
    codec = coreapi.codecs.CoreJSONCodec()
    content = codec.encode(document)
    assert content == encoded


def test_get(monkeypatch):
    def mockreturn(self, request, *args, **kwargs):
        return MockResponse(b'{"_type": "document", "example": 123}')

    monkeypatch.setattr(requests.Session, 'send', mockreturn)

    client = coreapi.Client()
    doc = client.get('http://example.org')
    assert doc == {'example': 123}


def test_follow(monkeypatch, document):
    def mockreturn(self, request, *args, **kwargs):
        return MockResponse(b'{"_type": "document", "example": 123}')

    monkeypatch.setattr(requests.Session, 'send', mockreturn)

    client = coreapi.Client()
    doc = client.action(document, ['next'])
    assert doc == {'example': 123}


def test_reload(monkeypatch):
    def mockreturn(self, request, *args, **kwargs):
        return MockResponse(b'{"_type": "document", "example": 123}')

    monkeypatch.setattr(requests.Session, 'send', mockreturn)

    client = coreapi.Client()
    doc = coreapi.Document(url='http://example.org')
    doc = client.reload(doc)
    assert doc == {'example': 123}


def test_error(monkeypatch, document):
    def mockreturn(self, request, *args, **kwargs):
        return MockResponse(b'{"_type": "error", "message": ["failed"]}')

    monkeypatch.setattr(requests.Session, 'send', mockreturn)

    client = coreapi.Client()
    with pytest.raises(coreapi.exceptions.ErrorMessage):
        client.action(document, ['next'])