class TestServices(unittest.TestCase): def setUp(self): self.helper = Secure_Track_Helper("localhost", ("username", "password")) self.patcher = patch( 'pytos.common.rest_requests.requests.Session.send') self.mock_get_uri = self.patcher.start() self.mock_get_uri.return_value.status_code = 200 def tearDown(self): self.patcher.stop() def test_01_get_services_for_device(self): self.mock_get_uri.return_value.content = fake_request_response( "services") services = self.helper.get_services_for_device(158) self.assertIsInstance(services, Services_List) def test_02_get_service_for_device_by_name(self): self.mock_get_uri.return_value.content = fake_request_response( "services") with patch( 'pytos.common.rest_requests.requests.Request') as mock_get_uri: service = self.helper.get_service_for_device_by_name( 158, 'service1') mock_get_uri.assert_called_with( 'GET', 'https://localhost/securetrack/api/devices/158/services?name=service1', auth=('username', 'password'), headers={}, params=None) def test_03_get_service_by_device_and_object_id(self): self.mock_get_uri.return_value.content = fake_request_response( "services") with patch( 'pytos.common.rest_requests.requests.Request') as mock_get_uri: service = self.helper.get_service_by_device_and_object_id( 158, 17973529) mock_get_uri.assert_called_with( 'GET', 'https://localhost/securetrack/api/devices/158/services/17973529', auth=('username', 'password'), headers={}, params=None) def test_04_get_member_services_for_group_service(self): self.mock_get_uri.return_value.content = fake_request_response( "network_objects") g_network_object = self.helper.get_network_objects_for_device(158)[-1] self.mock_get_uri.return_value.content = fake_request_response( "services") members = self.helper.get_member_network_objects_for_group_network_object( g_network_object, 158) self.assertIsInstance(members, list)
class TestNetworkObjects(unittest.TestCase): def setUp(self): self.helper = Secure_Track_Helper("localhost", ("username", "password")) self.patcher = patch( 'pytos.common.rest_requests.requests.Session.send') self.mock_get_uri = self.patcher.start() self.mock_get_uri.return_value.status_code = 200 def tearDown(self): self.patcher.stop() def test_01_get_network_objects_for_device(self): self.mock_get_uri.return_value.content = fake_request_response( "network_objects") network_objects = self.helper.get_network_objects_for_device(158) self.assertIsInstance(network_objects, Network_Objects_List) def test_02_network_object_text_search(self): self.mock_get_uri.return_value.content = fake_request_response( "network_objects") with patch( 'pytos.common.rest_requests.requests.Request') as mock_get_uri: network_objects = self.helper.network_object_text_search( "192.168", "any_field") mock_get_uri.assert_called_with( 'GET', 'https://localhost/securetrack/api/network_objects/search?filter=text&any_field=192.168', auth=('username', 'password'), headers={}, params=None) self.assertIsInstance(network_objects, Network_Objects_List) def test_03_network_object_subnet_search(self): self.mock_get_uri.return_value.content = fake_request_response( "network_objects") with patch( 'pytos.common.rest_requests.requests.Request') as mock_get_uri: network_objects = self.helper.network_object_subnet_search( "192.168.0.0", "contained_in") mock_get_uri.assert_called_with( 'GET', 'https://localhost/securetrack/api/network_objects/search?filter=subnet&contained_in=192.168.0.0', auth=('username', 'password'), headers={}, params=None) # def test_04_get_network_objects(self): # network_objects = self.helper.get_network_objects() # self.assertIsInstance(network_objects, dict) # self.assertTrue(len(network_objects) > 0) # def test_04_get_network_object_by_device_and_object_id(self): self.mock_get_uri.return_value.content = fake_request_response( "network_objects") with patch( 'pytos.common.rest_requests.requests.Request') as mock_get_uri: network_object = self.helper.get_network_object_by_device_and_object_id( 158, 3418214) mock_get_uri.assert_called_with( 'GET', 'https://localhost/securetrack/api/devices/158/network_objects/3418214', auth=('username', 'password'), headers={}, params=None) def test_05_get_member_network_objects_for_group_network_object(self): self.mock_get_uri.return_value.content = fake_request_response( "network_objects") g_network_object = self.helper.get_network_objects_for_device(158)[-1] members = self.helper.get_member_network_objects_for_group_network_object( g_network_object, 158) self.assertIsInstance(members, list)