def test_kafka_domain_pillow(self):
# make a domain
domain_name = 'domain-pillowtest-kafka'
with drop_connected_signals(commcare_domain_post_save):
domain = create_domain(domain_name)
# send to kafka
since = get_topic_offset(topics.DOMAIN)
publish_domain_saved(domain)
# send to elasticsearch
pillow = get_domain_kafka_to_elasticsearch_pillow()
pillow.process_changes(since=since, forever=False)
self.elasticsearch.indices.refresh(self.index_info.index)
# verify there
self._verify_domain_in_es(domain_name)
def test_kafka_domain_pillow_deletions(self):
# run the other test to ensure domain is created and in ES
self.test_kafka_domain_pillow()
domain_obj = Domain.get_by_name('domain-pillowtest-kafka')
domain_obj.doc_type = 'Domain-DUPLICATE'
# send to kafka
since = get_topic_offset(topics.DOMAIN)
publish_domain_saved(domain_obj)
# send to elasticsearch
pillow = get_domain_kafka_to_elasticsearch_pillow()
pillow.process_changes(since=since)
self.elasticsearch.indices.refresh(self.index_info.index)
# ensure removed from ES
self.assertEqual(0, DomainES().run().total)
def test_reverted_domain_pillow_deletion(self):
domain_name = 'domain-pillow-delete'
with drop_connected_signals(commcare_domain_post_save):
domain = create_domain(domain_name)
# send to kafka
since = get_topic_offset(topics.DOMAIN)
publish_domain_saved(domain)
# send to elasticsearch
pillow = get_domain_kafka_to_elasticsearch_pillow()
pillow.process_changes(since=since, forever=False)
self.elasticsearch.indices.refresh(self.index_info.index)
# verify there
self._verify_domain_in_es(domain_name)
domain_obj = Domain.get_by_name(domain_name)
domain_obj.doc_type = 'Domain-DUPLICATE'
domain_obj.save()
# send to kafka
since = get_topic_offset(topics.DOMAIN)
publish_domain_saved(domain_obj)
# undelete
domain_obj = Domain.get_by_name(domain_name)
domain_obj.doc_type = 'Domain'
domain_obj.save()
# process pillow changes
pillow = get_domain_kafka_to_elasticsearch_pillow()
pillow.process_changes(since=since, forever=False)
self.elasticsearch.indices.refresh(self.index_info.index)
# confirm domain still exists
self._verify_domain_in_es(domain_name)