def test_documents_add_to_register(self):
registry = DocumentRegistry()
with patch('django_elasticsearch_dsl.indices.registry', new=registry):
index = Index('test')
index.doc_type(self.DocA1)
docs = list(registry.get_documents())
self.assertEqual(len(docs), 1)
self.assertIsInstance(docs[0], self.DocA1)
index.doc_type(self.DocA2)
docs = list(registry.get_documents())
self.assertEqual(len(docs), 2)
self.assertIsInstance(docs[0], self.DocA1)
self.assertIsInstance(docs[1], self.DocA2)
class DocumentRegistryTestCase(TestCase):
class ModelA():
pass
class ModelB():
pass
class ModelC():
pass
def setUp(self):
self.registry = DocumentRegistry()
self.index_1 = Mock()
self.index_2 = Mock()
self.doc_a1 = Mock()
self.doc_a1._doc_type.model = self.ModelA
self.doc_a1._doc_type.ignore_signals = False
self.doc_a2 = Mock()
self.doc_a2._doc_type.model = self.ModelA
self.doc_a2._doc_type.ignore_signals = False
self.doc_b1 = Mock()
self.doc_b1._doc_type.model = self.ModelB
self.doc_b1._doc_type.ignore_signals = False
self.doc_c1 = Mock()
self.doc_c1._doc_type.model = self.ModelC
self.doc_c1._doc_type.ignore_signals = False
self.registry.register(self.index_1, self.doc_a1)
self.registry.register(self.index_1, self.doc_a2)
self.registry.register(self.index_2, self.doc_b1)
self.registry.register(self.index_1, self.doc_c1)
def test_empty_registry(self):
registry = DocumentRegistry()
self.assertEqual(registry._indices, {})
self.assertEqual(registry._models, {})
def test_register(self):
self.assertEqual(self.registry._models[self.ModelA],
set([self.doc_a1, self.doc_a2]))
self.assertEqual(self.registry._models[self.ModelB],
set([self.doc_b1]))
self.assertEqual(self.registry._indices[self.index_1],
set([self.doc_a1, self.doc_a2, self.doc_c1]))
self.assertEqual(self.registry._indices[self.index_2],
set([self.doc_b1]))
def test_get_models(self):
self.assertEqual(self.registry.get_models(),
set([self.ModelA, self.ModelB, self.ModelC]))
def test_get_documents(self):
self.assertEqual(
self.registry.get_documents(),
set([self.doc_a1, self.doc_a2, self.doc_b1, self.doc_c1]))
def test_get_documents_by_model(self):
self.assertEqual(self.registry.get_documents([self.ModelA]),
set([self.doc_a1, self.doc_a2]))
def test_get_documents_by_unregister_model(self):
ModelC = Mock()
self.assertFalse(self.registry.get_documents([ModelC]))
def test_get_indices(self):
self.assertEqual(self.registry.get_indices(),
set([self.index_1, self.index_2]))
def test_get_indices_by_model(self):
self.assertEqual(self.registry.get_indices([self.ModelA]),
set([self.index_1]))
def test_get_indices_by_unregister_model(self):
ModelC = Mock()
self.assertFalse(self.registry.get_indices([ModelC]))
def test_update_instance(self):
doc_a3 = Mock()
doc_a3._doc_type.model = self.ModelA
doc_a3._doc_type.ignore_signals = True
self.registry.register(self.index_1, doc_a3)
instance = self.ModelA()
self.registry.update(instance)
self.assertFalse(doc_a3.update.called)
self.assertFalse(self.doc_b1.update.called)
self.doc_a1.update.assert_called_once_with(instance)
self.doc_a2.update.assert_called_once_with(instance)
def test_delete_instance(self):
doc_a3 = Mock()
doc_a3._doc_type.model = self.ModelA
doc_a3._doc_type.ignore_signals = True
self.registry.register(self.index_1, doc_a3)
instance = self.ModelA()
self.registry.delete(instance)
self.assertFalse(doc_a3.update.called)
self.assertFalse(self.doc_b1.update.called)
self.doc_a1.update.assert_called_once_with(instance, action='delete')
self.doc_a2.update.assert_called_once_with(instance, action='delete')
def test_autosync(self):
settings.ELASTICSEARCH_DSL_AUTOSYNC = False
instance = self.ModelA()
self.registry.update(instance)
self.assertFalse(self.doc_a1.update.called)
settings.ELASTICSEARCH_DSL_AUTOSYNC = True
class DocumentRegistryTestCase(WithFixturesMixin, TestCase):
def setUp(self):
self.registry = DocumentRegistry()
self.index_1 = Index(name='index_1')
self.index_2 = Index(name='index_2')
self.doc_a1 = self._generate_doc_mock(self.ModelA, self.index_1)
self.doc_a2 = self._generate_doc_mock(self.ModelA, self.index_1)
self.doc_b1 = self._generate_doc_mock(self.ModelB, self.index_2)
self.doc_c1 = self._generate_doc_mock(self.ModelC, self.index_1)
def test_empty_registry(self):
registry = DocumentRegistry()
self.assertEqual(registry._indices, {})
self.assertEqual(registry._models, {})
def test_register(self):
self.assertEqual(self.registry._models[self.ModelA],
set([self.doc_a1, self.doc_a2]))
self.assertEqual(self.registry._models[self.ModelB],
set([self.doc_b1]))
self.assertEqual(self.registry._indices[self.index_1],
set([self.doc_a1, self.doc_a2, self.doc_c1]))
self.assertEqual(self.registry._indices[self.index_2],
set([self.doc_b1]))
def test_get_models(self):
self.assertEqual(self.registry.get_models(),
set([self.ModelA, self.ModelB, self.ModelC]))
def test_get_documents(self):
self.assertEqual(self.registry.get_documents(),
set([self.doc_a1, self.doc_a2,
self.doc_b1, self.doc_c1]))
def test_get_documents_by_model(self):
self.assertEqual(self.registry.get_documents([self.ModelA]),
set([self.doc_a1, self.doc_a2]))
def test_get_documents_by_unregister_model(self):
ModelC = Mock()
self.assertFalse(self.registry.get_documents([ModelC]))
def test_get_indices(self):
self.assertEqual(self.registry.get_indices(),
set([self.index_1, self.index_2]))
def test_get_indices_by_model(self):
self.assertEqual(self.registry.get_indices([self.ModelA]),
set([self.index_1]))
def test_get_indices_by_unregister_model(self):
ModelC = Mock()
self.assertFalse(self.registry.get_indices([ModelC]))
def test_update_instance(self):
doc_a3 = self._generate_doc_mock(
self.ModelA, self.index_1, _ignore_signals=True
)
instance = self.ModelA()
self.registry.update(instance)
self.assertFalse(doc_a3.update.called)
self.assertFalse(self.doc_b1.update.called)
self.doc_a1.update.assert_called_once_with(instance)
self.doc_a2.update.assert_called_once_with(instance)
def test_update_related_instances(self):
doc_d1 = self._generate_doc_mock(
self.ModelD, self.index_1,
_related_models=[self.ModelE, self.ModelB]
)
doc_d2 = self._generate_doc_mock(
self.ModelD, self.index_1, _related_models=[self.ModelE]
)
instance_e = self.ModelE()
instance_b = self.ModelB()
related_instance = self.ModelD()
doc_d2.get_instances_from_related.return_value = related_instance
doc_d1.get_instances_from_related.return_value = related_instance
self.registry.update_related(instance_e)
doc_d1.get_instances_from_related.assert_called_once_with(instance_e)
doc_d1.update.assert_called_once_with(related_instance)
doc_d2.get_instances_from_related.assert_called_once_with(instance_e)
doc_d2.update.assert_called_once_with(related_instance)
doc_d1.get_instances_from_related.reset_mock()
doc_d1.update.reset_mock()
doc_d2.get_instances_from_related.reset_mock()
doc_d2.update.reset_mock()
self.registry.update_related(instance_b)
doc_d1.get_instances_from_related.assert_called_once_with(instance_b)
doc_d1.update.assert_called_once_with(related_instance)
doc_d2.get_instances_from_related.assert_not_called()
doc_d2.update.assert_not_called()
def test_update_related_instances_not_defined(self):
doc_d1 = self._generate_doc_mock(_model=self.ModelD, index=self.index_1,
_related_models=[self.ModelE])
instance = self.ModelE()
doc_d1.get_instances_from_related.return_value = None
self.registry.update_related(instance)
doc_d1.get_instances_from_related.assert_called_once_with(instance)
doc_d1.update.assert_not_called()
def test_delete_instance(self):
doc_a3 = self._generate_doc_mock(
self.ModelA, self.index_1, _ignore_signals=True
)
instance = self.ModelA()
self.registry.delete(instance)
self.assertFalse(doc_a3.update.called)
self.assertFalse(self.doc_b1.update.called)
self.doc_a1.update.assert_called_once_with(instance, action='delete')
self.doc_a2.update.assert_called_once_with(instance, action='delete')
def test_autosync(self):
settings.ELASTICSEARCH_DSL_AUTOSYNC = False
instance = self.ModelA()
self.registry.update(instance)
self.assertFalse(self.doc_a1.update.called)
settings.ELASTICSEARCH_DSL_AUTOSYNC = True
class DocumentRegistryTestCase(TestCase):
class ModelA():
pass
class ModelB():
pass
class ModelC():
pass
def _generate_doc_mock(self,
model,
index=None,
ignore_signals=False,
related_models=None):
doc = Mock()
doc._doc_type.model = model
doc._doc_type.ignore_signals = ignore_signals
doc._doc_type.related_models = related_models if (
related_models) is not None else []
if index:
self.registry.register(index, doc)
return doc
def setUp(self):
self.registry = DocumentRegistry()
self.index_1 = Mock()
self.index_2 = Mock()
self.doc_a1 = self._generate_doc_mock(self.ModelA, self.index_1)
self.doc_a2 = self._generate_doc_mock(self.ModelA, self.index_1)
self.doc_b1 = self._generate_doc_mock(self.ModelB, self.index_2)
self.doc_c1 = self._generate_doc_mock(self.ModelC, self.index_1)
def test_empty_registry(self):
registry = DocumentRegistry()
self.assertEqual(registry._indices, {})
self.assertEqual(registry._models, {})
def test_register(self):
self.assertEqual(self.registry._models[self.ModelA],
set([self.doc_a1, self.doc_a2]))
self.assertEqual(self.registry._models[self.ModelB],
set([self.doc_b1]))
self.assertEqual(self.registry._indices[self.index_1],
set([self.doc_a1, self.doc_a2, self.doc_c1]))
self.assertEqual(self.registry._indices[self.index_2],
set([self.doc_b1]))
def test_get_models(self):
self.assertEqual(self.registry.get_models(),
set([self.ModelA, self.ModelB, self.ModelC]))
def test_get_documents(self):
self.assertEqual(
self.registry.get_documents(),
set([self.doc_a1, self.doc_a2, self.doc_b1, self.doc_c1]))
def test_get_documents_by_model(self):
self.assertEqual(self.registry.get_documents([self.ModelA]),
set([self.doc_a1, self.doc_a2]))
def test_get_documents_by_unregister_model(self):
ModelC = Mock()
self.assertFalse(self.registry.get_documents([ModelC]))
def test_get_indices(self):
self.assertEqual(self.registry.get_indices(),
set([self.index_1, self.index_2]))
def test_get_indices_by_model(self):
self.assertEqual(self.registry.get_indices([self.ModelA]),
set([self.index_1]))
def test_get_indices_by_unregister_model(self):
ModelC = Mock()
self.assertFalse(self.registry.get_indices([ModelC]))
def test_update_instance(self):
doc_a3 = self._generate_doc_mock(self.ModelA,
self.index_1,
ignore_signals=True)
instance = self.ModelA()
self.registry.update(instance)
self.assertFalse(doc_a3.update.called)
self.assertFalse(self.doc_b1.update.called)
self.doc_a1.update.assert_called_once_with(instance)
self.doc_a2.update.assert_called_once_with(instance)
def test_update_related_instances(self):
class ModelD():
pass
class ModelE():
pass
doc_d1 = self._generate_doc_mock(ModelD,
self.index_1,
related_models=[ModelE])
instance = ModelE()
related_instance = ModelD()
doc_d1.get_instances_from_related.return_value = related_instance
self.registry.update(instance)
doc_d1.get_instances_from_related.assert_called_once_with(instance)
doc_d1.update.assert_called_once_with(related_instance)
def test_update_related_isntances_not_defined(self):
class ModelD():
pass
class ModelE():
pass
doc_d1 = self._generate_doc_mock(ModelD,
self.index_1,
related_models=[ModelE])
instance = ModelE()
doc_d1.get_instances_from_related.return_value = None
self.registry.update(instance)
doc_d1.get_instances_from_related.assert_called_once_with(instance)
doc_d1.update.assert_not_called()
def test_delete_instance(self):
doc_a3 = self._generate_doc_mock(self.ModelA,
self.index_1,
ignore_signals=True)
instance = self.ModelA()
self.registry.delete(instance)
self.assertFalse(doc_a3.update.called)
self.assertFalse(self.doc_b1.update.called)
self.doc_a1.update.assert_called_once_with(instance, action='delete')
self.doc_a2.update.assert_called_once_with(instance, action='delete')
def test_autosync(self):
settings.ELASTICSEARCH_DSL_AUTOSYNC = False
instance = self.ModelA()
self.registry.update(instance)
self.assertFalse(self.doc_a1.update.called)
settings.ELASTICSEARCH_DSL_AUTOSYNC = True