Merge pull request #462 from awslabs/fix-rdklib-test-templates

remove hardcoded rule name from rdklib unit tests
awslabs · Sep 27, 2023 · 396f58b · 396f58b
2 parents 9da09ad + 8c2284d
commit 396f58b
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 10 deletions.
diff --git a/rdk/template/runtime/python3.10-lib/rule_test.py b/rdk/template/runtime/python3.10-lib/rule_test.py
@@ -19,8 +19,8 @@
 # Main Code #
 #############
 
-MODULE = __import__("check_security_hub_aggregator")
-RULE = MODULE.check_security_hub_aggregator()
+MODULE = __import__('<%RuleName%>')
+RULE = MODULE.<%RuleName%>()
 
 CLIENT_FACTORY = MagicMock()
 

diff --git a/rdk/template/runtime/python3.11-lib/rule_test.py b/rdk/template/runtime/python3.11-lib/rule_test.py
@@ -19,8 +19,8 @@
 # Main Code #
 #############
 
-MODULE = __import__("check_security_hub_aggregator")
-RULE = MODULE.check_security_hub_aggregator()
+MODULE = __import__('<%RuleName%>')
+RULE = MODULE.<%RuleName%>()
 
 CLIENT_FACTORY = MagicMock()
 

diff --git a/rdk/template/runtime/python3.7-lib/rule_test.py b/rdk/template/runtime/python3.7-lib/rule_test.py
@@ -19,8 +19,8 @@
 # Main Code #
 #############
 
-MODULE = __import__("check_security_hub_aggregator")
-RULE = MODULE.check_security_hub_aggregator()
+MODULE = __import__('<%RuleName%>')
+RULE = MODULE.<%RuleName%>()
 
 CLIENT_FACTORY = MagicMock()
 

diff --git a/rdk/template/runtime/python3.8-lib/rule_test.py b/rdk/template/runtime/python3.8-lib/rule_test.py
@@ -19,8 +19,8 @@
 # Main Code #
 #############
 
-MODULE = __import__("check_security_hub_aggregator")
-RULE = MODULE.check_security_hub_aggregator()
+MODULE = __import__('<%RuleName%>')
+RULE = MODULE.<%RuleName%>()
 
 CLIENT_FACTORY = MagicMock()
 

diff --git a/rdk/template/runtime/python3.9-lib/rule_test.py b/rdk/template/runtime/python3.9-lib/rule_test.py
@@ -19,8 +19,8 @@
 # Main Code #
 #############
 
-MODULE = __import__("check_security_hub_aggregator")
-RULE = MODULE.check_security_hub_aggregator()
+MODULE = __import__('<%RuleName%>')
+RULE = MODULE.<%RuleName%>()
 
 CLIENT_FACTORY = MagicMock()