change tests

Author: YunchuWang

tests/test_eventgrid.py

@@ -2,6 +2,7 @@
 # Licensed under the MIT License.
 
 from datetime import datetime
+import sys
 import unittest
 from typing import List
 
@@ -13,6 +14,10 @@ class MyTestCase(unittest.TestCase):
     def test_eventgrid_input_type(self):
         check_input_type = azf_event_grid.EventGridEventInConverter.\
             check_input_type_annotation
+        
+        if sys.version_info >= (3, 9):
+            self.assertTrue(check_input_type(list[func.EventGridEvent]))
+
         self.assertTrue(check_input_type(func.EventGridEvent))
         self.assertFalse(check_input_type(List[func.EventGridEvent]))
         self.assertFalse(check_input_type(str))
@@ -21,6 +26,11 @@ def test_eventgrid_input_type(self):
     def test_eventgrid_output_type(self):
         check_output_type = azf_event_grid.EventGridEventOutConverter.\
             check_output_type_annotation
+        
+        if sys.version_info >= (3, 9):
+            self.assertTrue(check_output_type(list[func.EventGridEvent]))
+            self.assertTrue(check_output_type(list[str]))
+        
         self.assertTrue(check_output_type(func.EventGridOutputEvent))
         self.assertTrue(check_output_type(List[func.EventGridOutputEvent]))
         self.assertTrue(check_output_type(str))

tests/test_eventhub.py

@@ -1,6 +1,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 
+import sys
 from typing import List, Mapping
 import unittest
 import json
@@ -21,6 +22,10 @@ def test_eventhub_input_type(self):
         check_input_type = (
             azf_eh.EventHubConverter.check_input_type_annotation
         )
+        if sys.version_info >= (3, 9):
+            self.assertTrue(check_input_type(list[func.EventGridEvent]))        
+            self.assertTrue(check_input_type(list[str]))    
+
         self.assertTrue(check_input_type(func.EventHubEvent))
         self.assertTrue(check_input_type(List[func.EventHubEvent]))
         self.assertFalse(check_input_type(str))
@@ -31,6 +36,11 @@ def test_eventhub_output_type(self):
         check_output_type = (
             azf_eh.EventHubTriggerConverter.check_output_type_annotation
         )
+
+        if sys.version_info >= (3, 9):
+            self.assertTrue(check_output_type(list[func.EventGridEvent]))        
+            self.assertTrue(check_output_type(list[str]))           
+        
         self.assertTrue(check_output_type(bytes))
         self.assertTrue(check_output_type(str))
         self.assertTrue(check_output_type(List[str]))

tests/test_kafka.py

@@ -1,6 +1,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 
+import sys
 from typing import List
 import unittest
 import json
@@ -33,6 +34,11 @@ def test_kafka_input_type(self):
         check_input_type = (
             azf_ka.KafkaConverter.check_input_type_annotation
         )
+
+        if sys.version_info >= (3, 9):
+            self.assertTrue(check_input_type(list[func.KafkaEvent]))
+            self.assertTrue(check_input_type(list[str]))
+                    
         self.assertTrue(check_input_type(func.KafkaEvent))
         self.assertTrue(check_input_type(List[func.KafkaEvent]))
         self.assertFalse(check_input_type(str))
@@ -43,6 +49,12 @@ def test_kafka_output_type(self):
         check_output_type = (
             azf_ka.KafkaTriggerConverter.check_output_type_annotation
         )
+
+        if sys.version_info >= (3, 9):
+            self.assertTrue(check_output_type(list[func.KafkaEvent]))
+            self.assertTrue(check_output_type(list[str]))
+            self.assertTrue(check_output_type(list[bytes]))
+
         self.assertTrue(check_output_type(bytes))
         self.assertTrue(check_output_type(str))
         self.assertTrue(check_output_type(List[str]))