def test_create_error(self): new_user = User(email='*****@*****.**', password='******') new_user.save() biz_card = BusinessCard(user_id=new_user) biz_card.save() with self.assertRaises(ValueError): new_contact = Contacts(businesscard_id=biz_card, user_id=new_user.id) new_contact.save()
class TestContactMedia(TestCase): def setUp(self): self.user = User( email="*****@*****.**", password="******" ) self.user.save() self.biz_card = BusinessCard(user_id=self.user) self.biz_card.save() self.contact = Contacts(businesscard_id=self.biz_card, user_id=self.user) self.contact.save()
class TestContactModel(TestCase): def setUp(self): self.user = User( email="*****@*****.**", password="******" ) self.user.save() self.biz_card = BusinessCard(user_id=self.user) self.biz_card.save() self.contact = Contacts(businesscard_id=self.biz_card, user_id=self.user) self.contact.save() def test_user_create(self): self.assertIsInstance(self.user, User) def test_contact_create(self): self.assertIsInstance(self.biz_card, BusinessCard) contact = Contacts(businesscard_id=self.biz_card) # contact.save() self.assertIsInstance(contact, Contacts) def test_contact_media_create(self): contact_media = ContactMedia(user_id=self.user, contact_id=self.contact, img_url=create_image_file() ) contact_media.save() self.assertIsInstance(contact_media, ContactMedia) self.assertRegexpMatches( contact_media.img_url.url, re.compile(r'.*bcards_gallery/*test.*\.png')) def test_contact_profile_create(self): request = HttpRequest() request.user = self.user contact_view_set = storeContactsViewSet() request.data = {} request.data['contact_profile_image'] = create_image_file() request.data['contact_id'] = self.contact.id request.data['user_id'] = self.contact.id response = contact_view_set.upload_contact_profile(request) self.assertEqual(response.status_code, 200) def test_invalidcontact_profile_create(self): request = HttpRequest() request.user = self.user contact_view_set = storeContactsViewSet() request.data = {} request.data['contact_profile_image'] = create_image_file() request.data['contact_id'] = 5000 response = contact_view_set.upload_contact_profile(request) self.assertEqual(response.status_code, 400, "Invalid Contact id") def test_create_error(self): new_user = User(email='*****@*****.**', password='******') new_user.save() biz_card = BusinessCard(user_id=new_user) biz_card.save() with self.assertRaises(ValueError): new_contact = Contacts(businesscard_id=biz_card, user_id=new_user.id) new_contact.save() def tearDown(self): self.user.delete() self.biz_card.delete() self.contact.delete()