def setUp(self):
self.mock = MockMongo()
self.builder = ReportBuilder(self.mock)
self.report = {'name': "TestReport", 'keywords': ['key', 'word']}
self.thread = {'title': (self.report['keywords'][0]), 'msgs': []}.copy()
self.keyword = self.report['keywords'][0]
self.builder.report_request = self.report
class TestReportBuilder(unittest.TestCase):
def setUp(self):
self.mock = MockMongo()
self.builder = ReportBuilder(self.mock)
self.report = {'name': "TestReport", 'keywords': ['key', 'word']}
self.thread = {'title': (self.report['keywords'][0]), 'msgs': []}.copy()
self.keyword = self.report['keywords'][0]
self.builder.report_request = self.report
def assertLen(self, count, elements):
self.assertEquals(count, len(elements))
def test_when_report_has_no_keywords_launch_exception(self):
emptyReport = dict()
self.assertRaises(ValueError, self.builder.build_report, emptyReport)
def test_when_report_has_keywords_calls_hreads(self):
self.builder.build_report(self.report)
self.assertEquals(self.mock.threadscalled, 1)
def test_report_has_title(self):
self.builder._create_report()
result = self.builder.report
self.assertEqual(result['title'], "Result for report TestReport")
def test_when_add_first_therad_first_create_report(self):
self.builder._add_thead_to_report("", dict())
result = self.builder.report
self.assertEqual(result['title'], "Result for report TestReport")
def test_when_add_first_therad_report_contains_the_keyword(self):
self.builder._add_thead_to_report('key', self.thread)
result = self.builder.report
self.assertIn(self.keyword, result)
self.assertLen(1, result[self.keyword])
def test_when_add_second_therad_report_contains_both(self):
self.builder._add_thead_to_report(self.keyword, self.thread)
self.builder._add_thead_to_report(self.keyword, self.thread)
result = self.builder.report
self.assertLen(2, result[self.keyword])
def test_when_keyword_is_foun_in_therad_titles_then_msgs_are_not_added(self):
self.builder._find_keywords(self.thread, ['key'])
keyline = self.builder.report[self.keyword][0]
self.assertNotIn('msgs', keyline)
def test_words_are_searched_in_title_ignoring_case(self):
self.thread['title']= self.keyword.lower()
self.builder._add_thead_to_report(self.keyword.upper(), self.thread)
result = self.builder.report
self.assertIn(self.keyword.upper(), result)
def test_word_in_msgs_when_msg_coitains_keyword(self):
self.thread['msgs'].append({'body': self.keyword})
result = self.builder._word_in_msgs(self.keyword, self.thread)
self.assertLen(1, result)
def test_word_in_msgs_when_msg_not_coitains_keyword(self):
self.thread['msgs'].append({'body': "No word"})
result = self.builder._word_in_msgs(self.keyword, self.thread)
self.assertLen(0, result)
def test_word_in_msgs_when_msg_is_empty(self):
result = self.builder._word_in_msgs(self.keyword, self.thread)
self.assertLen(0, result)
def test_word_in_with_zacatrus(self):
word = PreGeneratedReports.tienda_zacatrus['keywords'][0]
msg_title = u'Clash of Cultures a 33,75\x80 en Zacatrus (ZMAN)'
self.assertTrue(self.builder._word_in(word, msg_title))
def test_add_msgs_to_report(self):
msg = {'body': "Message"}
msglist = [msg]
self.builder._add_msgs_to_report(self.keyword, self.thread, msglist)
result = self.builder.report
self.assertLen(1, result[self.keyword])
thread = result[self.keyword][0]
self.assertIn('msgs', thread)
msgs = thread['msgs']
self.assertEquals(msgs[0]['body'], msg['body'])
def test_report_includes_creation_date_of_thread(self):
self.thread['msgs'] = Reports.asylum_msg_list
self.mock.saveThread(self.thread)
self.assertLen(1, self.mock.threads())
result = self.builder.build_report(self.report)
# print result
self.assertLen(1, result[self.keyword])
first_thread = result[self.keyword][0]
self.assertEquals(first_thread['creation_date'], Reports.asylum_msg_list[0]['date'])
def test_add_creation_date_two_messages(self):
self.thread['msgs'].append({'date': 'yes'})
self.thread['msgs'].append({'date': 'no'})
self.builder._add_creation_and_last_msg_date(self.thread)
self.assertIn('creation_date', self.thread)
self.assertEquals('yes', self.thread['creation_date'])
def test_when_thread_already_has_creation_date_then_exit(self):
self.thread['creation_date'] = "Already exists"
self.builder._add_creation_and_last_msg_date(self.thread)
self.assertEqual("Already exists", self.thread['creation_date'])
def test_report_includes_date_of_last_msg(self):
self.thread['msgs'] = Reports.asylum_msg_list
self.mock.saveThread(self.thread)
self.assertLen(1, self.mock.threads())
result = self.builder.build_report(self.report)
# print result
self.assertLen(1, result[self.keyword])
first_thread = result[self.keyword][0]
self.assertEquals(first_thread['last_msg_date'], Reports.asylum_msg_list[0]['date'])
def test_msg_has_a_link(self):
self.thread['msgs'] = [{'id': 'xxx'}]
self.thread['link'] = 'yyy'
self.builder._add_link(self.thread, self.thread['msgs'])
msg = self.thread['msgs'][0]
self.assertIn('link', msg)
self.assertEquals(msg['link'], 'yyy#xxx')
def test_when_msg_has_no_id_link_is_the_same_than_the_thread(self):
self.thread['msgs'] = [{}]
self.thread['link'] = 'yyy'
self.builder._add_link(self.thread, self.thread['msgs'])
msg = self.thread['msgs'][0]
self.assertNotIn('link', msg)
def test_hilo_con_mas_de_una_keyword(self):
thread = {u'title': u'Guardian cross para Android/Iphone/Ipad', u'answers': 1, 'creation_date': u' 03 de Junio de 2013, 10:55:11 am \xbb', u'source': u'LaBSK', u'link': u'http://labsk.net/index.php?topic=111335.0', u'location': u'Videojuegos'}
#threadobj = ThreadModel(thread)
self.builder._find_keywords(thread, ["iphone", "ios", "android", "tablet"])
report = self.builder.report
self.assertEquals(1, len(report["iphone"]))
self.assertEquals(1, len(report["android"]))
