class SetLocationTest(TestCase):
def setUp(self):
self.client = Client()
self.url = '/set_location/'
self.location = any_model(MyCustomLocation)
self.location_model_patcher = patch.object(settings, 'GEOIP_LOCATION_MODEL', 'test_app.models.MyCustomLocation')
self.location_model = self.location_model_patcher.start()
def tearDown(self):
self.location_model_patcher.stop()
def test_get(self):
response = self.client.get(self.url, data={'location_id': self.location.id})
self.assertFalse(settings.GEOIP_COOKIE_NAME in response.cookies)
self.assertRedirects(response, 'http://testserver/')
def test_get_or_post_next_url(self):
for method in ['get', 'post']:
method_call = getattr(self.client, method)
response = method_call(self.url, data={'next': '/hello/',
'location_id': self.location.id})
self.assertRedirects(response, 'http://testserver/hello/')
def test_post_ok(self):
response = self.client.post(self.url, data={'location_id': self.location.id})
self.assertEqual(response.cookies[settings.GEOIP_COOKIE_NAME].value, str(self.location.id))
self.assertRedirects(response, 'http://testserver/')
def test_post_fake_location(self):
response = self.client.post(self.url, data={'location_id': self.location.id+1})
self.assertFalse(settings.GEOIP_COOKIE_NAME in response.cookies)
self.assertRedirects(response, 'http://testserver/')
class DjangoAnyClient(TestCase):
urls = 'django_any.tests.test_client'
def setUp(self):
self.client = Client()
def test_login_as_super_user(self):
# TODO make test page which will not require is_staff like /admin
self.assertTrue(self.client.login_as(is_superuser=True, is_staff=True))
response = self.client.get('/admin/')
self.assertEquals(200, response.status_code)
def test_login_as_custom_user(self):
user = User.objects.create_user('john', '*****@*****.**',
'johnpassword')
self.assertTrue(self.client.login_as(user=user))
def test_login_as_failed(self):
user = None
self.assertRaises(AssertionError, self.client.login_as, user=user)
def test_post_any_data(self):
response = self.client.post_any_data('/view/')
self.assertRedirects(response, '/view/')
class TestBidActions(TestCase):
__metaclass__ = WithTestDataSeed
def setUp(self):
self.client = Client()
self.user = self.client.login_as()
self.avatar = any_model(
Avatar, owner=self.user,
today=0, hours_spent=0,
health=50)
self.market = any_model(
Market, buy_item_type=INVENTORY_TYPE.DUBLONS,
sell_item_type=INVENTORY_TYPE.BANANAS)
self.market_url = reverse('market_index', kwargs={'avatar_pk':self.avatar.pk, 'market_pk':self.market.pk})
def test_create_bid_succed(self):
self.client.post(self.market_url, {'action':'create_bid', 'quantity':'10', 'rate':'200', 'direction':'SELL'})
self.assertEqual(1, Bid.objects.filter(owner=self.avatar).count())
bid = Bid.objects.get(owner=self.avatar)
self.assertEqual(10, bid.quantity)
self.assertEqual(self.market, bid.market)
def test_delete_bid_succed(self):
bid = any_model(Bid, owner=self.avatar, market=self.market)
self.client.post(self.market_url, {'action':'delete_bid', 'bid_pk':'%s' % bid.pk})
self.assertEqual(0, Bid.objects.filter(owner=self.avatar).count())
class DjangoAnyClient(TestCase):
urls = 'django_any.tests.test_client'
def setUp(self):
self.client = Client()
def test_login_as_super_user(self):
self.assertTrue(self.client.login_as(is_superuser=True))
response = self.client.get('/admin/')
self.assertEquals(200, response.status_code)
def test_post_any_data(self):
response = self.client.post_any_data('/view/')
self.assertRedirects(response, '/view/')
class TestVacancyActions(TestCase):
__metaclass__ = WithTestDataSeed
def setUp(self):
self.client = Client()
self.user = self.client.login_as()
self.avatar = any_model(Avatar, owner=self.user, today=0, hours_spent=0, health=50)
self.vacancy_url = reverse("vacancy_index", kwargs={"avatar_pk": self.avatar.pk})
def test_accept_vacancy_succed(self):
vacancy = any_model(Vacancy)
self.client.post(self.vacancy_url, {"action": "accept_vacancy", "vacancy_pk": "%s" % vacancy.pk})
self.assertEqual(0, Vacancy.objects.count())
class DjangoAnyClient(TestCase):
urls = 'django_any.tests.test_client'
def setUp(self):
self.client = Client()
def test_login_as_super_user(self):
self.assertTrue(self.client.login_as(is_superuser=True))
response = self.client.get('/admin/')
self.assertEquals(200, response.status_code)
def test_post_any_data(self):
response = self.client.post_any_data('/view/')
self.assertRedirects(response, '/view/')
class MiddlewareTest(TestCase):
def setUp(self, *args, **kwargs):
self.client = Client()
self.factory = RequestFactory()
self.request = self.factory.get("/", **{"REMOTE_ADDR": "6.6.6.6"})
self.middleware = django_geoip.middleware.LocationMiddleware()
self.get_location_patcher = patch.object(middleware, "get_location")
self.get_location_patcher.start()
self.get_location_mock = self.get_location_patcher.start()
def tearDown(self):
self.get_location_patcher.stop()
def test_get_location_lazy(self):
self.client.get("/")
self.assertEqual(self.get_location_mock.call_count, 0)
def test_process_request(self):
self.get_location_mock.return_value = None
self.middleware.process_request(self.request)
self.assertEqual(self.request.location, None)
self.assertEqual(self.get_location_mock.call_count, 1)
@patch("django_geoip.storage.LocationCookieStorage.set")
@patch.object(LocationCookieStorage, "__init__")
def test_process_response_ok(self, mock, mock_location_set):
mock.return_value = None
base_response = HttpResponse()
self.get_location_mock.return_value = mycity = any_model(City)
self.middleware.process_request(self.request)
self.middleware.process_response(self.request, base_response)
mock.assert_called_once_with(request=self.request, response=base_response)
# workaround simplelazyobject
self.assertEqual(str(mycity), str(mock_location_set.call_args[1]["location"]))
@patch("django_geoip.storage.LocationCookieStorage._do_set")
def test_process_response_empty_request_location(self, mock_do_set):
base_response = HttpResponse()
self.request.location = None
self.middleware.process_response(self.request, base_response)
self.assertFalse(mock_do_set.called)
@patch("django_geoip.storage.LocationCookieStorage.set")
def test_process_response_no_request_location(self, mock_set):
base_response = HttpResponse()
self.middleware.process_response(self.request, base_response)
self.assertFalse(mock_set.called)
def setUp(self):
self.client = Client()
self.url = '/set_location/'
self.location = any_model(MyCustomLocation)
self.location_model_patcher = patch.object(settings, 'GEOIP_LOCATION_MODEL', 'test_app.models.MyCustomLocation')
self.location_model = self.location_model_patcher.start()
def setUp(self):
self.client = Client()
self.user = self.client.login_as()
self.avatar = any_model(Avatar, owner=self.user, today=0, hours_spent=0, health=50)
self.company = any_model(Company, owner=self.avatar)
self.company_url = reverse("company_index", kwargs={"avatar_pk": self.avatar.pk, "company_pk": self.company.pk})
def setUp(self):
self.client = Client()
self.user = self.client.login_as()
self.avatar = any_model(
Avatar, owner=self.user,
today=0, hours_spent=0,
health=50)
self.avatar_url = reverse('avatar_index', kwargs={'avatar_pk':self.avatar.pk})
class TestCompanyActions(TestCase):
__metaclass__ = WithTestDataSeed
def setUp(self):
self.client = Client()
self.user = self.client.login_as()
self.avatar = any_model(Avatar, owner=self.user, today=0, hours_spent=0, health=50)
self.company = any_model(Company, owner=self.avatar)
self.company_url = reverse("company_index", kwargs={"avatar_pk": self.avatar.pk, "company_pk": self.company.pk})
def test_create_vacancy_succed(self):
self.client.post(self.company_url, {"action": "create_vacancy", "salary_per_hour": "1"})
self.assertEqual(1, Vacancy.objects.filter(company=self.company).count())
vacancy = Vacancy.objects.get(company=self.company)
self.assertEqual(1, vacancy.salary_per_hour)
def test_delete_vacancy_succed(self):
bid = any_model(Vacancy, company=self.company)
self.client.post(self.company_url, {"action": "delete_vacancy", "vacancy_pk": "%s" % bid.pk})
self.assertEqual(0, Vacancy.objects.count())
def test_work_with_enough_company_money_succeed(self):
CompanyWorker.objects.create(worker=self.avatar, company=self.company, salary_per_hour=1)
any_model(InventoryItem, owner=self.avatar, quantity=10, item_type=INVENTORY_TYPE.DUBLONS)
initial_health = self.avatar.health
self.client.post(self.company_url, {"action": "work"})
new_health = Avatar.objects.get(pk=self.avatar.pk).health
self.assertTrue(initial_health > new_health)
def setUp(self, *args, **kwargs):
self.client = Client()
self.factory = RequestFactory()
self.request = self.factory.get("/", **{"REMOTE_ADDR": "6.6.6.6"})
self.middleware = django_geoip.middleware.LocationMiddleware()
self.get_location_patcher = patch.object(middleware, "get_location")
self.get_location_patcher.start()
self.get_location_mock = self.get_location_patcher.start()
def setUp(self):
self.client = Client()
self.user = self.client.login_as()
self.avatar = any_model(
Avatar, owner=self.user,
today=0, hours_spent=0,
health=50)
self.market = any_model(
Market, buy_item_type=INVENTORY_TYPE.DUBLONS,
sell_item_type=INVENTORY_TYPE.BANANAS)
self.market_url = reverse('market_index', kwargs={'avatar_pk':self.avatar.pk, 'market_pk':self.market.pk})
class DjangoAnyClient(TestCase):
def setUp(self):
self.client = Client()
def test_login_as_super_user(self):
# TODO make test page which will not require is_staff like /admin
self.assertTrue(self.client.login_as(is_superuser=True, is_staff=True))
response = self.client.get('/admin/')
self.assertEqual(200, response.status_code)
def test_login_as_custom_user(self):
user = User.objects.create_user('john', '*****@*****.**', 'johnpassword')
self.assertTrue(self.client.login_as(user=user))
def test_login_as_failed(self):
user = None
self.assertRaises(AssertionError, self.client.login_as, user=user)
def test_post_any_data(self):
response = self.client.post_any_data('/view/')
self.assertRedirects(response, '/view/')
class DjangoAnyClient(TestCase):
urls = 'django_any.tests.test_client'
def setUp(self):
self.client = Client()
def test_login_as_super_user(self):
self.assertTrue(self.client.login_as(is_superuser=True))
response = self.client.get('/admin/')
self.assertEquals(200, response.status_code)
def test_login_as_custom_user(self):
user = User.objects.create_user('john', '*****@*****.**', 'johnpassword')
self.assertTrue(self.client.login_as(user=user))
def test_login_as_failed(self):
user = None
self.assertRaises(AssertionError, self.client.login_as, user=user)
def test_post_any_data(self):
response = self.client.post_any_data('/view/')
self.assertRedirects(response, '/view/')
def setUp(self):
self.client = Client()
def setUp(self):
self.client = Client()
class TestAvatarActions(TestCase):
__metaclass__ = WithTestDataSeed
def setUp(self):
self.client = Client()
self.user = self.client.login_as()
self.avatar = any_model(
Avatar, owner=self.user,
today=0, hours_spent=0,
health=50)
self.avatar_url = reverse('avatar_index', kwargs={'avatar_pk':self.avatar.pk})
def test_sleep_action_restore_health_points(self):
initial_health = self.avatar.health
self.client.post(self.avatar_url, {'action' : 'sleep'})
new_health = Avatar.objects.get(pk=self.avatar.pk).health
self.assertTrue(initial_health < new_health)
def test_gather_bananas_extend_inventory(self):
self.client.post(self.avatar_url, {'action' : 'gather_bananas'})
bananas = self.avatar.get_inventory_item(INVENTORY_TYPE.BANANAS)
self.assertEqual(1, bananas.quantity)
def test_eathing_shoes_is_bad_for_heals(self):
initial_health = self.avatar.health
self.client.post(self.avatar_url, {'action' : 'eat_bananas'})
new_health = Avatar.objects.get(pk=self.avatar.pk).health
self.assertTrue(initial_health > new_health)
def test_eathing_bananas_is_good_for_heals(self):
bananas = self.avatar.get_inventory_item(INVENTORY_TYPE.BANANAS)
bananas.quantity = 1
bananas.save()
initial_health = self.avatar.health
self.client.post(self.avatar_url, {'action' : 'eat_bananas'})
new_health = Avatar.objects.get(pk=self.avatar.pk).health
self.assertTrue(initial_health < new_health)
def test_buy_company_with_enough_gold_succeed(self):
gold = self.avatar.get_inventory_item(INVENTORY_TYPE.GOLD)
gold.quantity = 10
gold.save()
self.client.post(self.avatar_url, {'action':'buy_company', 'name' : 'TheCompany'})
self.assertEqual(1, Company.objects.filter(owner=self.avatar).count())
company = Company.objects.get(owner=self.avatar)
self.assertEqual('TheCompany', company.name)
