from django.conf import settings from django.test import TestCase from picampaign.campaign.models import Campaign, Argumentary, CampaignContact from picampaign.contact.models import Contact class CampaignMethodTests(TestCase): def test_unicode_is_title(self): campaign = Campaign( title='Campaign Title', start_date='2000-01-01', end_date='2100-12-31' ) self.assertEqual(campaign.__unicode__(), 'Campaign Title') def test_unicode_is_unicode(self): campaign = Campaign( title='Campaign Title', start_date='2000-01-01', end_date='2100-12-31' ) self.assertEqual( str(type(campaign.__unicode__())), "" ) class ArgumentaryMethodTests(TestCase): def test_unicode_is_title(self): campaign = Campaign( title='Campaign Title', start_date='2000-01-01', end_date='2100-12-31' ) argumentary = Argumentary( campaign=campaign, lang='en' ) self.assertEqual( argumentary.__unicode__(), "Argumentary in en for Campaign Title" ) def test_unicode_is_unicode(self): campaign = Campaign( title='Campaign Title', start_date='2000-01-01', end_date='2100-12-31' ) argumentary = Argumentary( campaign=campaign, lang='en' ) self.assertEqual( str(type(argumentary.__unicode__())), "" ) class CampaignContactMethodTests(TestCase): def test_unicode_is_contact_and_title(self): campaign = Campaign( title='Campaign Title', start_date='2000-01-01', end_date='2100-12-31' ) contact = Contact( first_name='Victor', last_name='Hugo', phone='+3312345678' ) campaigncontact = CampaignContact( campaign=campaign, contact=contact ) self.assertEqual( campaigncontact.__unicode__(), "Contact Victor Hugo on Campaign Title" ) def test_unicode_is_unicode(self): campaign = Campaign( title='Campaign Title', start_date='2000-01-01', end_date='2100-12-31' ) contact = Contact( first_name='Victor', last_name='Hugo', phone='+3312345678' ) campaigncontact = CampaignContact( campaign=campaign, contact=contact ) self.assertEqual( str(campaigncontact), "Contact Contact object on Campaign Title" ) def test_all_groups_is_empty(self): campaign = Campaign( title='Campaign Title', start_date='2000-01-01', end_date='2100-12-31' ) contact = Contact( first_name='Victor', last_name='Hugo', phone='+3312345678' ) campaigncontact = CampaignContact( campaign=campaign, contact=contact ) self.assertEqual(campaigncontact.all_groups(),[])