More tests

Author: aaltat

atest/DynamicTypesAnnotationsLibrary.py

@@ -1,6 +1,6 @@
 from enum import Enum
 from functools import wraps
-from typing import List, Union, NewType, Optional, Tuple
+from typing import List, Union, NewType, Optional, Tuple, Dict
 
 from robot.api import logger
 
@@ -152,3 +152,7 @@ def keyword_with_deco_and_signature(self, arg1: bool = False, arg2: bool = False
     @keyword
     def keyword_optional_with_none(self, arg: Optional[str] = None):
         return f"arg: {arg}, type: {type(arg)}"
+
+    @keyword
+    def keyword_union_with_none(self, arg: Union[None, Dict, str] = None):
+        return f"arg: {arg}, type: {type(arg)}"

atest/tests_types.robot

@@ -2,6 +2,9 @@
 Library     DynamicTypesLibrary.py
 Library     DynamicTypesAnnotationsLibrary.py    xxx
 
+*** Variables ***
+${CUSTOM NONE} =    ${None}
+
 *** Test Cases ***
 Keyword Default Argument As Abject None
     ${return} =    DynamicTypesLibrary.Keyword None    ${None}
@@ -74,3 +77,17 @@ Type Conversion With Optional And None
     ${types} =    Keyword Optional With None    ${None}
     Should Contain    ${types}    arg: None,
     Should Contain    ${types}    <class 'NoneType'>
+    ${types} =    Keyword Optional With None    arg=${CUSTOM NONE}
+    Should Contain    ${types}    arg: None,
+    Should Contain    ${types}    <class 'NoneType'>
+
+Type Conversion With Union And Multiple Types
+    ${types} =    Keyword Union With None
+    Should Contain    ${types}    arg: None,
+    Should Contain    ${types}    <class 'NoneType'>
+    ${types} =    Keyword Union With None    None
+    Should Contain    ${types}    arg: None,
+    Should Contain    ${types}    <class 'str'>
+    ${types} =    Keyword Union With None    {"key": 1}
+    Should Contain    ${types}    arg: {"key": 1},
+    Should Contain    ${types}    <class 'str'>

utest/test_get_keyword_types.py

@@ -186,3 +186,8 @@ def test_keyword_with_decorator_arguments(lib_types):
 def test_keyword_optional_with_none(lib_types):
     types = lib_types.get_keyword_types('keyword_optional_with_none')
     assert types == {'arg': typing.Union[str, type(None)]}
+
+
+def test_keyword_union_with_none(lib_types):
+    types = lib_types.get_keyword_types('keyword_union_with_none')
+    assert types == {'arg': typing.Union[type(None), typing.Dict, str]}