southbound/southbound-impl/src/test/java/org/opendaylight/ovsdb/southbound/transactions/md/OvsdbOperationalCommandAggregatorTest.java

   1 /*
   2  * Copyright (c) 2015 Inocybe Technologies and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.ovsdb.southbound.transactions.md;
  10
  11 import static org.junit.Assert.*;
  12 import static org.mockito.Mockito.*;
  13
  14 import java.util.ArrayList;
  15 import java.util.List;
  16
  17 import org.junit.Before;
  18 import org.junit.Test;
  19 import org.junit.runner.RunWith;
  20 import org.mockito.Mockito;
  21 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
  22 import org.opendaylight.ovsdb.lib.message.TableUpdates;
  23 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
  24 import org.opendaylight.ovsdb.southbound.OvsdbConnectionInstance;
  25 import org.powermock.api.mockito.PowerMockito;
  26 import org.powermock.api.support.membermodification.MemberModifier;
  27 import org.powermock.core.classloader.annotations.PrepareForTest;
  28 import org.powermock.modules.junit4.PowerMockRunner;
  29 import org.powermock.reflect.Whitebox;
  30
  31 @PrepareForTest({})
  32 @RunWith(PowerMockRunner.class)
  33 public class OvsdbOperationalCommandAggregatorTest {
  34     private static final int NUMBER_OF_COMMANDS = 9;
  35     private List<TransactionCommand> commands = new ArrayList<>();
  36     private OvsdbOperationalCommandAggregator ovsdbOperationalCommandAggregator;
  37
  38     @Before
  39     public void setUp() throws Exception {
  40         ovsdbOperationalCommandAggregator = PowerMockito.mock(OvsdbOperationalCommandAggregator.class, Mockito.CALLS_REAL_METHODS);
  41
  42         //mock commands field
  43         commands.add(mock(OpenVSwitchUpdateCommand.class));
  44         commands.add(mock(OvsdbManagersUpdateCommand.class));
  45         commands.add(mock(OvsdbManagersRemovedCommand.class));
  46         commands.add(mock(OvsdbBridgeUpdateCommand.class));
  47         commands.add(mock(OvsdbBridgeRemovedCommand.class));
  48         commands.add(mock(OvsdbControllerUpdateCommand.class));
  49         commands.add(mock(OvsdbControllerRemovedCommand.class));
  50         commands.add(mock(OvsdbPortUpdateCommand.class));
  51         commands.add(mock(OvsdbPortRemoveCommand.class));
  52         MemberModifier.field(OvsdbOperationalCommandAggregator.class, "commands").set(ovsdbOperationalCommandAggregator, commands);
  53     }
  54
  55     @Test
  56     public void testOvsdbOperationalCommandAggregator() throws Exception {
  57         OvsdbConnectionInstance key = mock(OvsdbConnectionInstance.class);
  58         TableUpdates updates = mock(TableUpdates.class);
  59         DatabaseSchema dbSchema = mock(DatabaseSchema.class);
  60         OvsdbOperationalCommandAggregator ovsdbOperationalCommandAggregator1 = new OvsdbOperationalCommandAggregator(key, updates, dbSchema);
  61         List<TransactionCommand> testCommands = Whitebox.getInternalState(ovsdbOperationalCommandAggregator1, "commands");
  62         assertEquals(NUMBER_OF_COMMANDS, testCommands.size());
  63     }
  64
  65     @Test
  66     public void testExecute() {
  67         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
  68         for (TransactionCommand command: commands) {
  69             doNothing().when(command).execute(any(ReadWriteTransaction.class));
  70         }
  71         ovsdbOperationalCommandAggregator.execute(transaction);
  72         for (TransactionCommand command: commands) {
  73             verify(command).execute(any(ReadWriteTransaction.class));
  74         }
  75     }
  76 }