nemo-impl/src/test/java/org/opendaylight/nemo/user/transactionmanager/TransactionEndTest.java

   1 /*\r
   2  * Copyright (c) 2015 Huawei, Inc. and others. All rights reserved.\r
   3  *\r
   4  * This program and the accompanying materials are made available under the\r
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,\r
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html\r
   7  */\r
   8 package user.transactionmanager;\r
   9 \r
  10 import junit.framework.TestCase;\r
  11 import org.junit.Assert;\r
  12 import org.junit.Before;\r
  13 import org.junit.Test;\r
  14 import org.opendaylight.nemo.user.tenantmanager.AAA;\r
  15 import org.opendaylight.nemo.user.transactionmanager.TransactionEnd;\r
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.EndTransactionInput;\r
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.common.rev151010.UserId;\r
  18 import static org.junit.Assert.*;\r
  19 import static org.mockito.Mockito.*;\r
  20 /**\r
  21  * Created by zhangmeng on 2015/12/15.\r
  22  */\r
  23 public class TransactionEndTest extends TestCase {\r
  24     private AAA aaa;\r
  25     private EndTransactionInput input;\r
  26     private TransactionEnd transactionEnd;\r
  27     @Before\r
  28     public void setUp() throws Exception {\r
  29         aaa = mock(AAA.class);\r
  30         input = mock(EndTransactionInput.class);\r
  31 \r
  32         transactionEnd = new TransactionEnd();\r
  33     }\r
  34 \r
  35     @Test\r
  36     public void testTransactionend() throws Exception {\r
  37         UserId userId = mock(UserId.class);\r
  38 \r
  39         when(aaa.checkUser(any(UserId.class))).thenReturn(new String("test"));\r
  40         when(input.getUserId()).thenReturn(userId);\r
  41 \r
  42         Assert.assertTrue(transactionEnd.transactionend(aaa,input).equals("test"));\r
  43     }\r
  44 }