still testing
This commit is contained in:
37
tests/io/test_json.py
Normal file
37
tests/io/test_json.py
Normal file
@ -0,0 +1,37 @@
|
||||
import pytest
|
||||
|
||||
from bonobo import to_json
|
||||
from bonobo.util.lifecycle import get_initializer, get_finalizer
|
||||
|
||||
|
||||
class ContextMock:
|
||||
pass
|
||||
|
||||
|
||||
def test_write_json_to_file(tmpdir):
|
||||
file = tmpdir.join('output.json')
|
||||
json_writer = to_json(file)
|
||||
context = ContextMock()
|
||||
|
||||
get_initializer(json_writer)(context)
|
||||
json_writer(context, {'foo': 'bar'})
|
||||
get_finalizer(json_writer)(context)
|
||||
|
||||
assert file.read() == '''[
|
||||
{"foo": "bar"}
|
||||
]'''
|
||||
|
||||
with pytest.raises(AttributeError):
|
||||
getattr(context, 'fp')
|
||||
|
||||
with pytest.raises(AttributeError):
|
||||
getattr(context, 'first')
|
||||
|
||||
|
||||
def test_write_json_without_initializer_should_not_work(tmpdir):
|
||||
file = tmpdir.join('output.json')
|
||||
json_writer = to_json(file)
|
||||
|
||||
context = ContextMock()
|
||||
with pytest.raises(AttributeError):
|
||||
json_writer(context, {'foo': 'bar'})
|
||||
Reference in New Issue
Block a user