From 16c5781815d4695fe89db3c5cec5191dc1b81c6f Mon Sep 17 00:00:00 2001
From: Vivek Anand <vivekanand1101@gmail.com>
Date: Sep 21 2017 10:26:39 +0000
Subject: read only repo: Add unit test for update_read_only_mode method


Signed-off-by: Vivek Anand <vivekanand1101@gmail.com>

---

diff --git a/tests/test_pagure_lib.py b/tests/test_pagure_lib.py
index 9f76e7b..0ebd630 100644
--- a/tests/test_pagure_lib.py
+++ b/tests/test_pagure_lib.py
@@ -5865,5 +5865,29 @@ foo bar
         )
         self.assertEqual(len(out), 0)
 
+    def test_update_read_only_mode(self):
+        """ Test the update_read_only_mode method of pagure.lib """
+
+        tests.create_projects(self.session)
+
+        project_obj = pagure.lib._get_project(self.session, 'test')
+        # Default mode of project is read only
+        self.assertEqual(project_obj.read_only, True)
+
+        # Remove read only
+        pagure.lib.update_read_only_mode(self.session, project_obj, False)
+        self.session.commit()
+
+        project_obj = pagure.lib._get_project(self.session, 'test')
+        self.assertEqual(project_obj.read_only, False)
+
+        # Try reversing it back
+        pagure.lib.update_read_only_mode(self.session, project_obj, True)
+        self.session.commit()
+
+        project_obj = pagure.lib._get_project(self.session, 'test')
+        self.assertEqual(project_obj.read_only, True)
+
+
 if __name__ == '__main__':
     unittest.main(verbosity=2)