test_transitions.py

# coding: utf-8
from coreapi import Document, Link, Client
from coreapi.transports import HTTPTransport
from coreapi.transports.http import _handle_inplace_replacements
import pytest


class MockTransport(HTTPTransport):
    schemes = ['mock']

    def transition(self, link, decoders, params=None, link_ancestors=None, stream=True):
        if link.action == 'get':
            document = Document(title='new', content={'new': 123})
        elif link.action in ('put', 'post'):
            if params is None:
                params = {}
            document = Document(title='new', content={'new': 123, 'foo': params.get('foo')})
        else:
            document = None

        return _handle_inplace_replacements(document, link, link_ancestors)


client = Client(transports=[MockTransport()])


@pytest.fixture
def doc():
    return Document(title='original', content={
        'nested': Document(content={
            'follow': Link(url='mock://example.com', action='get'),
            'action': Link(url='mock://example.com', action='post', transform='inplace', fields=['foo']),
            'create': Link(url='mock://example.com', action='post', fields=['foo']),
            'update': Link(url='mock://example.com', action='put', fields=['foo']),
            'delete': Link(url='mock://example.com', action='delete')
        })
    })


# Test valid transitions.

def test_get(doc):
    new = client.action(doc, ['nested', 'follow'])
    assert new == {'new': 123}
    assert new.title == 'new'


def test_inline_post(doc):
    new = client.action(doc, ['nested', 'action'], params={'foo': 123})
    assert new == {'nested': {'new': 123, 'foo': 123}}
    assert new.title == 'original'


def test_post(doc):
    new = client.action(doc, ['nested', 'create'], params={'foo': 456})
    assert new == {'new': 123, 'foo': 456}
    assert new.title == 'new'


def test_put(doc):
    new = client.action(doc, ['nested', 'update'], params={'foo': 789})
    assert new == {'nested': {'new': 123, 'foo': 789}}
    assert new.title == 'original'


def test_delete(doc):
    new = client.action(doc, ['nested', 'delete'])
    assert new == {}
    assert new.title == 'original'


# Test overrides

def test_override_action(doc):
    new = client.action(doc, ['nested', 'follow'], overrides={'action': 'put'})
    assert new == {'nested': {'new': 123, 'foo': None}}
    assert new.title == 'original'


def test_override_transform(doc):
    new = client.action(doc, ['nested', 'update'], params={'foo': 456}, overrides={'transform': 'new'})
    assert new == {'new': 123, 'foo': 456}
    assert new.title == 'new'