1 files changed, 39 insertions, 0 deletions
diff --git a/unittests.py b/unittests.py
index d70b3d6..4c962cb 100755
--- a/unittests.py
+++ b/unittests.py
@@ -36,6 +36,7 @@ from pygerrit.events import PatchsetCreatedEvent, \
     ErrorEvent, MergeFailedEvent, ReviewerAddedEvent, TopicChangedEvent
 from pygerrit.client import GerritClient
 from pygerrit import GerritReviewMessageFormatter
+from pygerrit.rest import GerritReview
 
 EXPECTED_TEST_CASE_FIELDS = ['header', 'footer', 'paragraphs', 'result']
 
@@ -427,5 +428,43 @@ class TestGerritReviewMessageFormatter(unittest.TestCase):
                              "Formatted message does not match expected "
                              "result in test case #%d:\n[%s]" % (i, m))
 
+
+class TestGerritReview(unittest.TestCase):
+
+    """ Test that the GerritReview class behaves properly. """
+
+    def test_str(self):
+        """ Test for str function. """
+        obj = GerritReview()
+        self.assertEqual(str(obj), '{}')
+
+        obj2 = GerritReview(labels={'Verified': 1, 'Code-Review': -1})
+        self.assertEqual(
+            str(obj2),
+            '{"labels": {"Verified": 1, "Code-Review": -1}}')
+
+        obj3 = GerritReview(comments=[{'filename': 'Makefile',
+                                      'line': 10, 'message': 'test'}])
+        self.assertEqual(
+            str(obj3),
+            '{"comments": {"Makefile": [{"line": 10, "message": "test"}]}}')
+
+        obj4 = GerritReview(labels={'Verified': 1, 'Code-Review': -1},
+                            comments=[{'filename': 'Makefile', 'line': 10,
+                                       'message': 'test'}])
+        self.assertEqual(
+            str(obj4),
+            '{"labels": {"Verified": 1, "Code-Review": -1},'
+            ' "comments": {"Makefile": [{"line": 10, "message": "test"}]}}')
+
+        obj5 = GerritReview(comments=[{'filename': 'Makefile', 'line': 15,
+                            'message': 'test'}, {'filename': 'Make',
+                            'line': 10, 'message': 'test1'}])
+        self.assertEqual(
+            str(obj5),
+            '{"comments": {"Make": [{"line": 10, "message": "test1"}],'
+            ' "Makefile": [{"line": 15, "message": "test"}]}}')
+
+
 if __name__ == '__main__':
     unittest.main()