Improved Attribute testcases on FileSets.

edaa-org · Aug 21, 2023 · c61f44d · c61f44d
1 parent 36fff0e
commit c61f44d
Showing 1 changed file with 29 additions and 2 deletions.
diff --git a/tests/unit/FileSet.py b/tests/unit/FileSet.py
@@ -318,6 +318,10 @@ def test_FileSet(self):
 		fileSet.Validate()
 
 
+class Attr(Attribute):
+	pass
+
+
 class Attributes(TestCase):
 	def test_AddAttribute_WrongType(self):
 		fileSet = FileSet("fileset")
@@ -328,7 +332,30 @@ def test_AddAttribute_WrongType(self):
 	def test_AddAttribute_Normal(self):
 		fileSet = FileSet("fileset")
 
-		class Attr(Attribute):
-			pass
+		fileSet[Attr] = 5
+
+	def test_GetAttribute_WrongType(self):
+		fileSet = FileSet("fileset")
+		fileSet[Attr] = 5
+
+		with self.assertRaises(TypeError):
+			_ = fileSet["attr"]
 
+	def test_GetAttribute_Normal(self):
+		fileSet = FileSet("fileset")
 		fileSet[Attr] = 5
+
+		_ = fileSet[Attr]
+
+	def test_DelAttribute_WrongType(self):
+		fileSet = FileSet("fileset")
+		fileSet[Attr] = 5
+
+		with self.assertRaises(TypeError):
+			del fileSet["attr"]
+
+	def test_DelAttribute_Normal(self):
+		fileSet = FileSet("fileset")
+		fileSet[Attr] = 5
+
+		del fileSet[Attr]