[#1563] add tests

remkop · Feb 10, 2022 · 2afd966 · 2afd966
1 parent 3468f19
commit 2afd966
Showing 1 changed file with 50 additions and 0 deletions.
diff --git a/picocli-spring-boot-starter/src/test/java/picocli/spring/PicocliSpringFactoryTest.java b/picocli-spring-boot-starter/src/test/java/picocli/spring/PicocliSpringFactoryTest.java
@@ -9,6 +9,7 @@
 import picocli.spring.boot.autoconfigure.sample.MyCommand;
 import picocli.spring.boot.autoconfigure.sample.MySpringApp;
 
+import java.lang.reflect.Field;
 import java.util.Arrays;
 
 import static org.junit.Assert.*;
@@ -83,4 +84,53 @@ private void load(Class<?> config, String... environment) {
         applicationContext.refresh();
         this.context = applicationContext;
     }
+
+    @Test(expected = NullPointerException.class)
+    public void testConstructorNullAppContext() {
+        new PicocliSpringFactory(null);
+    }
+
+    @Test(expected = NullPointerException.class)
+    public void testConstructorNullAppContextNullFallbackFactory() {
+        new PicocliSpringFactory(null, null);
+    }
+
+    @Test(expected = NullPointerException.class)
+    public void testConstructorNullAppContextValidFallbackFactory() {
+        new PicocliSpringFactory(null, CommandLine.defaultFactory());
+    }
+
+    @Test(expected = NullPointerException.class)
+    public void testConstructorValidAppContextNullFallbackFactory() {
+        load(MySpringApp.class);
+        new PicocliSpringFactory(context, null);
+    }
+
+    @Test
+    public void testConstructorAppContextUsesDefaultFactory() throws Exception {
+        load(MySpringApp.class);
+        PicocliSpringFactory factory = new PicocliSpringFactory(context);
+        IFactory fallbackFactory = extractFallbackFactory(factory);
+        assertNotNull(fallbackFactory);
+        assertEquals("picocli.CommandLine$DefaultFactory", fallbackFactory.getClass().getName());
+    }
+
+    @Test
+    public void testConstructorAppContextFactoryUsesSpecifiedFactory() throws Exception {
+        load(MySpringApp.class);
+        IFactory myFactory = new IFactory() {
+            @Override public <K> K create(Class<K> cls) {return null;}
+        };
+        PicocliSpringFactory factory = new PicocliSpringFactory(context, myFactory);
+        IFactory fallbackFactory = extractFallbackFactory(factory);
+        assertNotNull(fallbackFactory);
+        assertSame(myFactory, fallbackFactory);
+    }
+
+    private IFactory extractFallbackFactory(PicocliSpringFactory factory) throws NoSuchFieldException, IllegalAccessException {
+        Field fallbackFactoryField = PicocliSpringFactory.class.getDeclaredField("fallbackFactory");
+        fallbackFactoryField.setAccessible(true);
+        IFactory result = (IFactory) fallbackFactoryField.get(factory);
+        return result;
+    }
 }