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