def test_children_returns_when_no_hits(self, mock_search):
mock_search().filter().filter().sort().extra(
).execute.return_value.hits.__len__.return_value = 0
children = IndexedFile.children('test_user',
system='test.system',
path='/')
self.assertEqual(children, ([], None))
def test_children_returns_when_hits(self, mock_search, mock_get):
search_res = IndexedFile(**{
'name': 'res1',
'system': 'test.system',
'path': '/path/to/res1'
})
search_res.meta.id = 'MOCK ID'
mock_search().filter().filter().sort().extra(
).execute.return_value.hits.__len__.return_value = 1
mock_search().filter().filter().sort().extra().execute(
).__iter__.return_value = [search_res]
mock_search().filter().filter().sort().extra(
).execute.return_value.hits.hits = [{
'sort': 'MOCK SORTKEY'
}]
mock_get.return_value = search_res
children = IndexedFile.children('test_user',
system='test.system',
path='/')
mock_get.assert_called_with('MOCK ID')
self.assertEqual(children, ([search_res], 'MOCK SORTKEY'))
def test_children_raises_on_404(self, mock_search):
mock_search().filter().filter().sort().extra(
).execute.side_effect = TransportError(404)
with self.assertRaises(TransportError):
IndexedFile.children('test_user', system='test.system', path='/')
