diff options
Diffstat (limited to 'service/test/test_tags.py')
| -rw-r--r-- | service/test/test_tags.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/service/test/test_tags.py b/service/test/test_tags.py index 5e0ea426..9c57ee07 100644 --- a/service/test/test_tags.py +++ b/service/test/test_tags.py @@ -5,9 +5,10 @@ from app.tags import Tags, Tag class TagTestCase(unittest.TestCase): def test_create_tag(self): - tag = Tag('test') + tag = Tag('test', True) self.assertEqual(tag.name, 'test') + class TagsTestCase(unittest.TestCase): def test_add_tag_to_collection(self): @@ -28,10 +29,14 @@ class TagsTestCase(unittest.TestCase): tag_collection = Tags() tag_collection.add('test') tag_collection.add('test2') - self.assertEqual(tag_collection.find('test'), Tag('test')) + self.assertEqual(tag_collection.find('test'), Tag('test', True)) def test_special_tags_always_exist(self): special_tags = ['inbox', 'sent', 'drafts', 'trash'] tag_collection = Tags() for tag in special_tags: - self.assertIn(Tag(tag), tag_collection)
\ No newline at end of file + self.assertIn(Tag(tag, True), tag_collection) + + def test_special_tags_are_default(self): + tags = Tags() + self.assertTrue(tags.find('inbox').default) |