From ce02c2b3c3c354d9f972166f6eddcadba6089134 Mon Sep 17 00:00:00 2001
From: Pierre-Yves Chibon <pingou@pingoured.fr>
Date: Aug 02 2018 09:36:52 +0000
Subject: Add tests for the no new branch hook


Signed-off-by: Pierre-Yves Chibon <pingou@pingoured.fr>

---

diff --git a/tests/test_pagure_flask_ui_plugins_pagure_no_new_branch.py b/tests/test_pagure_flask_ui_plugins_pagure_no_new_branch.py
new file mode 100644
index 0000000..9c594bb
--- /dev/null
+++ b/tests/test_pagure_flask_ui_plugins_pagure_no_new_branch.py
@@ -0,0 +1,192 @@
+# -*- coding: utf-8 -*-
+
+"""
+ (c) 2015-2018 - Copyright Red Hat Inc
+
+ Authors:
+   Pierre-Yves Chibon <pingou@pingoured.fr>
+
+"""
+
+from __future__ import unicode_literals
+
+__requires__ = ['SQLAlchemy >= 0.8']
+
+import unittest
+import shutil
+import sys
+import os
+
+
+sys.path.insert(0, os.path.join(os.path.dirname(
+    os.path.abspath(__file__)), '..'))
+
+import pagure.lib
+import tests
+
+
+class PagureFlaskPluginPagureNoNewBranchHooktests(tests.SimplePagureTest):
+    """ Tests for pagure_no_new_branches plugin of pagure """
+
+    def setUp(self):
+        """ Set up the environnment, ran before every tests. """
+        super(PagureFlaskPluginPagureNoNewBranchHooktests, self).setUp()
+
+        tests.create_projects(self.session)
+        tests.create_projects_git(os.path.join(self.path, 'repos'))
+
+        pagure.config.config['GIT_FOLDER'] = os.path.join(
+            self.path, 'repos')
+
+        with tests.user_set(self.app.application, tests.FakeUser()):
+            self.csrf_token = self.get_csrf()
+
+    def test_plugin_pagure_ticket_no_data(self):
+        """ Test the pagure_ticket plugin on/off endpoint. """
+
+        user = tests.FakeUser(username='pingou')
+        with tests.user_set(self.app.application, user):
+            output = self.app.get(
+                '/test/settings/Prevent creating new branches by git push')
+            self.assertEqual(output.status_code, 200)
+            output_text = output.get_data(as_text=True)
+            self.assertIn(
+                '<title>Settings Prevent creating new branches by git '
+                'push - test - Pagure</title>', output_text)
+            self.assertIn(
+                '<input class="form-control" id="active" name="active" '
+                'type="checkbox" value="y">', output_text)
+
+            data = {}
+
+            output = self.app.post(
+                '/test/settings/Prevent creating new branches by git push',
+                data=data)
+            self.assertEqual(output.status_code, 200)
+            output_text = output.get_data(as_text=True)
+            self.assertIn(
+                '<title>Settings Prevent creating new branches by git push '
+                '- test - Pagure</title>', output_text)
+            self.assertIn(
+                '<input class="form-control" id="active" name="active" '
+                'type="checkbox" value="y">', output_text)
+
+    def test_plugin_pagure_ticket_deactivate(self):
+        """ Test the pagure_ticket plugin on/off endpoint. """
+        user = tests.FakeUser(username='pingou')
+        with tests.user_set(self.app.application, user):
+            data = {'csrf_token': self.csrf_token}
+
+            output = self.app.post(
+                '/test/settings/Prevent creating new branches by git push',
+                data=data, follow_redirects=True)
+            self.assertEqual(output.status_code, 200)
+            output_text = output.get_data(as_text=True)
+            self.assertIn(
+                '<h5 class="pl-2 font-weight-bold text-muted">'
+                'Project Settings</h5>\n', output_text)
+            self.assertIn(
+                'Hook Prevent creating new branches by git push deactivated',
+                output_text)
+
+            output = self.app.get(
+                '/test/settings/Prevent creating new branches by git push')
+            self.assertEqual(output.status_code, 200)
+            output_text = output.get_data(as_text=True)
+            self.assertIn(
+                '<title>Settings Prevent creating new branches by git push '
+                '- test - Pagure</title>', output_text)
+            self.assertIn(
+                '<input class="form-control" id="active" name="active" '
+                'type="checkbox" value="y">', output_text)
+
+            self.assertFalse(os.path.exists(os.path.join(
+                self.path, 'repos', 'test.git', 'hooks', 'post-receive.pagure')))
+
+    def test_plugin_pagure_ticket_activate(self):
+        """ Test the pagure_ticket plugin on/off endpoint. """
+
+        user = tests.FakeUser(username='pingou')
+        with tests.user_set(self.app.application, user):
+            # Activate hook
+            data = {
+                'csrf_token': self.csrf_token,
+                'active': 'y',
+            }
+
+            output = self.app.post(
+                '/test/settings/Prevent creating new branches by git push',
+                data=data, follow_redirects=True)
+            self.assertEqual(output.status_code, 200)
+            output_text = output.get_data(as_text=True)
+            self.assertIn(
+                '<h5 class="pl-2 font-weight-bold text-muted">'
+                'Project Settings</h5>\n', output_text)
+            self.assertIn(
+                'Hook Prevent creating new branches by git push activated',
+                output_text)
+
+            output = self.app.get(
+                '/test/settings/Prevent creating new branches by git push')
+            self.assertEqual(output.status_code, 200)
+            output_text = output.get_data(as_text=True)
+            self.assertIn(
+                '<title>Settings Prevent creating new branches by git push '
+                '- test - Pagure</title>', output_text)
+            self.assertIn(
+                '<input checked class="form-control" id="active" name="active" '
+                'type="checkbox" value="y">', output_text)
+
+            self.assertTrue(os.path.exists(os.path.join(
+                self.path, 'repos', 'test.git', 'hooks',
+                'pre-receive.pagure_no_new_branches')))
+
+            # De-Activate hook
+            data = {'csrf_token': self.csrf_token}
+            output = self.app.post(
+                '/test/settings/Prevent creating new branches by git push',
+                data=data, follow_redirects=True)
+            self.assertEqual(output.status_code, 200)
+            output_text = output.get_data(as_text=True)
+            self.assertIn(
+                '<h5 class="pl-2 font-weight-bold text-muted">'
+                'Project Settings</h5>\n', output_text)
+            self.assertIn(
+                'Hook Prevent creating new branches by git push deactivated',
+                output_text)
+
+            output = self.app.get(
+                '/test/settings/Prevent creating new branches by git push')
+            self.assertEqual(output.status_code, 200)
+            output_text = output.get_data(as_text=True)
+            self.assertIn(
+                '<title>Settings Prevent creating new branches by git push '
+                '- test - Pagure</title>', output_text)
+            self.assertIn(
+                '<input class="form-control" id="active" name="active" '
+                'type="checkbox" value="y">', output_text)
+
+            self.assertFalse(os.path.exists(os.path.join(
+                self.path, 'repos', 'test.git', 'hooks',
+                'pre-receive.pagure_no_new_branches')))
+
+    def test_plugin_pagure_ticket_activate_w_no_repo(self):
+        """ Test the pagure_ticket plugin on/off endpoint. """
+        shutil.rmtree(os.path.join(self.path, 'repos', 'test.git'))
+
+        user = tests.FakeUser(username='pingou')
+        with tests.user_set(self.app.application, user):
+            # Try re-activate hook w/o the git repo
+            data = {
+                'csrf_token': self.csrf_token,
+                'active': 'y',
+            }
+
+            output = self.app.post(
+                '/test/settings/Prevent creating new branches by git push',
+                data=data)
+            self.assertEqual(output.status_code, 404)
+
+
+if __name__ == '__main__':
+    unittest.main(verbosity=2)