Refactor AAA.checkUser()

[nemo.git] / nemo-impl / src / test / java / org / opendaylight / nemo / user / transactionmanager / TransactionBeginTest.java

diff --git a/nemo-impl/src/test/java/org/opendaylight/nemo/user/transactionmanager/TransactionBeginTest.java b/nemo-impl/src/test/java/org/opendaylight/nemo/user/transactionmanager/TransactionBeginTest.java
index ed8642c2e4c184e0e7de4539f1e51e5c73ff7783..ca5eeeacefad37c516a95c19856ef005d0234150 100644 (file)
--- a/nemo-impl/src/test/java/org/opendaylight/nemo/user/transactionmanager/TransactionBeginTest.java
+++ b/nemo-impl/src/test/java/org/opendaylight/nemo/user/transactionmanager/TransactionBeginTest.java
@@ -23,11 +23,8 @@ import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
 import org.opendaylight.nemo.user.tenantmanager.AAA;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.common.rev151010.UserId;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.common.rev151010.UserName;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.common.rev151010.UserPassword;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.common.rev151010.UserRoleName;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.BeginTransactionInput;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.user.rev151010.UserInstance;
 
 /**
  * Created by zhangmeng on 2015/11/20.
@@ -43,10 +40,9 @@ public class TransactionBeginTest extends TestCase {
     public void testTransactionbegin() throws Exception {
         AAA aaa = mock(AAA.class);
         BeginTransactionInput input = mock(BeginTransactionInput.class);
-        when(aaa.CheckUser(any(UserId.class),any(UserName.class),any(UserPassword.class),any(UserRoleName.class)))
-                .thenReturn(new String("test"));
+        when(aaa.checkUser(any(UserInstance.class))).thenReturn(new String("test"));
         String flag = transactionBegin.transactionbegin(aaa,input);
-        verify(aaa).CheckUser(any(UserId.class),any(UserName.class),any(UserPassword.class),any(UserRoleName.class));
+        verify(aaa).checkUser(any(UserInstance.class));
         Assert.assertEquals("test",flag);
     }
 }
\ No newline at end of file