def test_build_csv_custom_url(self, csv_mock, tmp_file_mock, os_mock): tmp_file_mock.return_value = (0, self.filename) writer_mock = Mock() csv_mock.writer.return_value = writer_mock url = 'http://www.google.com' user = facebook.client.get_user(1, 1) edge_collection = { self.notification_user.uuid: facebook.client.get_friend_edges(user, 1) } faces_email.build_csv( edge_collection.iteritems(), 3, self.campaign, self.content, url ) assert writer_mock.writerow.called assert 'http://www.google.com?efuuid=1' in writer_mock.writerow.call_args[0][0][4] self.assertEqual( writer_mock.writerow.call_args[0][0][1], '*****@*****.**' ) self.assertEqual( relational.NotificationEvent.objects.filter( event_type='shown').count(), 3 ) assert relational.NotificationEvent.objects.filter( event_type='generated').exists()
def test_build_csv(self, csv_mock, tmp_file_mock, os_mock): ''' Tests the build_csv method ''' tmp_file_mock.return_value = (0, self.filename) writer_mock = Mock() csv_mock.writer.return_value = writer_mock user = facebook.client.get_user(1, 1) edge_collection = { self.notification_user.uuid: facebook.client.get_friend_edges(user, 1) } faces_email.build_csv( edge_collection.iteritems(), 3, self.campaign, self.content, None ) assert writer_mock.writerow.called assert writer_mock.writerow.call_args[0][0][4].strip().startswith('<table') self.assertEqual( writer_mock.writerow.call_args[0][0][1], '*****@*****.**' ) self.assertEqual( relational.NotificationEvent.objects.filter( event_type='shown').count(), 3 ) assert relational.NotificationEvent.objects.filter( event_type='generated').exists()