def get_or_create_country(name, user_created=False): try: return Country.objects.get(name=name) except Country.DoesNotExist: new = Country(name=name, user_created=user_created) new.save() return new
def get_country(name): try: c = Country.objects.get(name=name) except Country.DoesNotExist: c = Country(name=name) c.save() return c
def get_country(name): try: c = Country.objects.get(name = name) except Country.DoesNotExist: c = Country(name = name) c.save() return c
def setUp(self): self.user = User.objects.create(username='******', email='*****@*****.**') self.user.save() self.ndt = NDT.objects.create(download_rate=444, upload_rate=33, latency=30, nominal_upload_rate=35, nominal_download_rate=500, latitude=170.23539, longitude=120.34) self.isp = ISP.objects.create(name='rogers', website='rogers.ca', phone=4124124141, support_phone=1800, rating=1.5, facebook='rogers.facebook.com', twitter='rogers.twitter.com', support_link='support.rogers.com') self.isp.save() self.country = Country(name=u'France') self.country.save() self.ndt_profile = NDTProfile.objects.create(user=self.user, name='name', service_type='PUBLIC', country=self.country, isp=self.isp) self.ndt_profile.save()
class NDTModelTests(TestCase): def setUp(self): self.user = User.objects.create(username='******', email='*****@*****.**') self.user.save() self.ndt = NDT.objects.create(download_rate=444, upload_rate=33, latency=30, nominal_upload_rate=35, nominal_download_rate=500, latitude=170.23539, longitude=120.34) self.isp = ISP.objects.create(name='rogers', website='rogers.ca', phone=4124124141, support_phone=1800, rating=1.5, facebook='rogers.facebook.com', twitter='rogers.twitter.com', support_link='support.rogers.com') self.isp.save() self.country = Country(name=u'France') self.country.save() self.ndt_profile = NDTProfile.objects.create(user=self.user, name='name', service_type='PUBLIC', country=self.country, isp=self.isp) self.ndt_profile.save() def tearDown(self): self.isp.delete() self.ndt.delete() self.ndt_profile.delete() self.user.delete() def test_check_ndt_using_id(self): ndt = NDT.objects.get(id=1) self.assertEqual(ndt.upload_rate, 33) def test_update_ndt(self): ndt = NDT.objects.get(id=1) ndt.latitude = 99.999 ndt.save() validation_ndt = NDT.objects.get(id=1) self.assertEqual(approx_equal(float(validation_ndt.latitude), 99.999, 0.01), True) def test_check_ndt_profile_using_name(self): ndt_profile = NDTProfile.objects.get(name='name') self.assertEqual(ndt_profile.service_type, 'PUBLIC') def test_ndt_profile_update(self): ndt_profile = NDTProfile.objects.get(id=1) ndt_profile.name = 'name2' ndt_profile.save() validation_ndt_profile = NDTProfile.objects.get(id=1) self.assertEqual(validation_ndt_profile.name, 'name2') def test_ndt_profile_null_values(self): ndt_profile = NDTProfile.objects.get(id=1) self.assertEqual(ndt_profile.latitude, None) self.assertEqual(ndt_profile.longitude, None) self.assertEqual(ndt_profile.nominal_download_rate, None) self.assertEqual(ndt_profile.nominal_upload_rate, None) self.assertEqual(ndt_profile.bandwidth, None) self.assertEqual(ndt_profile.price, None) self.assertEqual(ndt_profile.contract, False) self.assertEqual(ndt_profile.vpn, None) self.assertEqual(ndt_profile.rating_general, None) self.assertEqual(ndt_profile.rating_customer_service, None) self.assertEqual(ndt_profile.country, self.country) self.assertEqual(ndt_profile.city, None) self.assertEqual(ndt_profile.promotion, None) self.assertNotEqual(ndt_profile.service_type, None) self.assertNotEqual(ndt_profile.name, None) def test_ndt_profile_update_null_field(self): ndt_profile = NDTProfile.objects.get(id=1) ndt_profile.nominal_download_rate = 65.54 ndt_profile.save() validation_ndt_profile = NDTProfile.objects.get(id=1) self.assertEqual(approx_equal(float(validation_ndt_profile.nominal_download_rate), 65.54, 0.01), True) def test_ndt_profile_update_null_boolean_field(self): ndt_profile = NDTProfile.objects.get(id=1) ndt_profile.vpn = True ndt_profile.save() validation_ndt_profile = NDTProfile.objects.get(id=1) self.assertTrue(validation_ndt_profile.vpn) def test_check_if_user_and_ndt_profile_connected(self): ndt_profile = NDTProfile.objects.get(id=1) self.assertEqual(ndt_profile.user.id, self.user.id) def test_ndt_profile_isp_connected(self): ndt_profile = NDTProfile.objects.get(id=1) self.assertEqual(ndt_profile.isp.name, self.isp.name) self.assertEqual(ndt_profile.isp.facebook, self.isp.facebook) def test_ndt_profile_isp_connected_get_isp(self): isp = ISP.objects.get(id=1) self.assertEqual(self.ndt_profile.isp.phone, isp.phone) def test_ndt_profile_isp_connected_after_update(self): isp = ISP.objects.get(id=1) isp.name = 'Rogers2' isp.save() ndt_profile = NDTProfile.objects.get(isp=self.isp) self.assertEqual(isp.name, ndt_profile.isp.name)
def create(self, validated_data): country = Country(**validated_data) country.save() return country