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_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
