def setUp(self): self.user = User.objects.get(username='******') # Phonebook model self.phonebook = Phonebook( name='test_phonebook', user=self.user, ) self.phonebook.save() # Contact model self.contact = Contact( phonebook=self.phonebook, contact=123456789, first_name='Tom', last_name='Gun', ) self.contact.save() self.assertEqual(self.contact.__unicode__(), u'123456789 (Gun)') self.assertEqual(self.contact.contact_name(), 'Tom Gun') self.assertTrue(get_url_campaign_status(1, 1)) self.assertTrue(get_url_campaign_status(1, 2)) self.assertTrue(get_url_campaign_status(1, 3)) self.assertTrue(get_url_campaign_status(1, 4))
class DialerContactModel(TestCase): """Test Phonebook, Contact models""" fixtures = ['auth_user.json', 'phonebook.json', 'contact.json'] def setUp(self): self.user = User.objects.get(username='******') # Phonebook model self.phonebook = Phonebook( name='test_phonebook', user=self.user, ) self.phonebook.save() # Contact model self.contact = Contact( phonebook=self.phonebook, contact=123456789, first_name='Tom', last_name='Gun', ) self.contact.save() self.assertEqual(self.contact.__unicode__(), u'123456789 (Gun)') self.assertEqual(self.contact.contact_name(), 'Tom Gun') self.assertTrue(get_url_campaign_status(1, 1)) self.assertTrue(get_url_campaign_status(1, 2)) self.assertTrue(get_url_campaign_status(1, 3)) self.assertTrue(get_url_campaign_status(1, 4)) def test_phonebook_form(self): self.assertEqual(self.phonebook.name, 'test_phonebook') form = PhonebookForm({'name': 'sample_phonebook'}) obj = form.save(commit=False) obj.user = self.user obj.save() form = PhonebookForm(instance=self.phonebook) self.assertTrue(isinstance(form.instance, Phonebook)) def test_contact_form(self): self.assertEqual(self.contact.phonebook, self.phonebook) form = ContactForm(self.user) form.contact = '123456' obj = form.save(commit=False) obj.phonebook = self.phonebook obj.save() form = ContactForm(self.user, instance=self.contact) self.assertTrue(isinstance(form.instance, Contact)) def teardown(self): self.phonebook.delete() self.contact.delete()
def setUp(self): self.user = User.objects.get(username='******') # Phonebook model self.phonebook = Phonebook( name='test_phonebook', user=self.user, ) self.phonebook.save() # Contact model self.contact = Contact( phonebook=self.phonebook, contact=123456789, first_name='Tom', last_name='Gun', ) self.contact.save() self.assertEqual(self.contact.__unicode__(), u'123456789') self.assertEqual(self.contact.contact_name(), 'Tom Gun')