def test_ping_exception(self, mock_ping): mock_ping.side_effect = ConnectionError( "Failed to establish a new connection") transmission = StixTransmission("trendmicro_vision_one", CONNECTION, CONFIG) ping_response = transmission.ping() self.assertFalse(ping_response["success"])
def test_ping_unknown_failure(self, mock_ping): response = { "error": { "code": "InternalError", } } mock_ping.side_effect = [MockResponse(503, json.dumps(response))] transmission = StixTransmission("trendmicro_vision_one", CONNECTION, CONFIG) ping_response = transmission.ping() self.assertFalse(ping_response["success"]) self.assertEqual(ping_response["code"], "unknown")
def test_ping(self, mock_ping): response = { "data": [{ "modelId": "modelId", "name": "name", "enabled": False }] } mock_ping.side_effect = [MockResponse(200, json.dumps(response))] transmission = StixTransmission("trendmicro_vision_one", CONNECTION, CONFIG) ping_response = transmission.ping() self.assertTrue(ping_response["success"])
def test_ping_failure(self, mock_ping): response = { "error": { "code": "Unsupported", "message": "The specified search parameters are invalid. Verify the parameters and try again." } } mock_ping.side_effect = [MockResponse(400, json.dumps(response))] transmission = StixTransmission("trendmicro_vision_one", CONNECTION, CONFIG) ping_response = transmission.ping() self.assertFalse(ping_response["success"]) self.assertEqual(ping_response["code"], "invalid_parameter")
def test_ping_auth_failure(self, mock_ping): response = { "error": { "code": "Unauthorized", "message": "No authorization token was found", "innererror": { "code": "InvalidToken", "innererror": { "code": "CredentialsRequired", "service": "svp" } } } } mock_ping.side_effect = [MockResponse(401, json.dumps(response))] transmission = StixTransmission("trendmicro_vision_one", CONNECTION, CONFIG) ping_response = transmission.ping() self.assertFalse(ping_response["success"]) self.assertEqual(ping_response["code"], "authentication_fail")