nemo-impl/src/test/java/org/opendaylight/nemo/user/vnspacemanager/structurestyle/updateintent/UpdateFlowTest.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 org.opendaylight.nemo.user.vnspacemanager.structurestyle.updateintent;\r
   9 import org.opendaylight.nemo.user.vnspacemanager.structurestyle.updateintent.UpdateFlow;\r
  10 \r
  11 import static org.mockito.Mockito.mock;\r
  12 import static org.mockito.Mockito.*;\r
  13 import org.junit.Assert;\r
  14 import org.junit.Before;\r
  15 import org.junit.Test;\r
  16 \r
  17 import org.opendaylight.controller.md.sal.binding.api.DataBroker;\r
  18 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;\r
  19 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;\r
  20 import org.opendaylight.nemo.user.tenantmanager.TenantManage;\r
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.common.rev151010.UserId;\r
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.Users;\r
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.user.intent.Objects;\r
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.user.intent.objects.Flow;\r
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.user.intent.objects.FlowBuilder;\r
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.user.intent.objects.FlowKey;\r
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.users.User;\r
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.users.UserKey;\r
  29 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;\r
  30 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;\r
  31 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;\r
  32 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;\r
  33 import com.google.common.util.concurrent.CheckedFuture;\r
  34 import com.google.common.util.concurrent.FutureCallback;\r
  35 import com.google.common.util.concurrent.Futures;\r
  36 import org.slf4j.Logger;\r
  37 import org.slf4j.LoggerFactory;\r
  38 \r
  39 public class UpdateFlowTest {\r
  40 \r
  41     private UpdateFlow updateFlow;\r
  42 \r
  43     private DataBroker dataBroker;\r
  44     private TenantManage tenantManage;\r
  45     private UserId userId;\r
  46     private Flow flow;\r
  47     \r
  48     @org.junit.Before\r
  49     public void setUp() throws Exception {\r
  50         dataBroker = mock(DataBroker.class);\r
  51         tenantManage = mock(TenantManage.class);\r
  52         userId = mock(UserId.class);\r
  53         flow = mock(Flow.class);\r
  54         \r
  55         updateFlow = new UpdateFlow(dataBroker, tenantManage);\r
  56     }\r
  57 \r
  58     @org.junit.Test\r
  59     public void testFlowHandling() throws Exception {\r
  60         \r
  61         String errorDefinition = new String("The match item has not been defined.");\r
  62         \r
  63         CheckedFuture connectiondefinitionFuture = mock(CheckedFuture.class);\r
  64         ReadOnlyTransaction readOnlyTransaction = mock(ReadOnlyTransaction.class);\r
  65         when(dataBroker.newReadOnlyTransaction()).thenReturn(readOnlyTransaction);\r
  66         when(readOnlyTransaction.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class))).thenReturn(connectiondefinitionFuture); \r
  67          \r
  68         Assert.assertEquals(updateFlow.FlowHandling(userId,flow),errorDefinition);\r
  69         \r
  70     }\r
  71 }