typing test_generic_protocols

Author: MaxNumerique

src/tests/test_generic_protocols.py

@@ -1,7 +1,8 @@
+from typing import Callable
 from opengeodeweb_viewer.rpc.generic.generic_protocols import VtkGenericView
 
 
-def test_register_mesh(server, dataset_factory):
+def test_register_mesh(server: "ServerMonitor", dataset_factory: Callable[..., str]) -> None:
     data_id = "123456789"
     dataset_factory(id=data_id, viewable_file_name="hat.vtp", geode_object="mesh")
 
@@ -13,7 +14,7 @@ def test_register_mesh(server, dataset_factory):
     assert server.compare_image(3, "mesh/register.jpeg") is True
 
 
-def test_register_model(server, dataset_factory):
+def test_register_model(server: "ServerMonitor", dataset_factory: Callable[..., str]) -> None:
     data_id = "123456789"
     dataset_factory(id=data_id, viewable_file_name="CrossSection.vtm")
 
@@ -25,7 +26,7 @@ def test_register_model(server, dataset_factory):
     assert server.compare_image(3, "model/register.jpeg") is True
 
 
-def test_deregister_mesh(server, dataset_factory):
+def test_deregister_mesh(server: "ServerMonitor", dataset_factory: Callable[..., str]) -> None:
     test_register_mesh(server, dataset_factory)
 
     server.call(
@@ -36,7 +37,7 @@ def test_deregister_mesh(server, dataset_factory):
     assert server.compare_image(3, "mesh/deregister.jpeg") == True
 
 
-def test_deregister_model(server, dataset_factory):
+def test_deregister_model(server: "ServerMonitor", dataset_factory: Callable[..., str]) -> None:
     test_register_model(server, dataset_factory)
 
     server.call(