Test string UDAs.

taskw/test/test_datas.py

@@ -30,7 +30,11 @@ def setup(self):
         dname = tempfile.mkdtemp(prefix='taskw-tests-data')
 
         with open(fname, 'w') as f:
-            f.writelines(['data.location=%s' % dname])
+            f.writelines([
+                'data.location=%s\n' % dname,
+                'uda.somestring.label=Testing Date\n',
+                'uda.somestring.type=string\n',
+            ])
 
         # Create empty .data files
         for piece in ['completed', 'pending', 'undo']:
@@ -185,6 +189,17 @@ def test_add_datetime(self):
         # ... we'll just "roughly" test it instead of mocking.
         assert(tasks['pending'][0]['entry'].startswith("20110101T"))
 
+    def test_add_with_uda(self):
+        self.tw.task_add(
+            "foobar",
+            somestring="this is a uda",
+        )
+        tasks = self.tw.load_tasks()
+        eq_(len(tasks['pending']), 1)
+        task = tasks['pending'][0]
+
+        eq_(task['somestring'], "this is a uda")
+
     @raises(ValueError)
     def test_completing_completed_task(self):
         task = self.tw.task_add("foobar")