Merge pull request #493 from alexandrehassan/484-test-pick

Add tests for @pick

Author: fliiiix

tests/exploratory/pick/features/pick.feature

@@ -0,0 +1,10 @@
+Feature: Verify the pick feature
+    
+    Scenario: Add a method to the world using pick
+        Given I don't have a pick decorated method
+        Then I expect the pick step not to be in the world
+        Given I have a method decorated with pick
+        Then I expect the pick step to be in the world
+        When I call the pick decorated method
+        Then I expect the pick result to be 42
+

tests/exploratory/pick/radish/steps.py

@@ -0,0 +1,39 @@
+from radish.stepregistry import given, step, then, when
+from radish.terrain import pick, world
+
+
+@step("I don't have a pick decorated method")
+def dont_have_pick_method(step):
+    if hasattr(world, "pick_decorated"):
+        delattr(world, "pick_decorated")
+
+
+@given("I have a method decorated with pick")
+def add_pick_step(step):
+    @pick
+    def pick_decorated():
+        return 42
+
+    step.context.decorate_method_name = pick_decorated.__name__
+
+
+@then("I expect the pick step to be in the world")
+def expect_pick_step(step):
+    assert hasattr(world, "pick_decorated"), "World is missing the decorated method pick_decorated"
+
+
+@then("I expect the pick step not to be in the world")
+def expect_pick_step_missing(step):
+    assert not hasattr(world, "pick_decorated"), "World has the decorated method pick_decorated"
+
+
+@when("I call the pick decorated method")
+def call_pick_decorated_method(step):
+    step.context.pick_result = world.pick_decorated()
+
+
+@then("I expect the pick result to be {expected_result:d}")
+def expect_pick_result(step, expected_result):
+    assert step.context.pick_result == expected_result, (
+        f"Expected pick result to be {expected_result} but was {step.context.pick_result}"
+    )

tests/unit/test_terrain.py

@@ -0,0 +1,29 @@
+from radish.terrain import pick, world
+
+
+def test_pick_decorator_adds_function_to_world():
+    """Test that the pick decorator adds the function to the world object."""
+    assert not hasattr(world, "sample_function")
+
+    @pick
+    def sample_function():
+        return "test"
+
+    assert hasattr(world, "sample_function")
+    assert world.sample_function() == "test"
+
+    @pick
+    def another_function(x, y=0):
+        return x + y
+
+    assert hasattr(world, "another_function")
+    assert world.another_function(2, y=3) == 5, "Parameters not passed correctly."
+
+    def func():
+        return "overridden"
+
+    func.__name__ = "sample_function"
+
+    pick(func)
+
+    assert world.sample_function() == "overridden", "The function was not overridden correctly."