diff options
Diffstat (limited to 'src/test/java/com/beust/jcommander/SimpleExample.java')
-rw-r--r-- | src/test/java/com/beust/jcommander/SimpleExample.java | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/src/test/java/com/beust/jcommander/SimpleExample.java b/src/test/java/com/beust/jcommander/SimpleExample.java new file mode 100644 index 0000000..b8000ca --- /dev/null +++ b/src/test/java/com/beust/jcommander/SimpleExample.java @@ -0,0 +1,50 @@ +package com.beust.jcommander; + +import org.testng.annotations.BeforeMethod; +import org.testng.annotations.Test; + +import java.io.PrintWriter; +import java.io.StringWriter; + +import static org.testng.Assert.assertEquals; + +class Main { + static PrintWriter out; + @Parameter(names={"--length", "-l"}) + int length; + @Parameter(names={"--pattern", "-p"}) + int pattern; + + public static void main(String ... args) { + Main main = new Main(); + new JCommander(main, args); + main.run(); + } + + public void run() { + out.printf("%d %d", length, pattern); + } +} + +public class SimpleExample { + StringWriter out; + + @BeforeMethod + public void setupMain(){ + out=new StringWriter(); + Main.out=new PrintWriter(out); + } + + @Test + public void testLongArgs() { + Main.main("--length", "512", "--pattern", "2"); + assertEquals("512 2", out.toString()); + } + + @Test + public void testShortArgs() { + Main.main("-l", "256", "-p", "171"); + assertEquals("256 171", out.toString()); + } + +} |