11import os
22import sys
3+ from jsonschema import validate
34from unittest import TestCase
45from unittest .mock import patch , MagicMock
56
67sys .path .append (os .path .abspath ("../" ))
78from insightconnect_plugin_runtime .exceptions import PluginException
89from komand_mimecast .actions import DeleteBlockedSenderPolicy
9- from komand_mimecast .util .constants import BASIC_ASSISTANCE_MESSAGE , ERROR_CASES , MANAGED_URL_NOT_FOUND_ERROR
10+ from komand_mimecast .actions .delete_blocked_sender_policy .schema import (
11+ DeleteBlockedSenderPolicyOutput ,
12+ DeleteBlockedSenderPolicyInput ,
13+ )
1014
1115from util import Util
1216
@@ -17,12 +21,15 @@ class TestDeleteBlockedSenderPolicy(TestCase):
1721 def setUpClass (cls ) -> None :
1822 cls .action = Util .default_connector (DeleteBlockedSenderPolicy ())
1923
20- def test_delete_blocked_sender_policy (self , mocked_request : MagicMock ):
21- actual = self .action .run (Util .load_json ("inputs/delete_blocked_sender_policy.json.exp" ))
24+ def test_delete_blocked_sender_policy (self , _mocked_request : MagicMock ):
25+ input_data = Util .load_json ("inputs/delete_blocked_sender_policy.json.exp" )
26+ validate (input_data , DeleteBlockedSenderPolicyInput .schema )
27+ actual = self .action .run (input_data )
2228 expect = Util .load_json ("expected/delete_blocked_sender_policy.json.exp" )
2329 self .assertEqual (expect , actual )
30+ validate (actual , DeleteBlockedSenderPolicyOutput .schema )
2431
25- def test_bad_delete_blocked_sender_policy (self , mocked_request : MagicMock ):
32+ def test_bad_delete_blocked_sender_policy (self , _mocked_request : MagicMock ):
2633 with self .assertRaises (PluginException ) as exception :
2734 self .action .run (Util .load_json ("inputs/delete_blocked_sender_policy_bad.json.exp" ))
2835 self .assertEqual (exception .exception .cause , PluginException .causes [PluginException .Preset .NOT_FOUND ])
0 commit comments