From f0c3f22bc23a269a35b0f04e8a8f2ecc124bfd30 Mon Sep 17 00:00:00 2001
From: amatellanes <dev.amatellanes@gmail.com>
Date: Wed, 15 Jan 2014 21:50:33 +0100
Subject: [PATCH] Added tests for gen_ipv4() and gen_ipv6() methods

---
 test/generic/forms.py            |  8 ++++++++
 test/generic/models.py           |  4 ++++
 test/generic/tests/test_mommy.py | 16 ++++++++++++++++
 3 files changed, 28 insertions(+)
 create mode 100644 test/generic/forms.py

diff --git a/test/generic/forms.py b/test/generic/forms.py
new file mode 100644
index 0000000..a2dcec2
--- /dev/null
+++ b/test/generic/forms.py
@@ -0,0 +1,8 @@
+from django.forms import ModelForm
+
+from test.generic.models import DummyIPAddressesFieldModel
+
+
+class DummyIPAddressesFieldForm(ModelForm):
+    class Meta:
+        model = DummyIPAddressesFieldModel
\ No newline at end of file
diff --git a/test/generic/models.py b/test/generic/models.py
index 03e34d3..287c9a7 100755
--- a/test/generic/models.py
+++ b/test/generic/models.py
@@ -199,3 +199,7 @@ class CustomFieldWithoutGeneratorModel(models.Model):
 
 class DummyUniqueIntegerFieldModel(models.Model):
     value = models.IntegerField(unique=True)
+
+
+class DummyIPAddressesFieldModel(models.Model):
+    ip = models.GenericIPAddressField()
diff --git a/test/generic/tests/test_mommy.py b/test/generic/tests/test_mommy.py
index 621131c..24c128f 100644
--- a/test/generic/tests/test_mommy.py
+++ b/test/generic/tests/test_mommy.py
@@ -8,6 +8,7 @@
 from mock import patch, Mock
 
 from model_mommy import mommy
+from model_mommy import generators
 from model_mommy.exceptions import ModelNotFound, AmbiguousModelName, InvalidQuantityException
 from model_mommy.timezone import smart_datetime as datetime
 from test.generic.models import Person, Dog, Store, LonelyPerson, School, SchoolEnrollment, ModelWithImpostorField, Classroom, GuardDog
@@ -16,6 +17,7 @@
 from test.generic.models import DummyNullFieldsModel, DummyBlankFieldsModel
 from test.generic.models import DummyDefaultFieldsModel, DummyMultipleInheritanceModel
 from test.generic.models import DummyGenericForeignKeyModel, NonAbstractPerson
+from test.generic.forms import DummyIPAddressesFieldForm
 
 
 class ModelFinderTest(TestCase):
@@ -340,3 +342,17 @@ def test_skip_fields_with_default(self):
         self.assertEqual(dummy.default_decimal_field, Decimal('0'))
         self.assertEqual(dummy.default_email_field, 'foo@bar.org')
         self.assertEqual(dummy.default_slug_field, 'a-slug')
+
+
+class MommyGeneratesIPAdresses(TestCase):
+    def test_create_model_with_valid_ipv4(self):
+        form_data = {
+            'ip': generators.gen_ipv4(),
+        }
+        self.assertTrue(DummyIPAddressesFieldForm(form_data).is_valid())
+
+    def test_create_model_with_valid_ipv6(self):
+        form_data = {
+            'ip': generators.gen_ipv6(),
+        }
+        self.assertTrue(DummyIPAddressesFieldForm(form_data).is_valid())