Added missing argument in Jython snippets. Closes #324

Author: aslakhellesoy

History.md

@@ -1,5 +1,6 @@
 ## [Git master](https://github.com/cucumber/cucumber-jvm/compare/v1.0.7...master)
 
+* [Jython] Added missing `self` argument in Jython snippets. ([#324](https://github.com/cucumber/cucumber-jvm/issues/324) Aslak Hellesøy)
 * [Scala] Fixed regression from v1.0.6 in Scala module - glue code wasn't loaded at all. ([#321](https://github.com/cucumber/cucumber-jvm/issues/321) Aslak Hellesøy)
 
 ## [1.0.8](https://github.com/cucumber/cucumber-jvm/compare/v1.0.7...v1.0.8)

jython/src/main/java/cucumber/runtime/jython/JythonSnippet.java

@@ -24,7 +24,8 @@ public String tableHint() {
 
     @Override
     public String arguments(List<Class<?>> argumentTypes) {
-        return untypedArguments(argumentTypes);
+        String args = untypedArguments(argumentTypes);
+        return args.equals("") ? "self" : "self, " + args;
     }
 
     @Override

jython/src/test/java/cucumber/runtime/jython/JythonSnippetTest.java

@@ -12,16 +12,27 @@
 public class JythonSnippetTest {
 
     @Test
-    public void generatesPlainSnippet() {
+    public void generatesSnippetWithTwoArgs() {
         String expected = "" +
                 "@Given('^I have (\\d+) cukes in my \"([^\"]*)\" belly$')\n" +
-                "def I_have_cukes_in_my_belly(arg1, arg2):\n" +
+                "def I_have_cukes_in_my_belly(self, arg1, arg2):\n" +
                 "  # Express the Regexp above with the code you wish you had\n" +
                 "  raise(PendingException())\n" +
                 "";
         assertEquals(expected, snippetFor("I have 4 cukes in my \"big\" belly"));
     }
 
+    @Test
+    public void generatesSnippetWithZeroArgs() {
+        String expected = "" +
+                "@Given('^I have no cukes belly$')\n" +
+                "def I_have_no_cukes_belly(self):\n" +
+                "  # Express the Regexp above with the code you wish you had\n" +
+                "  raise(PendingException())\n" +
+                "";
+        assertEquals(expected, snippetFor("I have no cukes belly"));
+    }
+
     private String snippetFor(String name) {
         Step step = new Step(Collections.<Comment>emptyList(), "Given ", name, 0, null, null);
         return new SnippetGenerator(new JythonSnippet()).getSnippet(step);