X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=auth%2Fauth-cass%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Faaf%2Fauth%2Fdirect%2Ftest%2FJU_DirectRegistrar.java;h=833b8970ccbeefa5e74eb54b82cd5103079bbf98;hb=1296352d8eafee57f982a4342ad79ada4aa56d28;hp=1cc2410767635a63eed72c2dfaef6d472257ab4b;hpb=4c39dda98b8af6911c33085307f2d8eec4fc4837;p=aaf%2Fauthz.git

diff --git a/auth/auth-cass/src/test/java/org/onap/aaf/auth/direct/test/JU_DirectRegistrar.java b/auth/auth-cass/src/test/java/org/onap/aaf/auth/direct/test/JU_DirectRegistrar.java
index 1cc24107..833b8970 100644
--- a/auth/auth-cass/src/test/java/org/onap/aaf/auth/direct/test/JU_DirectRegistrar.java
+++ b/auth/auth-cass/src/test/java/org/onap/aaf/auth/direct/test/JU_DirectRegistrar.java
@@ -7,9 +7,9 @@
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
- * 
+ *
  *      http://www.apache.org/licenses/LICENSE-2.0
- * 
+ *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@@ -38,49 +38,76 @@ import org.onap.aaf.cadi.Access;
 import org.onap.aaf.cadi.CadiException;
 import org.onap.aaf.cadi.config.Config;
 
-
-@RunWith(MockitoJUnitRunner.class) 
+@RunWith(MockitoJUnitRunner.class)
 public class JU_DirectRegistrar {
 
-	@Mock
-	LocateDAO ldao;
-	
-//	@Mock
-//	Data locate;
-	
-	@Mock
-	Access access;
-	
-	@Mock
-	AuthzEnv env;
-	
-//	@Mock
-	AuthzTrans trans;
-	
-	@Before
-	public void setUp() throws Exception {
-		initMocks(this);
-	}
-	
-	@Test
-	public void testUpdate() {
-		DirectRegistrar registrarObj=null;
-		try {
-			Mockito.doReturn("20").when(access).getProperty(Config.CADI_LATITUDE,null);
-			Mockito.doReturn("20").when(access).getProperty(Config.CADI_LONGITUDE,null);
-			Mockito.doReturn("20").when(access).getProperty(Config.AAF_LOCATOR_CONTAINER,"");
-			registrarObj = new DirectRegistrar( access, ldao, 9080);
-		} catch (CadiException e) {
-			// TODO Auto-generated catch block
-			e.printStackTrace();
-		}
-		org.onap.aaf.auth.layer.Result<Void> retVal1 = new Result<Void>(null,0,"test",new String[0]);
-		Mockito.doReturn(trans).when(env).newTransNoAvg();
-//		Mockito.doReturn(retVal1).when(ldao).update(trans,locate);
-		registrarObj.update(env);
-//		System.out.println(retVal1);
-	}
-	
-	
-	
+    @Mock
+    LocateDAO ldao;
+
+    // @Mock
+    // Data locate;
+
+    @Mock
+    Access access;
+
+    @Mock
+    AuthzEnv env;
+
+    // @Mock
+    AuthzTrans trans;
+
+    @Before
+    public void setUp() throws Exception {
+        initMocks(this);
+    }
+
+    @Test
+    public void testUpdate() {
+        DirectRegistrar registrarObj = null;
+        org.onap.aaf.auth.layer.Result<Void> rs = null;
+        try {
+            Mockito.doReturn("20").when(access).getProperty(Config.CADI_LATITUDE, null);
+            Mockito.doReturn("20").when(access).getProperty(Config.CADI_LONGITUDE, null);
+            Mockito.doReturn("20").when(access).getProperty(Config.AAF_LOCATOR_CONTAINER, "");
+            Mockito.doReturn("20,test,test").when(access).getProperty(Config.AAF_LOCATOR_ENTRIES, "");
+            rs = new org.onap.aaf.auth.layer.Result<Void>(null, 0, "test",    new Object[0]);
+            Mockito.doReturn(rs).when(ldao).update(Mockito.any(), Mockito.any());
+            registrarObj = new DirectRegistrar(access, ldao, 9080);
+
+
+            org.onap.aaf.auth.layer.Result<Void> retVal1 = new Result<Void>(null, 0, "test", new Object[0]);
+            Mockito.doReturn(trans).when(env).newTransNoAvg();
+            // Mockito.doReturn(retVal1).when(ldao).update(trans,locate);
+            registrarObj.update(env);
+
+            rs = new org.onap.aaf.auth.layer.Result<Void>(null, 1, "test", new Object[0]);
+            Mockito.doReturn(rs).when(ldao).update(Mockito.any(), Mockito.any());
+            registrarObj = new DirectRegistrar(access, ldao, 9080);
+            registrarObj.update(env);
+        } catch (CadiException e) {
+            // TODO Auto-generated catch block
+            e.printStackTrace();
+        }
+        // System.out.println(retVal1);
+    }
+
+    @Test
+    public void testCancel() {
+        DirectRegistrar registrarObj = null;
+        try {
+            Mockito.doReturn("20").when(access).getProperty(Config.CADI_LATITUDE, null);
+            Mockito.doReturn("20").when(access).getProperty(Config.CADI_LONGITUDE, null);
+            Mockito.doReturn("20").when(access).getProperty(Config.AAF_LOCATOR_CONTAINER, "");
+            Mockito.doReturn("20,test,test").when(access).getProperty(Config.AAF_LOCATOR_ENTRIES, "");
+            registrarObj = new DirectRegistrar(access, ldao, 9080);
+
+            org.onap.aaf.auth.layer.Result<Void> rs = new org.onap.aaf.auth.layer.Result<Void>(null, 1, "test", new Object[0]);
+            Mockito.doReturn(rs).when(ldao).delete(Mockito.any(), Mockito.any(), Mockito.anyBoolean());
+        } catch (CadiException e) {
+            // TODO Auto-generated catch block
+            e.printStackTrace();
+        }
+         registrarObj.cancel(env);
+    }
+
 }
\ No newline at end of file