def _set_up_sms_es(self):
sms_doc = {
'_id': 'some_sms_id',
'domain': self.domain.name,
'direction': INCOMING,
'date': json_format_datetime(datetime.datetime.utcnow()),
'doc_type': SMS_INDEX_INFO.type,
}
send_to_elasticsearch("sms", sms_doc)
refresh_elasticsearch_index('sms')
def create_sms_in_es(domain_name, direction):
sms_doc = {
'_id': 'some_sms_id',
'domain': domain_name,
'direction': direction,
'date': json_format_datetime(datetime.datetime.utcnow()),
'doc_type': SMS_INDEX_INFO.type,
}
send_to_elasticsearch("sms", sms_doc)
refresh_elasticsearch_index('sms')
return sms_doc
def check_elasticsearch():
cluster_health = check_es_cluster_health()
if cluster_health != 'green':
return ServiceStatus(False, "Cluster health at %s" % cluster_health)
doc = {'_id': 'elasticsearch-service-check',
'date': datetime.datetime.now().isoformat()}
send_to_elasticsearch('groups', doc)
refresh_elasticsearch_index('groups')
hits = GroupES().remove_default_filters().doc_id(doc['_id']).run().hits
send_to_elasticsearch('groups', doc, delete=True) # clean up
if doc in hits:
return ServiceStatus(True, "Successfully sent a doc to ES and read it back")
return ServiceStatus(False, "Something went wrong sending a doc to ES")
def check_elasticsearch():
cluster_health = check_es_cluster_health()
if cluster_health != 'green':
return ServiceStatus(False, "Cluster health at %s" % cluster_health)
doc = {'_id': 'elasticsearch-service-check-{}'.format(random_hex()[:7]),
'date': datetime.datetime.now().isoformat()}
try:
send_to_elasticsearch('groups', doc)
refresh_elasticsearch_index('groups')
hits = GroupES().remove_default_filters().doc_id(doc['_id']).run().hits
if doc in hits:
return ServiceStatus(True, "Successfully sent a doc to ES and read it back")
else:
return ServiceStatus(False, "Something went wrong sending a doc to ES")
finally:
send_to_elasticsearch('groups', doc, delete=True) # clean up
def _set_up_sms_es(self):
sms_doc = {
'_id': 'some_sms_id',
'domain': self.domain.name,
'direction': INCOMING,
'date': json_format_datetime(datetime.datetime.utcnow()),
'doc_type': SMS_INDEX_INFO.type,
}
send_to_elasticsearch(
alias=SMS_INDEX_INFO.alias,
doc_type=SMS_INDEX_INFO.type,
doc_id=sms_doc['_id'],
es_getter=get_es_new,
name='ElasticProcessor',
data=sms_doc,
update=False,
)
refresh_elasticsearch_index('sms')
def delete_sms_in_es(sms_doc):
send_to_elasticsearch("sms", sms_doc, delete=True)
refresh_elasticsearch_index('sms')
def test_deleting_location_updates_user(self):
self.user.reset_locations(self.loc_ids)
refresh_elasticsearch_index('users')
self.loc1.sql_location.full_delete()
self.loc2.sql_location.full_delete()
self.assertAssignedLocations([])
