from django.core.urlresolvers import reverse as r
from django.test import TestCase, SimpleTestCase
from leaflet.forms.widgets import LeafletWidget

from ..models import Barragem


class BarragemNewPostValidTest(TestCase):
    def setUp(self):
        data = dict(
            nome="Coremas",
            finalidade="Irrigação",
            capacidade_m3=720000000,
            altura_m=47,
            material_macico='Terra',
            geom='{"type":"Point","coordinates":[-37.012, -8.212]}'
        )
        self.resp = self.client.post(r('hidrology:barragem_new'), data)

    def test_post_valid(self):
        """Valid POST should redirect to /barragens/"""
        self.assertRedirects(self.resp, r('hidrology:barragem_list'))

    def test_save_barragem(self):
        self.assertTrue(Barragem.objects.exists())


class BarragemNewPostInvalidTest(TestCase):
    def setUp(self):
        data = dict(
            nome="Coremas",
            finalidade="Irrigação",
            capacidade_m3=720000000,
            altura_m=47,
            material_macico='Terra',
            geom=''
        )
        self.resp = self.client.post(r('hidrology:barragem_new'), data)

    def test_post(self):
        """Invalid POST should not redirect"""
        self.assertEqual(200, self.resp.status_code)

    def test_dont_save_subscription(self):
        self.assertFalse(Barragem.objects.exists())

    def test_form_has_errors(self):
        form = self.resp.context['form']
        self.assertTrue(form.errors)


class SettingsOverridesTest(SimpleTestCase):
    def test_settings_overrides(self):
        widget = LeafletWidget(attrs={
            'settings_overrides': {
                'DEFAULT_CENTER': (8.0, 3.14),
            }
        })
        output = widget.render('geom', '', {'id': 'geom'})
        self.assertIn('"center": [8.0, 3.14]', output)