def test_set_data(self):
data = Data('name')
expected = Query('href', 'rel', data=[data])
query = Query('href', 'rel')
query.data = [data]
self.assertIsInstance(query.data, Array)
self.assertEqual(query, expected)
def test_to_dict_minimal(self):
query = Query('href', 'rel')
expected = {
'href': 'href',
'rel': 'rel',
}
self.assertEqual(query.to_dict(), expected)
def test_to_dict_with_data(self):
data = [Data('name')]
query = Query('href', 'rel', 'name', 'prompt', data)
expected = {
'href': 'href',
'rel': 'rel',
'name': 'name',
'prompt': 'prompt',
'data': [
{'name': 'name'}
]
}
self.assertEqual(query.to_dict(), expected)
def test_set_data_invalid(self):
query = Query('href', 'rel')
invalid_obj = object()
with self.assertRaises(TypeError):
query.data = invalid_obj
