def test_after_delete3(self, dummy_request):
from kotti.resources import Document
document1 = Document()
document1.request = dummy_request
document2 = Document()
document2.request = dummy_request
from kotti_es.sqla import _after_delete
_after_delete(None, None, document1)
_after_delete(None, None, document2)
assert len(getattr(dummy_request, '_index_list')) == 2
def test_after_delete3(self, dummy_request):
from kotti.resources import Document
document1 = Document()
document1.request = dummy_request
document2 = Document()
document2.request = dummy_request
from kotti_es.sqla import _after_delete
_after_delete(None, None, document1)
_after_delete(None, None, document2)
assert len(getattr(dummy_request, '_index_list')) == 2
def test_after_commit_delete(self, dummy_request):
from kotti.resources import Document
from kotti_es.sqla import _after_delete
from kotti_es.sqla import _after_commit
document = Document()
document.request = dummy_request
_after_delete(None, None, document)
import mock
with mock.patch('kotti_es.sqla.get_current_request') as \
get_current_request:
get_current_request.return_value = dummy_request
with mock.patch('kotti_es.sqla.get_client') as \
get_client:
magic = mock.MagicMock()
get_client.return_value = magic
session = mock.Mock()
session.configure_mock(
**{
'query.return_value.filter_by.'
'return_value.one.return_value':
document
})
_after_commit(session)
magic.delete_object.assert_called_with(document,
immediate=True)
def test_after_delete2(self, dummy_request):
from kotti.resources import Document
document = Document()
document.id = 1
document.request = dummy_request
document.request._index_list = []
from kotti_es.sqla import _after_delete
from kotti_es.sqla import DELETE_CODE
_after_delete(None, None, document)
assert getattr(document.request,
'_index_list') == [(document, document.id, DELETE_CODE)]
def test_after_insert(self, dummy_request):
from kotti.resources import Document
document = Document()
document.id = 1
document.request = dummy_request
assert hasattr(document.request, '_index_list') is False
from kotti_es.sqla import _after_insert_update
from kotti_es.sqla import INSERT_CODE
_after_insert_update(None, None, document)
assert getattr(document.request,
'_index_list') == [(document, document.id, INSERT_CODE)]
def test_after_delete2(self, dummy_request):
from kotti.resources import Document
document = Document()
document.id = 1
document.request = dummy_request
document.request._index_list = []
from kotti_es.sqla import _after_delete
from kotti_es.sqla import DELETE_CODE
_after_delete(None, None, document)
assert getattr(document.request, '_index_list') == [
(document, document.id, DELETE_CODE)
]
def test_after_insert(self, dummy_request):
from kotti.resources import Document
document = Document()
document.id = 1
document.request = dummy_request
assert hasattr(document.request, '_index_list') is False
from kotti_es.sqla import _after_insert_update
from kotti_es.sqla import INSERT_CODE
_after_insert_update(None, None, document)
assert getattr(document.request, '_index_list') == [
(document, document.id, INSERT_CODE)
]
def test_insert_delete(self, es_client, root,
dummy_request, db_session):
import transaction
from kotti.resources import Document
document = Document(title=u'mytitle', description=u'mydescription',
body=u'<span>mybody</span>')
document.request = dummy_request
root[u'mydoc'] = document
# with body=None no exceptions should be raised (utils/html_to_text)
transaction.commit()
es_client.flush()
assert len(es_client.es.search(q='mytitle')['hits']['hits']) == 1
def test_insert(self, es_client, root, dummy_request, db_session):
import transaction
from kotti.resources import Document
document = Document(title=u'mydoc', description=u'mydescription',)
document.request = dummy_request
root[u'mydoc'] = document
# with body=None no exceptions should be raised (utils/html_to_text)
transaction.commit()
es_client.flush()
results = es_client.es.search(q='mydescription')
_type = results['hits']['hits'][0]['_type']
assert _type == 'Document'
from kotti.resources import get_root
root = get_root()
assert results['hits']['hits'][0]['_source']['path'] == root[u'mydoc'].path
def test_after_commit_delete(self, dummy_request):
from kotti.resources import Document
from kotti_es.sqla import _after_delete
from kotti_es.sqla import _after_commit
document = Document()
document.request = dummy_request
_after_delete(None, None, document)
import mock
with mock.patch('kotti_es.sqla.get_current_request') as \
get_current_request:
get_current_request.return_value = dummy_request
with mock.patch('kotti_es.sqla.get_client') as \
get_client:
magic = mock.MagicMock()
get_client.return_value = magic
session = mock.Mock()
session.configure_mock(**{
'query.return_value.filter_by.'
'return_value.one.return_value': document})
_after_commit(session)
magic.delete_object.assert_called_with(document,
immediate=True)
