Fix NPE when creating a producer after shard was unregistered
[mdsal.git] / dom / mdsal-dom-broker / src / test / java / org / opendaylight / mdsal / dom / broker / MountPointServiceTest.java
1 /*
2  * Copyright (c) 2014 Cisco Systems, Inc. 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 package org.opendaylight.mdsal.dom.broker;
9
10 import static org.junit.Assert.assertFalse;
11 import static org.junit.Assert.assertSame;
12 import static org.junit.Assert.assertTrue;
13 import static org.mockito.Mockito.doNothing;
14 import static org.mockito.Mockito.mock;
15
16 import com.google.common.collect.ClassToInstanceMap;
17 import java.lang.reflect.Field;
18 import org.junit.Before;
19 import org.junit.Test;
20 import org.opendaylight.mdsal.dom.api.DOMMountPoint;
21 import org.opendaylight.mdsal.dom.api.DOMMountPointListener;
22 import org.opendaylight.mdsal.dom.api.DOMMountPointService;
23 import org.opendaylight.mdsal.dom.api.DOMMountPointService.DOMMountPointBuilder;
24 import org.opendaylight.mdsal.dom.api.DOMService;
25 import org.opendaylight.mdsal.dom.broker.DOMMountPointServiceImpl.DOMMountPointBuilderImpl;
26 import org.opendaylight.yangtools.concepts.ObjectRegistration;
27 import org.opendaylight.yangtools.yang.common.QName;
28 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
29 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
30
31 public class MountPointServiceTest {
32
33     private static DOMMountPointService mountService;
34     private static final YangInstanceIdentifier PATH =
35             YangInstanceIdentifier.of(QName.create("namespace", "12-12-2012", "top"));
36
37     @Before
38     public void setup() {
39         mountService = new DOMMountPointServiceImpl();
40     }
41
42     @Test
43     public void createSimpleMountPoint() throws Exception {
44         final DOMMountPointListener listener = mock(DOMMountPointListener.class);
45         doNothing().when(listener).onMountPointCreated(PATH);
46         mountService.registerProvisionListener(listener);
47
48         assertFalse(mountService.getMountPoint(PATH).isPresent());
49
50         mountService.createMountPoint(PATH).register();
51
52         assertTrue(mountService.getMountPoint(PATH).isPresent());
53     }
54
55     @Test
56     public void unregisterTest() throws Exception {
57         final DOMMountPointListener listener = mock(DOMMountPointListener.class);
58         doNothing().when(listener).onMountPointCreated(PATH);
59         doNothing().when(listener).onMountPointRemoved(PATH);
60         final DOMMountPointServiceImpl mountService = new DOMMountPointServiceImpl();
61         mountService.registerProvisionListener(listener);
62         mountService.createMountPoint(PATH).register();
63
64         assertTrue(mountService.getMountPoint(PATH).isPresent());
65
66         mountService.unregisterMountPoint(PATH);
67
68         assertFalse(mountService.getMountPoint(PATH).isPresent());
69     }
70
71     @Test
72     public void mountRegistrationTest() throws Exception {
73         final DOMMountPointBuilder mountBuilder = mountService.createMountPoint(PATH);
74         final ObjectRegistration<DOMMountPoint> objectRegistration = mountBuilder.register();
75
76         assertTrue(mountService.getMountPoint(PATH).isPresent());
77         assertSame(objectRegistration.getInstance(), mountService.getMountPoint(PATH).get());
78
79         objectRegistration.close();
80
81         assertFalse(mountService.getMountPoint(PATH).isPresent());
82     }
83
84     @Test
85     public void mountBuilderTest() throws Exception {
86         final DOMMountPointBuilderImpl mountBuilder = (DOMMountPointBuilderImpl) mountService.createMountPoint(PATH);
87         mountBuilder.register();
88
89         final SchemaContext mockSchemaContext = mock(SchemaContext.class);
90         mountBuilder.addInitialSchemaContext(mockSchemaContext);
91
92         final Field schemaContextField = DOMMountPointBuilderImpl.class.getDeclaredField("schemaContext");
93         schemaContextField.setAccessible(true);
94
95         final SchemaContext schemaContext = (SchemaContext) schemaContextField.get(mountBuilder);
96
97         assertSame(mockSchemaContext, schemaContext);
98
99         final Field servicesField = DOMMountPointBuilderImpl.class.getDeclaredField("services");
100         servicesField.setAccessible(true);
101
102         final ClassToInstanceMap<DOMService> services =
103                 (ClassToInstanceMap<DOMService>) servicesField.get(mountBuilder);
104         assertTrue(services.isEmpty());
105         assertFalse(services.containsKey(DOMService.class));
106         mountBuilder.addService(DOMService.class, null);
107         assertFalse(services.isEmpty());
108         assertTrue(services.containsKey(DOMService.class));
109     }
110 }