Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions flask_hal/document.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,9 @@ def __init__(self, data=None, links=None, embedded=None):
self.embedded = embedded or {}
self.links = links or link.Collection()

def __str__(self):
return str(self.to_dict())

@property
def links(self):
return self._links
Expand Down
48 changes: 48 additions & 0 deletions tests/test_document.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,54 @@
from flask_hal.document import Document, Embedded


def test_empty_document_str():
app = flask.Flask(__name__)
with app.test_request_context('/entity/231'):
document = Document()
assert str(document) == str({'_links': {'self': {'href': '/entity/231'}}})


def test_str_with_embedded_document():
app = flask.Flask(__name__)
with app.test_request_context('/entity/231'):
document = Document(
embedded={
'orders': Embedded(
embedded={'details': Embedded(
data={'details': {}}
)},
links=link.Collection(
link.Link('foo', 'www.foo.com'),
link.Link('boo', 'www.boo.com')
),
data={'total': 30},
)
},
data={'currentlyProcessing': 14}
)
expected = {
'_links': {
'self': {
'href': flask.request.path
}
},
'_embedded': {
'orders': {
'_links': {
'foo': {'href': 'www.foo.com'},
'boo': {'href': 'www.boo.com'}
},
'_embedded': {
'details': {'details': {}}
},
'total': 30,
}
},
'currentlyProcessing': 14
}
assert str(expected) == str(document)


def test_document_should_have_link_self():
app = flask.Flask(__name__)
with app.test_request_context('/entity/231'):
Expand Down