- }
-
- /**
- * Test method for {@link org.opendaylight.unimgr.cli.UniAddShellCommand#getSpeed()}.
- * @throws Exception
- */
- @Test
- public void testgetSpeed() throws Exception {
- // Test 10M
- Whitebox.setInternalState(uniAddShellCommand, "speed", "10M");
- final Speed10M speed10M = mock(Speed10M.class);
- final Speed10MBuilder speed10MBuilder = mock(Speed10MBuilder.class);
- when(speed10MBuilder.build()).thenReturn(speed10M);
- PowerMockito.whenNew(Speed10MBuilder.class).withNoArguments().thenReturn(speed10MBuilder);
- Object getSpeed = Whitebox.invokeMethod(uniAddShellCommand, "getSpeed");
- assertEquals(speed10M, getSpeed);
-
- // Test 100M
- Whitebox.setInternalState(uniAddShellCommand, "speed", "100M");
- final Speed100M speedObject100M = mock(Speed100M.class);
- final Speed100MBuilder speed100M = mock(Speed100MBuilder.class);
- PowerMockito.whenNew(Speed100MBuilder.class).withNoArguments().thenReturn(speed100M);
- when(speed100M.build()).thenReturn(speedObject100M);
- getSpeed = Whitebox.invokeMethod(uniAddShellCommand, "getSpeed");
- assertEquals(speedObject100M, getSpeed);
-
- // Test 1G
- Whitebox.setInternalState(uniAddShellCommand, "speed", "1G");
- final Speed1G speedObject1G = mock(Speed1G.class);
- final Speed1GBuilder speed1G = mock(Speed1GBuilder.class);
- PowerMockito.whenNew(Speed1GBuilder.class).withNoArguments().thenReturn(speed1G);
- when(speed1G.build()).thenReturn(speedObject1G);
- getSpeed = Whitebox.invokeMethod(uniAddShellCommand, "getSpeed");
- assertEquals(speedObject1G, getSpeed);
-
- // Test 10G
- Whitebox.setInternalState(uniAddShellCommand, "speed", "10G");
- final Speed10G speedObject10G = mock(Speed10G.class);
- final Speed10GBuilder speed10G = mock(Speed10GBuilder.class);
- PowerMockito.whenNew(Speed10GBuilder.class).withNoArguments().thenReturn(speed10G);
- when(speed10G.build()).thenReturn(speedObject10G);
- getSpeed = Whitebox.invokeMethod(uniAddShellCommand, "getSpeed");
- assertEquals(speedObject10G, getSpeed);
-
- // Test other
- Whitebox.setInternalState(uniAddShellCommand, "speed", "other");
- getSpeed = Whitebox.invokeMethod(uniAddShellCommand, "getSpeed");
- assertEquals(null, getSpeed);