1 files changed, 46 insertions, 0 deletions
diff --git a/odl-aaa-moon/aaa-authz/aaa-authz-service/src/test/java/org/opendaylight/aaa/authz/srv/AuthzConsumerContextImplTest.java b/odl-aaa-moon/aaa-authz/aaa-authz-service/src/test/java/org/opendaylight/aaa/authz/srv/AuthzConsumerContextImplTest.java
new file mode 100644
index 00000000..fb033341
--- /dev/null
+++ b/odl-aaa-moon/aaa-authz/aaa-authz-service/src/test/java/org/opendaylight/aaa/authz/srv/AuthzConsumerContextImplTest.java
@@ -0,0 +1,46 @@
+/*
+ * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.aaa.authz.srv;
+
+import org.junit.Assert;
+import org.junit.Before;
+import org.mockito.Mockito;
+import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
+import org.opendaylight.controller.sal.core.api.Broker;
+import org.opendaylight.controller.sal.core.api.Provider;
+
+public class AuthzConsumerContextImplTest {
+
+    private Broker.ConsumerSession realconsumercontext;
+    private Provider realprovidercontext;
+    private AuthzBrokerImpl authzBroker;
+    private Broker realbroker;
+
+    @Before
+    public void beforeTest() {
+        realconsumercontext = Mockito.mock(Broker.ConsumerSession.class);
+        realprovidercontext = Mockito.mock(Provider.class);
+        realbroker = Mockito.mock(Broker.class);
+        realbroker.registerProvider(realprovidercontext);
+        authzBroker = Mockito.mock(AuthzBrokerImpl.class);
+    }
+
+    @org.junit.Test
+    public void testGetService() throws Exception {
+        AuthzConsumerContextImpl authzConsumerContext = new AuthzConsumerContextImpl(
+                realconsumercontext, authzBroker);
+
+        Assert.assertEquals("Expected Authz session context",
+                authzConsumerContext.getService(DOMDataBroker.class).getClass(),
+                AuthzDomDataBroker.class);
+        // Assert.assertEquals("Expected Authz session context",
+        // authzConsumerContext.getService(SchemaService.class).getClass(),
+        // SchemaService.class);
+    }
+}
+\ No newline at end of file