diff --git a/taskw/test/test_datas.py b/taskw/test/test_datas.py index 5f3e3b5..db89063 100644 --- a/taskw/test/test_datas.py +++ b/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")