Update unit tests for commit 29f188

2 files changed, 5 insertions, 0 deletions

diff --git a/kittystore/test/__init__.py b/kittystore/test/__init__.py
index 2a2b228..b6833aa 100644
--- a/kittystore/test/__init__.py
+++ b/kittystore/test/__init__.py
@@ -14,5 +14,6 @@ class FakeList(object):
     def __init__(self, name):
         self.fqdn_listname = name
         self.display_name = None
+        self.description = None
 
 
diff --git a/kittystore/test/test_storm_store.py b/kittystore/test/test_storm_store.py
index a4ed61b..cea9b60 100644
--- a/kittystore/test/test_storm_store.py
+++ b/kittystore/test/test_storm_store.py
@@ -81,13 +81,17 @@ class TestStormStore(unittest.TestCase):
         msg.set_payload("Dummy message")
         ml = FakeList("example-list")
         ml.display_name = u"name 1"
+        ml.description = u"desc 1"
         self.store.add_to_list(ml, msg)
         ml_db = self.store.db.find(List).one()
         self.assertEqual(ml_db.display_name, "name 1")
+        self.assertEqual(ml_db.description, "desc 1")
         ml.display_name = u"name 2"
+        ml.description = u"desc 2"
         self.store.add_to_list(ml, msg)
         ml_db = self.store.db.find(List).one()
         self.assertEqual(ml_db.display_name, "name 2")
+        self.assertEqual(ml_db.description, "desc 2")
 
     #def test_non_ascii_payload(self):
     #    """add_to_list must handle non-ascii messages"""