forked from DefectDojo/django-DefectDojo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathendpoint_test.py
99 lines (86 loc) · 4.38 KB
/
endpoint_test.py
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
import sys
import unittest
from base_test_class import BaseTestCase
from product_test import ProductTest
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
class EndpointTest(BaseTestCase):
def test_create_endpoint(self):
# Login to the site.
# Username and password will be gotten from environ
driver = self.driver
# Navigate to the Endpoint page
driver.get(self.base_url + "endpoint")
# "Click" the dropdown button to see options
driver.find_element(By.ID, "dropdownMenu1").click()
# "Click" the New Endpoint
driver.find_element(By.LINK_TEXT, "New Endpoint").click()
# Keep a good practice of clearing field before entering value
# Endpoints
driver.find_element(By.ID, "id_endpoint").clear()
driver.find_element(By.ID, "id_endpoint").send_keys("moving.com.rnd")
# Select product to assign endpoint to
Select(driver.find_element(By.ID, "id_product")).select_by_visible_text("QA Test")
# submit
driver.find_element(By.CSS_SELECTOR, "input.btn.btn-primary").click()
# Query the site to determine if the finding has been added
# Assert ot the query to dtermine status of failure
self.assertTrue(self.is_success_message_present(text="Endpoint added successfully"))
def test_edit_endpoint(self):
# Login to the site. Password will have to be modified
# to match an admin password in your own container
driver = self.driver
# Navigate to the endpoint page
driver.get(self.base_url + "endpoint")
# Select one of the previously created endpoint to edit
driver.find_element(By.LINK_TEXT, "moving.com.rnd").click()
# "Click" the dropdown button to see options
driver.find_element(By.ID, "dropdownMenu1").click()
# "Click" the Edit Endpoint
driver.find_element(By.LINK_TEXT, "Edit Endpoint").click()
# Clear the old endpoint host name
driver.find_element(By.ID, "id_host").clear()
# Fill in the endpoint host name
driver.find_element(By.ID, "id_host").send_keys("rnd.moving.com")
# Fill in port for endpoint
driver.find_element(By.ID, "id_port").clear()
driver.find_element(By.ID, "id_port").send_keys("8080")
# "Click" the submit button to complete the transaction
driver.find_element(By.CSS_SELECTOR, "input.btn.btn-primary").click()
# Query the site to determine if the product has been added
# Assert ot the query to dtermine status of failure
self.assertTrue(self.is_success_message_present(text="Endpoint updated successfully"))
def test_delete_endpoint(self):
# Login to the site. Password will have to be modified
# to match an admin password in your own container
driver = self.driver
# Navigate to the endpoint page
driver.get(self.base_url + "endpoint")
# Select one of the previously created endpoint to delete
driver.find_element(By.LINK_TEXT, "rnd.moving.com:8080").click()
# "Click" the dropdown button to see options
driver.find_element(By.ID, "dropdownMenu1").click()
# "Click" the Delete Endpoint
driver.find_element(By.LINK_TEXT, "Delete Endpoint").click()
# "Click" the delete button to complete the transaction
driver.find_element(By.CSS_SELECTOR, "button.btn.btn-danger").click()
# Query the site to determine if the product has been added
# Assert ot the query to dtermine status of failure
self.assertTrue(self.is_success_message_present(text="Endpoint and relationships removed."))
def suite():
suite = unittest.TestSuite()
# Add each test the the suite to be run
# success and failure is output by the test
suite.addTest(BaseTestCase("test_login"))
suite.addTest(BaseTestCase("disable_block_execution"))
suite.addTest(ProductTest("test_create_product"))
suite.addTest(EndpointTest("test_create_endpoint"))
suite.addTest(EndpointTest("test_edit_endpoint"))
suite.addTest(EndpointTest("test_delete_endpoint"))
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)