def test_multi_field(self):
c = CRECDoc(title='foo',
content='bar',
date_issued=datetime(2017, 1, 1))
c2 = CRECDoc(title='foo',
content='baz',
date_issued=datetime(2016, 1, 1))
c.save(refresh=True)
c2.save(refresh=True)
start_date = datetime(2017, 1, 1)
end_date = datetime(2017, 1, 30)
query_args = {
'start_date': start_date.strftime('%Y-%m-%d'),
'end_date': end_date.strftime('%Y-%m-%d'),
'content': 'bar',
'title': 'foo',
}
response = self.client.get('/cwapi/search/', query_args)
response_content = response.json()
results = response_content['data']
self.assertEquals(1, len(results))
def test_date_filter(self):
c = CRECDoc(title='should be in results',
content='blah',
date_issued=datetime(2017, 1, 1))
c2 = CRECDoc(title='should NOT be in results',
content='blah',
date_issued=datetime(2016, 1, 1))
c.save(refresh=True)
c2.save(refresh=True)
start_date = datetime(2017, 1, 1)
end_date = datetime(2017, 1, 30)
query_args = {
'start_date': start_date.strftime('%Y-%m-%d'),
'end_date': end_date.strftime('%Y-%m-%d'),
'content': 'blah',
}
response = self.client.get('/cwapi/search/', query_args)
response_content = response.json()
results = response_content['data']
self.assertEquals(1, len(results))
self.assertEquals('should be in results', results[0]['title'])
def setUp(self):
self.es_conn = connections.get_connection()
self.test_crecs = []
for i in range(20):
self.test_crecs.append(
CRECDoc(title=str(i),
content='foo bar baz Foo',
date_issued=datetime(2017, 1, i % 5 + 1)))
self.index = Index(settings.ES_CW_INDEX)
CRECDoc.init()
for c in self.test_crecs:
c.save(refresh=True)
self.client = Client()
def to_es_doc(self):
"""Returns the CRECParser as a dict ready to be uploaded to
elasticsearch.
Returns:
dict: A dict representation of this document.
"""
return CRECDoc(
title=self.title,
title_part=self.title_part,
date_issued=self.date_issued,
content=self.content,
crec_id=self.id,
pdf_url=self.pdf_url,
html_url=self.html_url,
page_start=self.page_start,
page_end=self.page_end,
speakers=','.join(self.speakers),
segments=self.segments,
)