def test_document_object(self):
d = Document()
d.features = {
"item1": "val1"
}
d.content = {
"content1": "val1"
}
d.url = "http://example.com"
json = d.toJSON()
ob = ujson.loads(json)
self.assertEqual(
ob['features']['item1'],
d.features['item1']
)
self.assertEqual(ob['url'], d.url)
self.assertEqual(len(ob['content']), 1)
self.assertEqual(str(d), d.toJSON())
with mock.patch("ujson.dump") as dump:
dump.return_value = None
fp, fo = tempfile.mkstemp()
d.toJSON(fp)
self.assertEqual(fp, dump.call_args[0][0])
