django-DefectDojo/tests/engagement_extended_test.py at master · DefectDojo/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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
import logging
import os
import sys
import time
import unittest
from pathlib import Path

from base_test_class import BaseTestCase, on_exception_html_source_logger, set_suite_settings
from product_test import ProductTest
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select

logger = logging.getLogger(__name__)
dir_path = Path(os.path.realpath(__file__)).parent


class EngagementExtendedTest(BaseTestCase):

    @on_exception_html_source_logger
    def test_create_engagement_for_extended_tests(self):
        driver = self.driver
        self.goto_product_overview(driver)
        driver.find_element(By.LINK_TEXT, "QA Test").click()
        driver.find_element(By.CSS_SELECTOR, ".dropdown-toggle.active").click()
        driver.find_element(By.LINK_TEXT, "Add New Interactive Engagement").click()
        driver.find_element(By.ID, "id_name").clear()
        driver.find_element(By.ID, "id_name").send_keys("Extended Test Engagement")
        driver.find_element(By.ID, "id_name").send_keys(Keys.TAB, "Extended engagement for testing.")
        Select(driver.find_element(By.ID, "id_lead")).select_by_visible_text("Admin User (admin)")
        Select(driver.find_element(By.ID, "id_status")).select_by_visible_text("In Progress")
        driver.find_element(By.CSS_SELECTOR, "input.btn.btn-primary").click()

        self.assertTrue(self.is_success_message_present(text="Engagement added successfully"))

    @on_exception_html_source_logger
    def test_close_engagement_for_reopen(self):
        driver = self.driver
        self.goto_product_overview(driver)
        driver.find_element(By.LINK_TEXT, "QA Test").click()
        driver.find_element(By.CSS_SELECTOR, ".dropdown-toggle.active").click()
        driver.find_element(By.LINK_TEXT, "View Engagements").click()
        driver.find_element(By.LINK_TEXT, "Extended Test Engagement").click()
        driver.find_element(By.ID, "dropdownMenu1").click()
        driver.find_element(By.LINK_TEXT, "Close Engagement").click()

        self.assertTrue(self.is_success_message_present(text="Engagement closed successfully."))

    @on_exception_html_source_logger
    def test_reopen_closed_engagement(self):
        driver = self.driver
        self.goto_product_overview(driver)
        driver.find_element(By.LINK_TEXT, "QA Test").click()
        driver.find_element(By.CSS_SELECTOR, ".dropdown-toggle.active").click()
        driver.find_element(By.LINK_TEXT, "View Engagements").click()
        self.wait_for_datatable_if_content("no_active_engagements", "open_wrapper")
        driver.find_element(By.LINK_TEXT, "Extended Test Engagement").click()
        driver.find_element(By.ID, "dropdownMenu1").click()
        driver.find_element(By.LINK_TEXT, "Reopen Engagement").click()

        self.assertTrue(self.is_success_message_present(text="Engagement reopened successfully."))

    @on_exception_html_source_logger
    def test_copy_engagement(self):
        driver = self.driver
        self.goto_product_overview(driver)
        driver.find_element(By.LINK_TEXT, "QA Test").click()
        driver.find_element(By.CSS_SELECTOR, ".dropdown-toggle.active").click()
        driver.find_element(By.LINK_TEXT, "View Engagements").click()
        driver.find_element(By.LINK_TEXT, "Extended Test Engagement").click()
        driver.find_element(By.ID, "dropdownMenu1").click()
        driver.find_element(By.LINK_TEXT, "Copy Engagement").click()
        driver.find_element(By.ID, "id_done").click()
        driver.find_element(By.CSS_SELECTOR, "input.btn.btn-primary").click()

        self.assertTrue(self.is_success_message_present(text="Engagement Copied successfully."))

    @on_exception_html_source_logger
    def test_engagement_ics_export(self):
        driver = self.driver
        self.goto_product_overview(driver)
        driver.find_element(By.LINK_TEXT, "QA Test").click()
        driver.find_element(By.CSS_SELECTOR, ".dropdown-toggle.active").click()
        driver.find_element(By.LINK_TEXT, "View Engagements").click()
        driver.find_element(By.LINK_TEXT, "Extended Test Engagement").click()
        driver.find_element(By.ID, "dropdownMenu1").click()
        driver.find_element(By.LINK_TEXT, "Add To Calendar").click()
        # ICS export should trigger a download, verify no error page
        time.sleep(2)

    @on_exception_html_source_logger
    def test_all_engagements_page_loads(self):
        driver = self.driver
        driver.get(self.base_url + "engagement/all")
        # Wait for the DataTable to initialize
        time.sleep(2)
        body_text = driver.find_element(By.TAG_NAME, "body").text
        self.assertIsNotNone(body_text)

    @on_exception_html_source_logger
    def test_delete_extended_engagement(self):
        driver = self.driver
        self.goto_product_overview(driver)
        driver.find_element(By.LINK_TEXT, "QA Test").click()
        driver.find_element(By.CSS_SELECTOR, ".dropdown-toggle.active").click()
        driver.find_element(By.LINK_TEXT, "View Engagements").click()
        driver.find_element(By.LINK_TEXT, "Extended Test Engagement").click()
        driver.find_element(By.ID, "dropdownMenu1").click()
        driver.find_element(By.LINK_TEXT, "Delete Engagement").click()
        driver.find_element(By.NAME, "delete_name").click()

        self.assertTrue(self.is_success_message_present(text="Engagement and relationships removed."))


def suite():
    suite = unittest.TestSuite()
    suite.addTest(BaseTestCase("test_login"))
    set_suite_settings(suite, jira=False, github=False, block_execution=False)
    suite.addTest(ProductTest("test_create_product"))
    suite.addTest(EngagementExtendedTest("test_create_engagement_for_extended_tests"))
    suite.addTest(EngagementExtendedTest("test_close_engagement_for_reopen"))
    suite.addTest(EngagementExtendedTest("test_reopen_closed_engagement"))
    suite.addTest(EngagementExtendedTest("test_copy_engagement"))
    suite.addTest(EngagementExtendedTest("test_engagement_ics_export"))
    suite.addTest(EngagementExtendedTest("test_all_engagements_page_loads"))
    suite.addTest(EngagementExtendedTest("test_delete_extended_engagement"))
    suite.addTest(ProductTest("test_delete_product"))
    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)