Source

django-easy-maps / easy_maps_tests / test_app / tests.py


import mock
from django.test import TestCase
from django import template
# from django.template import Template, Context
from django.utils import unittest
try:
    from django.test.utils import override_settings
except ImportError:
    def override_settings(*args, **kwargs):
        return unittest.skip("overriding settings is not supported by this django version")

from easy_maps.models import Address

class AddressTests(TestCase):
    fake_default_center  = (1, 2)

    def test_empty_dont_save_on_db(self):
        """If we pass an empty address we don't save nothing in the database"""
        n_addresses_before = len(Address.objects.all())

        simple_template_string = """{% load easy_maps_tags %}
        {% easy_map "" 500 500 10 %}
        """
        t = template.Template(simple_template_string)
        t.render(template.Context({}))

        n_addresses_after = len(Address.objects.all())

        self.assertEqual(n_addresses_after, n_addresses_before)

    @override_settings(EASY_MAPS_CENTER=fake_default_center)
    def test_empty_address_use_defaults(self):
        """When an empty address is passed uses the EASY_MAPS_CENTER setting"""
        a = ""
        simple_template_string = """{%% load easy_maps_tags %%}
        {%% easy_map "%s" 500 500 10 %%}
        """ % a

        address = [None]  # nonlocal

        # below we patch the render_to_string in order to retrieve the map
        # context variable and check its coordinate
        def get_map_context_instance(*args, **kwargs):
            address[0] = kwargs['context_instance']['map']
            return ''

        t = template.Template(simple_template_string)
        with mock.patch('easy_maps.templatetags.easy_maps_tags.render_to_string', get_map_context_instance):
            t.render(template.Context({}))

        self.assertEqual(address[0].latitude, AddressTests.fake_default_center[0])
        self.assertEqual(address[0].longitude, AddressTests.fake_default_center[1])

    @override_settings(EASY_MAPS_CENTER=fake_default_center)
    def test_normal_address(self):
        """If we pass an address don't use the defaults"""
        n_addresses_before = len(Address.objects.all())

        a = "Ekaterinburg, Mira 33"
        simple_template_string = """{%% load easy_maps_tags %%}
        {%% easy_map "%s" 500 500 10 %%}
        """ % a
        t = template.Template(simple_template_string)
        t.render(template.Context({}))

        address = Address.objects.get(address=a)
        self.assertNotEqual(address.latitude, AddressTests.fake_default_center[0])
        self.assertNotEqual(address.longitude, AddressTests.fake_default_center[1])

        n_addresses_after = len(Address.objects.all())

        self.assertEqual(n_addresses_after, n_addresses_before + 1)

    @override_settings(EASY_MAPS_CENTER=fake_default_center)
    def test_use_address_instance(self):
        """It's possible to pass directly to the easy_map tag an Address instance.

        This test checks also that the database is not hit.
        """
        # create a fake address
        Address.objects.create(address='fake')
        n_addresses_before = len(Address.objects.all())

        simple_template_string = """{% load easy_maps_tags %}
        {% easy_map address 500 500 10 %}
        """
        address = Address.objects.all()[0]
        t = template.Template(simple_template_string)
        # this assert works only from django1.3
        ctx = template.Context({'address': address,})
        self.assertNumQueries(0, lambda: t.render(ctx))

        n_addresses_after = len(Address.objects.all())

        # no Address is created in the process
        self.assertEqual(n_addresses_after, n_addresses_before)