Skip to content

Commit 69feeb7

Browse files
committed
tests(plugin): improve type of fixture
1 parent 0a7263f commit 69feeb7

4 files changed

Lines changed: 6 additions & 6 deletions

File tree

packages/polywrap-plugin/tests/conftest.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,12 @@
66

77
@fixture
88
def get_greeting_module():
9-
class GreetingModule(PluginModule[Any, str]):
10-
def __init__(self, config: Any):
9+
class GreetingModule(PluginModule[None, str]):
10+
def __init__(self, config: None):
1111
super().__init__(config)
1212

1313
def greeting(self, args: Dict[str, Any], client: Invoker):
1414
return f"Greetings from: {args['name']}"
1515

16-
instance = GreetingModule({})
16+
instance = GreetingModule(None)
1717
return instance

packages/polywrap-plugin/tests/test_plugin_module.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from polywrap_plugin import PluginModule
88

99
@pytest.mark.asyncio
10-
async def test_plugin_module(get_greeting_module: PluginModule[Any, Any]):
10+
async def test_plugin_module(get_greeting_module: PluginModule[None, str]):
1111
module = get_greeting_module
1212

1313
client = PolywrapClient()

packages/polywrap-plugin/tests/test_plugin_package.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from polywrap_result import Ok
88

99
@pytest.mark.asyncio
10-
async def test_plugin_package_invoke(get_greeting_module: PluginModule[Any, Any]):
10+
async def test_plugin_package_invoke(get_greeting_module: PluginModule[None, str]):
1111
module = get_greeting_module
1212
manifest = cast(AnyWrapManifest, {})
1313
plugin_package = PluginPackage(module, manifest)

packages/polywrap-plugin/tests/test_plugin_wrapper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from polywrap_plugin import PluginWrapper, PluginModule
1010

1111
@pytest.mark.asyncio
12-
async def test_plugin_wrapper_invoke(get_greeting_module: PluginModule[Any, Any]):
12+
async def test_plugin_wrapper_invoke(get_greeting_module: PluginModule[None, str]):
1313
module = get_greeting_module
1414
manifest = cast(AnyWrapManifest, {})
1515

0 commit comments

Comments
 (0)