class TestReportEntriesModel(unittest.TestCase):
def setUp(self):
self.model = ReportEntriesModel()
def test_to_json_empty_with_title_and_date(self):
self.model.set_title("Demo")
self.model.set_report_date("Demo date")
json = self.model.json()
self.assertEqual(json, {'title': "Demo", 'report_date': "Demo date"})
def test_to_json_empty_with_defailt_title_and_date(self):
json = self.model.json()
self.assertEqual(json, {'title': "No title", 'report_date': "No date"})
def test_to_json_with_thead_and_blog(self):
thread = ThreadModel({'title':"Demo T"})
blog = BlogEntry({'title': "Demo B"})
self.model.add_entry("1", thread)
self.model.add_entry("2", blog)
json = self.model.json()
#self.assertEqual(json, {'title': "", 'report_date': ""})
self.assertEqual(json['1'], [{'title': "Demo T"}])
self.assertEqual(json['2'], [{'title': "Demo B"}])
def transtale_report_into_threads(report_query, report):
new_report = ReportEntriesModel()
for kword in report_query['keywords']:
for thread in report[kword]:
new_report.add_entry(kword, ThreadModel(thread))
return new_report
