django-DefectDojo/tests/system_settings_test.py at master · Maffooch/django-DefectDojo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import sys
import unittest

from base_test_class import BaseTestCase, on_exception_html_source_logger
from selenium.webdriver.common.by import By


class SystemSettingsTest(BaseTestCase):

    @on_exception_html_source_logger
    def test_system_settings_page_loads(self):
        driver = self.driver
        driver.get(self.base_url + "system_settings")
        # Verify the settings page loads with the form
        self.assertTrue(self.is_element_by_css_selector_present("input.btn.btn-primary"))

    @on_exception_html_source_logger
    def test_toggle_deduplication(self):
        # Disable deduplication first, then re-enable it
        # (always end with enabled to avoid breaking dedupe_test.py)
        self.disable_system_setting("id_enable_deduplication")
        self.enable_system_setting("id_enable_deduplication")

    @on_exception_html_source_logger
    def test_toggle_false_positive_history(self):
        # Disable then re-enable to test both states
        # (always end with disabled since that's the default)
        self.enable_system_setting("id_false_positive_history")
        self.disable_system_setting("id_false_positive_history")

    @on_exception_html_source_logger
    def test_toggle_jira_integration(self):
        # Enable JIRA
        self.enable_jira()
        # Disable JIRA
        self.disable_jira()

    @on_exception_html_source_logger
    def test_toggle_github_integration(self):
        # Enable GitHub
        self.enable_github()
        # Disable GitHub
        self.disable_github()

    @on_exception_html_source_logger
    def test_change_max_dupes_setting(self):
        driver = self.driver
        driver.get(self.base_url + "system_settings")
        # Find and modify max duplicates field
        max_dupes_field = driver.find_element(By.ID, "id_max_dupes")
        original_value = max_dupes_field.get_attribute("value")
        max_dupes_field.clear()
        max_dupes_field.send_keys("10")
        driver.find_element(By.CSS_SELECTOR, "input.btn.btn-primary").click()
        # Verify saved
        driver.get(self.base_url + "system_settings")
        self.assertEqual(driver.find_element(By.ID, "id_max_dupes").get_attribute("value"), "10")
        # Reset to original
        driver.find_element(By.ID, "id_max_dupes").clear()
        driver.find_element(By.ID, "id_max_dupes").send_keys(original_value)
        driver.find_element(By.CSS_SELECTOR, "input.btn.btn-primary").click()

    @on_exception_html_source_logger
    def test_settings_save_and_reload(self):
        driver = self.driver
        driver.get(self.base_url + "system_settings")
        # Just verify the page loads and save button works
        driver.find_element(By.CSS_SELECTOR, "input.btn.btn-primary").click()
        # After save, the page should reload without errors
        self.assertFalse(self.is_error_message_present())


def suite():
    suite = unittest.TestSuite()
    suite.addTest(BaseTestCase("test_login"))
    suite.addTest(SystemSettingsTest("test_system_settings_page_loads"))
    suite.addTest(SystemSettingsTest("test_toggle_deduplication"))
    suite.addTest(SystemSettingsTest("test_toggle_false_positive_history"))
    suite.addTest(SystemSettingsTest("test_toggle_jira_integration"))
    suite.addTest(SystemSettingsTest("test_toggle_github_integration"))
    suite.addTest(SystemSettingsTest("test_change_max_dupes_setting"))
    suite.addTest(SystemSettingsTest("test_settings_save_and_reload"))
    return suite


if __name__ == "__main__":
    runner = unittest.TextTestRunner(descriptions=True, failfast=True, verbosity=2)
    ret = not runner.run(suite()).wasSuccessful()
    BaseTestCase.tearDownDriver()
    sys.exit(ret)