* ============LICENSE_END=========================================================\r
*/\r
\r
-\r
package com.att.nsa.cambria.service.impl;\r
\r
import static org.junit.Assert.*;\r
\r
import java.io.IOException;\r
\r
-import com.att.nsa.cambria.beans.DMaaPContext;\r
+import com.att.dmf.mr.backends.ConsumerFactory;\r
+import com.att.dmf.mr.beans.DMaaPContext;\r
+import com.att.dmf.mr.security.DMaaPAuthenticatorImpl;\r
+import com.att.dmf.mr.service.impl.AdminServiceImpl;\r
+import com.att.dmf.mr.utils.ConfigurationReader;\r
+import com.att.dmf.mr.utils.DMaaPResponseBuilder;\r
import com.att.nsa.configs.ConfigDbException;\r
+import com.att.nsa.limits.Blacklist;\r
import com.att.nsa.security.ReadWriteSecuredResource.AccessDeniedException;\r
+import com.att.nsa.security.db.simple.NsaSimpleApiKey;\r
\r
import org.junit.After;\r
import org.junit.Before;\r
import org.junit.Test;\r
+import org.junit.runner.RunWith;\r
+import org.mockito.InjectMocks;\r
+import org.mockito.Mock;\r
+import org.mockito.MockitoAnnotations;\r
+import org.powermock.api.mockito.PowerMockito;\r
+import org.powermock.core.classloader.annotations.PrepareForTest;\r
+import org.powermock.modules.junit4.PowerMockRunner;\r
\r
+@RunWith(PowerMockRunner.class)\r
+@PrepareForTest({ DMaaPAuthenticatorImpl.class, DMaaPResponseBuilder.class })\r
public class AdminServiceImplemTest {\r
\r
+ @InjectMocks\r
+ AdminServiceImpl adminServiceImpl;\r
+\r
+ @Mock\r
+ DMaaPContext dmaapContext;\r
+ @Mock\r
+ ConsumerFactory factory;\r
+\r
+ @Mock\r
+ ConfigurationReader configReader;\r
+ @Mock\r
+ Blacklist Blacklist;\r
+\r
@Before\r
public void setUp() throws Exception {\r
+\r
+ MockitoAnnotations.initMocks(this);\r
+ PowerMockito.mockStatic(DMaaPAuthenticatorImpl.class);\r
+ NsaSimpleApiKey user = new NsaSimpleApiKey("admin", "password");\r
+\r
+ PowerMockito.when(dmaapContext.getConfigReader()).thenReturn(configReader);\r
+ PowerMockito.when(configReader.getfConsumerFactory()).thenReturn(factory);\r
+ PowerMockito.when(configReader.getfIpBlackList()).thenReturn(Blacklist);\r
+\r
+ PowerMockito.when(DMaaPAuthenticatorImpl.getAuthenticatedUser(dmaapContext)).thenReturn(user);\r
+ PowerMockito.mockStatic(DMaaPResponseBuilder.class);\r
}\r
\r
@After\r
public void tearDown() throws Exception {\r
}\r
\r
- \r
- //ISSUES WITH AUTHENTICATION\r
+ // ISSUES WITH AUTHENTICATION\r
@Test\r
public void testShowConsumerCache() {\r
- \r
- AdminServiceImpl adminServiceImpl = new AdminServiceImpl();\r
+\r
try {\r
- adminServiceImpl.showConsumerCache(new DMaaPContext());\r
+ adminServiceImpl.showConsumerCache(dmaapContext);\r
} catch (IOException | AccessDeniedException e) {\r
// TODO Auto-generated catch block\r
e.printStackTrace();\r
} catch (NullPointerException e) {\r
// TODO Auto-generated catch block\r
- //e.printStackTrace();\r
+ // e.printStackTrace();\r
assertTrue(true);\r
}\r
- \r
- \r
+\r
String trueValue = "True";\r
assertTrue(trueValue.equalsIgnoreCase("True"));\r
- \r
- \r
+\r
}\r
- \r
+\r
@Test\r
public void testDropConsumerCache() {\r
- \r
- AdminServiceImpl adminServiceImpl = new AdminServiceImpl();\r
+\r
try {\r
- adminServiceImpl.dropConsumerCache(new DMaaPContext());\r
+ adminServiceImpl.dropConsumerCache(dmaapContext);\r
} catch (IOException | AccessDeniedException e) {\r
// TODO Auto-generated catch block\r
e.printStackTrace();\r
} catch (NullPointerException e) {\r
// TODO Auto-generated catch block\r
- //e.printStackTrace();\r
+ // e.printStackTrace();\r
assertTrue(true);\r
}\r
- \r
- \r
+\r
String trueValue = "True";\r
assertTrue(trueValue.equalsIgnoreCase("True"));\r
- \r
- \r
+\r
}\r
- \r
+\r
@Test\r
public void testGetBlacklist() {\r
- \r
- AdminServiceImpl adminServiceImpl = new AdminServiceImpl();\r
+\r
try {\r
- adminServiceImpl.getBlacklist(new DMaaPContext());\r
+ adminServiceImpl.getBlacklist(dmaapContext);\r
} catch (IOException | AccessDeniedException e) {\r
// TODO Auto-generated catch block\r
e.printStackTrace();\r
} catch (NullPointerException e) {\r
// TODO Auto-generated catch block\r
- //e.printStackTrace();\r
+ // e.printStackTrace();\r
assertTrue(true);\r
}\r
- \r
- \r
+\r
String trueValue = "True";\r
assertTrue(trueValue.equalsIgnoreCase("True"));\r
- \r
- \r
+\r
}\r
- \r
+\r
@Test\r
public void testAddToBlacklist() {\r
- \r
- AdminServiceImpl adminServiceImpl = new AdminServiceImpl();\r
+\r
try {\r
- adminServiceImpl.addToBlacklist(new DMaaPContext(), "120.120.120.120");\r
+ adminServiceImpl.addToBlacklist(dmaapContext, "120.120.120.120");\r
} catch (IOException | AccessDeniedException | ConfigDbException e) {\r
// TODO Auto-generated catch block\r
e.printStackTrace();\r
} catch (NullPointerException e) {\r
// TODO Auto-generated catch block\r
- //e.printStackTrace();\r
+ // e.printStackTrace();\r
assertTrue(true);\r
}\r
- \r
- \r
+\r
String trueValue = "True";\r
assertTrue(trueValue.equalsIgnoreCase("True"));\r
- \r
- \r
+\r
}\r
- \r
+\r
@Test\r
public void testRemoveFromBlacklist() {\r
- \r
- AdminServiceImpl adminServiceImpl = new AdminServiceImpl();\r
+\r
try {\r
- adminServiceImpl.addToBlacklist(new DMaaPContext(), "120.120.120.120");\r
+ adminServiceImpl.removeFromBlacklist(dmaapContext, "120.120.120.120");\r
} catch (IOException | AccessDeniedException | ConfigDbException e) {\r
// TODO Auto-generated catch block\r
e.printStackTrace();\r
} catch (NullPointerException e) {\r
// TODO Auto-generated catch block\r
- //e.printStackTrace();\r
+ // e.printStackTrace();\r
assertTrue(true);\r
}\r
- \r
- \r
+\r
String trueValue = "True";\r
assertTrue(trueValue.equalsIgnoreCase("True"));\r
- \r
- \r
+\r
}\r
- \r
- \r
\r
}\r