def test_get(self): with patch.object(Document, 'get') as mock_method: doc = DocumentCollection(Mock()) doc.get('collection_test', 'key1') mock_method.assert_called_with(kind='collections', collection='collection_test', key='key1')
def test_post(self): with patch.object(Document, 'post') as mock_method: doc = DocumentCollection(Mock()) doc.post('collection_test', {'doc': 1}) mock_method.assert_called_with(kind='collections', collection='collection_test', document={'doc': 1})
def test_list(self): with patch.object(Document, 'list') as mock_method: doc = DocumentCollection(Mock()) doc.list('collection_test', 10, 1) mock_method.assert_called_with(kind='collections', collection='collection_test', per_page=10, page=1)
def test_clear(self): with patch.object(Document, 'clear') as mock_method: doc = DocumentCollection(Mock()) query = [[{'field': 'name', 'operator': 'LIKE', 'value': 'test'}]] doc.clear('collection_test', query) mock_method.assert_called_with(kind='collections', collection='collection_test', query=query)
def test_search_with_pagination(self): with patch.object(Document, 'search') as mock_method: doc = DocumentCollection(Mock()) query = [[{'field': 'name', 'operator': 'LIKE', 'value': 'test'}]] doc.search('collection_test', query, 20, 2) mock_method.assert_called_with(kind='collections', collection='collection_test', query=query, per_page=20, page=2)
def test_search_many_coll_with_pagination(self): doc = DocumentCollection(Mock()) doc.make_request = Mock() collections = ['collection_test1', 'collection_test2'] query = [[{'field': 'name', 'operator': 'LIKE', 'value': 'test'}]] doc.search_many_coll(collections, query, 20, 2) params = { 'query': json.dumps(query), 'per_page': 20, 'collections': collections, 'page': 2 } doc.make_request.assert_called_once_with(method='GET', uri='collections/search', params=params)