diff options
Diffstat (limited to 'vmdb/tags_tests.py')
-rw-r--r-- | vmdb/tags_tests.py | 27 |
1 files changed, 20 insertions, 7 deletions
diff --git a/vmdb/tags_tests.py b/vmdb/tags_tests.py index d3171d8..5264c95 100644 --- a/vmdb/tags_tests.py +++ b/vmdb/tags_tests.py @@ -121,18 +121,31 @@ class TagsTests(unittest.TestCase): with self.assertRaises(vmdb.AlreadyHasFsType): tags.set_fstype("first", "ext4") - def test_set_uuid(self): + def test_set_fsuuid(self): tags = vmdb.Tags() tags.append("first") - tags.set_uuid("first", "uuid") - self.assertEqual(tags.get_uuid("first"), "uuid") + tags.set_fsuuid("first", "uuid") + self.assertEqual(tags.get_fsuuid("first"), "uuid") - def test_set_uuid_raises_error_for_double_fstype(self): + def test_set_fsuuid_raises_error_for_double_fstype(self): tags = vmdb.Tags() tags.append("first") - tags.set_uuid("first", "uuid") - with self.assertRaises(vmdb.AlreadyHasUuid): - tags.set_uuid("first", "other") + tags.set_fsuuid("first", "uuid") + with self.assertRaises(vmdb.AlreadyHasFsUuid): + tags.set_fsuuid("first", "other") + + def test_set_luksuuid(self): + tags = vmdb.Tags() + tags.append("first") + tags.set_luksuuid("first", "uuid") + self.assertEqual(tags.get_luksuuid("first"), "uuid") + + def test_set_luksuuid_raises_error_for_double_fstype(self): + tags = vmdb.Tags() + tags.append("first") + tags.set_luksuuid("first", "uuid") + with self.assertRaises(vmdb.AlreadyHasLuksUuid): + tags.set_luksuuid("first", "other") def test_set_dm(self): tags = vmdb.Tags() |