Add more tests for GeneralObject

* wqflask/tests/base/test_general_object.py: test object's magic methods

1 files changed, 6 insertions, 0 deletions

diff --git a/wqflask/tests/base/test_general_object.py b/wqflask/tests/base/test_general_object.py
index 699cb079..df5791e0 100644
--- a/wqflask/tests/base/test_general_object.py
+++ b/wqflask/tests/base/test_general_object.py
@@ -12,6 +12,7 @@ class TestGeneralObjectTests(unittest.TestCase):
         """Test whether base contents are stored properly"""
         test_obj = GeneralObject("a", "b", "c")
         self.assertEqual("abc", ''.join(test_obj.contents))
+        self.assertEqual(len(test_obj), 0)
 
     def test_object_dict(self):
         """Test whether the base class is printed properly"""
@@ -19,3 +20,8 @@ class TestGeneralObjectTests(unittest.TestCase):
         self.assertEqual(str(test_obj), "value = 1\nname = test\n")
         self.assertEqual(
             repr(test_obj), "value = 1\nname = test\ncontents = ['a']\n")
+        self.assertEqual(len(test_obj), 2)
+        self.assertEqual(getattr(test_obj, "value"), 1)
+        self.assertEqual(test_obj["value"], 1)
+        test_obj["test"] = 1
+        self.assertEqual(test_obj["test"], 1)