Update tests

sdk/python/cmd/pulumi-language-python/testdata/setuppy/sdks/simple-2.0.0/pulumi_simple/provider.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = ['ProviderArgs', 'Provider']

sdk/python/cmd/pulumi-language-python/testdata/setuppy/sdks/simple-2.0.0/pulumi_simple/resource.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = ['ResourceArgs', 'Resource']

sdk/python/cmd/pulumi-language-python/testdata/setuppy/sdks/simple-2.0.0/setup.py

@@ -32,6 +32,7 @@ def readme():
       install_requires=[
           'parver>=0.2.1',
           'pulumi',
-          'semver>=2.8.1'
+          'semver>=2.8.1',
+          'typing-extensions>=4.11; python_version < "3.11"'
       ],
       zip_safe=False)

sdk/python/cmd/pulumi-language-python/testdata/toml/sdks/simple-2.0.0/pulumi_simple/provider.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = ['ProviderArgs', 'Provider']

sdk/python/cmd/pulumi-language-python/testdata/toml/sdks/simple-2.0.0/pulumi_simple/resource.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = ['ResourceArgs', 'Resource']

sdk/python/cmd/pulumi-language-python/testdata/toml/sdks/simple-2.0.0/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
   name = "pulumi_simple"
-  dependencies = ["parver>=0.2.1", "pulumi", "semver>=2.8.1"]
+  dependencies = ["parver>=0.2.1", "pulumi", "semver>=2.8.1", "typing-extensions>=4.11; python_version < \"3.11\""]
   readme = "README.md"
   requires-python = ">=3.8"
   version = "2.0.0"

tests/testdata/codegen/assets-and-archives/python/pulumi_example/_inputs.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = [

tests/testdata/codegen/assets-and-archives/python/pulumi_example/get_assets.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = [

tests/testdata/codegen/assets-and-archives/python/pulumi_example/outputs.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = [

tests/testdata/codegen/assets-and-archives/python/pulumi_example/provider.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = ['ProviderArgs', 'Provider']

tests/testdata/codegen/assets-and-archives/python/pulumi_example/resource_with_assets.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 from . import outputs
 from ._inputs import *

tests/testdata/codegen/assets-and-archives/python/setup.py

@@ -32,6 +32,7 @@ def readme():
       install_requires=[
           'parver>=0.2.1',
           'pulumi',
-          'semver>=2.8.1'
+          'semver>=2.8.1',
+          'typing-extensions>=4.11; python_version < "3.11"'
       ],
       zip_safe=False)

tests/testdata/codegen/azure-native-nested-types/python/pulumi_azure_native/documentdb/outputs.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from .. import _utilities
 from . import outputs
 

tests/testdata/codegen/azure-native-nested-types/python/pulumi_azure_native/documentdb/sql_resource_sql_container.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from .. import _utilities
 from . import outputs
 

tests/testdata/codegen/azure-native-nested-types/python/pulumi_azure_native/provider.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = ['ProviderArgs', 'Provider']

tests/testdata/codegen/azure-native-nested-types/python/setup.py

@@ -39,6 +39,7 @@ def readme():
       install_requires=[
           'parver>=0.2.1',
           'pulumi>=3.0.0,<4.0.0',
-          'semver>=2.8.1'
+          'semver>=2.8.1',
+          'typing-extensions>=4.11; python_version < "3.11"'
       ],
       zip_safe=False)

tests/testdata/codegen/cyclic-types/python/pulumi_example/provider.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = ['ProviderArgs', 'Provider']

tests/testdata/codegen/cyclic-types/python/setup.py

@@ -32,6 +32,7 @@ def readme():
       install_requires=[
           'parver>=0.2.1',
           'pulumi',
-          'semver>=2.8.1'
+          'semver>=2.8.1',
+          'typing-extensions>=4.11; python_version < "3.11"'
       ],
       zip_safe=False)

tests/testdata/codegen/dash-named-schema/python/pulumi_foo_bar/_inputs.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = [

tests/testdata/codegen/dash-named-schema/python/pulumi_foo_bar/outputs.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = [

tests/testdata/codegen/dash-named-schema/python/pulumi_foo_bar/provider.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = ['ProviderArgs', 'Provider']

tests/testdata/codegen/dash-named-schema/python/pulumi_foo_bar/submodule1/foo_encrypted_bar_class.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from .. import _utilities
 
 __all__ = ['FOOEncryptedBarClassArgs', 'FOOEncryptedBarClass']

tests/testdata/codegen/dash-named-schema/python/pulumi_foo_bar/submodule1/module_resource.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from .. import _utilities
 from .. import _inputs as _root_inputs
 from .. import outputs as _root_outputs

tests/testdata/codegen/dash-named-schema/python/setup.py

@@ -32,6 +32,7 @@ def readme():
       install_requires=[
           'parver>=0.2.1',
           'pulumi',
-          'semver>=2.8.1'
+          'semver>=2.8.1',
+          'typing-extensions>=4.11; python_version < "3.11"'
       ],
       zip_safe=False)

tests/testdata/codegen/dashed-import-schema/python/pulumi_plant/_inputs.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 from ._enums import *
 

tests/testdata/codegen/dashed-import-schema/python/pulumi_plant/outputs.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 from ._enums import *
 

tests/testdata/codegen/dashed-import-schema/python/pulumi_plant/provider.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from . import _utilities
 
 __all__ = ['ProviderArgs', 'Provider']

tests/testdata/codegen/dashed-import-schema/python/pulumi_plant/tree/v1/nursery.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from ... import _utilities
 from ._enums import *
 

tests/testdata/codegen/dashed-import-schema/python/pulumi_plant/tree/v1/rubber_tree.py

@@ -4,9 +4,14 @@
 
 import copy
 import warnings
+import sys
 import pulumi
 import pulumi.runtime
 from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict
+else:
+    from typing_extensions import NotRequired, TypedDict
 from ... import _utilities
 from ... import _enums as _root_enums
 from ... import _inputs as _root_inputs