From 0e447ee3e20fa8bbccc0c257124ff22e37544d69 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Petr=20Dlouh=C3=BD?= <petr.dlouhy@email.cz>
Date: Sun, 14 Jun 2015 17:25:56 +0200
Subject: [PATCH] add tests for custom additional fields

---
 tests/settings.py | 9 +++++++++
 tests/storage.py  | 3 +++
 2 files changed, 12 insertions(+)

diff --git a/tests/settings.py b/tests/settings.py
index 04a428a..d7678e2 100644
--- a/tests/settings.py
+++ b/tests/settings.py
@@ -51,6 +51,14 @@ long_value = 123456
 if not six.PY3:
     long_value = long(long_value)
 
+CONSTANCE_ADDITIONAL_FIELDS = {
+     'yes_no_null_select': ['django.forms.fields.ChoiceField',
+         {
+         'widget': 'django.forms.Select',
+         'choices': (("-----", None), ("yes", "Yes"), ("no", "No"))
+         }],
+     }
+
 CONSTANCE_CONFIG = {
     'INT_VALUE': (1, 'some int'),
     'LONG_VALUE': (long_value, 'some looong int'),
@@ -63,6 +71,7 @@ CONSTANCE_CONFIG = {
     'FLOAT_VALUE': (3.1415926536, 'PI'),
     'DATE_VALUE': (date(2010, 12, 24), 'Merry Chrismas'),
     'TIME_VALUE': (time(23, 59, 59), 'And happy New Year'),
+    'CHOICE_VALUE': ('yes', 'select yes or no', 'yes_no_null_select'),
 }
 
 DEBUG = True
diff --git a/tests/storage.py b/tests/storage.py
index a78b02a..338daae 100644
--- a/tests/storage.py
+++ b/tests/storage.py
@@ -27,6 +27,7 @@ class StorageTestsMixin(object):
         self.assertEqual(self.config.FLOAT_VALUE, 3.1415926536)
         self.assertEqual(self.config.DATE_VALUE, date(2010, 12, 24))
         self.assertEqual(self.config.TIME_VALUE, time(23, 59, 59))
+        self.assertEqual(self.config.CHOICE_VALUE, 'yes')
 
         # set values
         self.config.INT_VALUE = 100
@@ -39,6 +40,7 @@ class StorageTestsMixin(object):
         self.config.FLOAT_VALUE = 2.718281845905
         self.config.DATE_VALUE = date(2001, 12, 20)
         self.config.TIME_VALUE = time(1, 59, 0)
+        self.config.CHOICE_VALUE = 'no'
 
         # read again
         self.assertEqual(self.config.INT_VALUE, 100)
@@ -51,6 +53,7 @@ class StorageTestsMixin(object):
         self.assertEqual(self.config.FLOAT_VALUE, 2.718281845905)
         self.assertEqual(self.config.DATE_VALUE, date(2001, 12, 20))
         self.assertEqual(self.config.TIME_VALUE, time(1, 59, 0))
+        self.assertEqual(self.config.CHOICE_VALUE, 'no')
 
     def test_nonexistent(self):
         try: