openstack/net-virt/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/impl/EventDispatcherImplTest.java

   1 /*
   2  * Copyright (c) 2015 Inocybe 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.openstack.netvirt.impl;
  10
  11 import static org.junit.Assert.assertEquals;
  12 import static org.junit.Assert.assertNotEquals;
  13 import static org.junit.Assert.assertNull;
  14 import static org.mockito.Mockito.mock;
  15 import static org.mockito.Mockito.when;
  16
  17 import java.lang.reflect.Field;
  18 import java.util.Random;
  19 import java.util.concurrent.BlockingQueue;
  20
  21 import org.junit.Before;
  22 import org.junit.Test;
  23 import org.junit.runner.RunWith;
  24 import org.mockito.InjectMocks;
  25 import org.mockito.Mock;
  26 import org.mockito.runners.MockitoJUnitRunner;
  27 import org.opendaylight.ovsdb.openstack.netvirt.AbstractEvent;
  28 import org.opendaylight.ovsdb.openstack.netvirt.AbstractHandler;
  29 import org.opendaylight.ovsdb.openstack.netvirt.api.Constants;
  30 import org.osgi.framework.ServiceReference;
  31
  32 /**
  33  * Unit test for {@link EventDispatcherImpl}
  34  */
  35 @RunWith(MockitoJUnitRunner.class)
  36 public class EventDispatcherImplTest {
  37
  38     @Mock AbstractHandler handler;
  39     @InjectMocks EventDispatcherImpl eventDispatcherImpl;
  40
  41     private AbstractEvent.HandlerType handlerTypeObject = AbstractEvent.HandlerType.NEUTRON_FLOATING_IP;
  42     private ServiceReference ref = mock(ServiceReference.class);
  43
  44     @Before
  45     public void setUp() {
  46         Random r = new Random();
  47
  48         eventDispatcherImpl.init();
  49         eventDispatcherImpl.start();
  50
  51         when(ref.getProperty(org.osgi.framework.Constants.SERVICE_ID)).thenReturn(r.nextLong());
  52         when(ref.getProperty(Constants.EVENT_HANDLER_TYPE_PROPERTY)).thenReturn(handlerTypeObject);
  53     }
  54
  55     /**
  56      * Test methods {@link EventDispatcherImpl#eventHandlerRemoved(ServiceReference)}
  57      * and {@link EventDispatcherImpl#eventHandlerAdded(ServiceReference, AbstractHandler)}
  58      */
  59     @Test
  60     public void testHandlerAddedAndRemoved() throws Exception{
  61         AbstractHandler[] handlers = ( AbstractHandler[]) getClassField("handlers");
  62
  63         assertNotEquals("Error, handler should be null", handlers[handlerTypeObject.ordinal()], handler);
  64
  65         eventDispatcherImpl.eventHandlerAdded(ref, handler);
  66
  67         assertEquals("Error, did not return the added handler", handlers[handlerTypeObject.ordinal()], handler);
  68
  69         eventDispatcherImpl.eventHandlerRemoved(ref);
  70
  71         assertNull("Error, handler should be null as it has just been removed", handlers[handlerTypeObject.ordinal()]);
  72     }
  73
  74     /**
  75      * Test method {@link EventDispatcherImpl#enqueueEvent(AbstractEvent)}
  76      */
  77     @Test
  78     public void testEnqueueEvent() throws Exception{
  79         BlockingQueue<AbstractEvent> events = (BlockingQueue<AbstractEvent>) getClassField("events");
  80
  81         assertEquals("Error, did not return the expected size, nothing has been added yet", 0, events.size());
  82
  83         eventDispatcherImpl.enqueueEvent(mock(AbstractEvent.class));
  84
  85         assertEquals("Error, did not return the expected size", 1, events.size());
  86     }
  87
  88     /**
  89      * Get the specified field from EventDispatcherImpl using reflection
  90      * @param fieldName - the field to retrieve
  91      * @return the desired field
  92      */
  93     private Object getClassField(String fieldName) throws NoSuchFieldException, SecurityException, IllegalArgumentException, IllegalAccessException{
  94         Field field = EventDispatcherImpl.class.getDeclaredField(fieldName);
  95         field.setAccessible(true);
  96         return field.get(eventDispatcherImpl);
  97     }
  98 }