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='/')