--- /dev/null
+/*
+ * Copyright (c) 2017 Pantheon Technologies s.r.o. 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.controller.cluster.databroker;
+
+import java.lang.reflect.Field;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.MockitoAnnotations;
+import org.opendaylight.controller.cluster.access.concepts.ClientIdentifier;
+import org.opendaylight.controller.cluster.access.concepts.FrontendIdentifier;
+import org.opendaylight.controller.cluster.access.concepts.FrontendType;
+import org.opendaylight.controller.cluster.access.concepts.LocalHistoryIdentifier;
+import org.opendaylight.controller.cluster.access.concepts.MemberName;
+import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
+import org.opendaylight.controller.cluster.databroker.actors.dds.AbstractClientHandle;
+import org.opendaylight.controller.cluster.databroker.actors.dds.ClientLocalHistory;
+import org.opendaylight.controller.cluster.databroker.actors.dds.ClientSnapshot;
+import org.opendaylight.controller.cluster.databroker.actors.dds.ClientTransaction;
+import org.opendaylight.controller.cluster.databroker.actors.dds.DataStoreClient;
+import org.opendaylight.controller.cluster.datastore.DatastoreContext;
+import org.opendaylight.controller.cluster.datastore.utils.ActorContext;
+import org.opendaylight.controller.md.cluster.datastore.model.TestModel;
+import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadTransaction;
+import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadWriteTransaction;
+import org.opendaylight.controller.sal.core.spi.data.DOMStoreTransactionChain;
+import org.opendaylight.controller.sal.core.spi.data.DOMStoreWriteTransaction;
+import org.opendaylight.yangtools.yang.model.api.SchemaContext;
+
+public class ClientBackedDataStoreTest {
+
+ private static final ClientIdentifier UNKNOWN_ID = ClientIdentifier.create(
+ FrontendIdentifier.create(MemberName.forName("local"), FrontendType.forName("unknown")), 0);
+
+ private static FrontendIdentifier FRONTEND_IDENTIFIER = FrontendIdentifier.create(
+ MemberName.forName("member"), FrontendType.forName("frontend"));
+ private static final ClientIdentifier CLIENT_IDENTIFIER = ClientIdentifier.create(FRONTEND_IDENTIFIER, 0);
+
+ private static LocalHistoryIdentifier HISTORY_ID = new LocalHistoryIdentifier(CLIENT_IDENTIFIER, 0);
+ private static final TransactionIdentifier TRANSACTION_IDENTIFIER = new TransactionIdentifier(HISTORY_ID, 0);
+
+ @Mock
+ private DataStoreClient clientActor;
+
+ @Mock
+ private ActorContext actorContext;
+
+ @Mock
+ private ClientLocalHistory clientLocalHistory;
+
+ @Mock
+ private ClientTransaction clientTransaction;
+
+ @Mock
+ private ClientSnapshot clientSnapshot;
+
+ @Before
+ public void setUp() throws Exception {
+ MockitoAnnotations.initMocks(this);
+
+ final SchemaContext schemaContext = TestModel.createTestContext();
+
+ Mockito.when(actorContext.getSchemaContext()).thenReturn(schemaContext);
+ Mockito.when(actorContext.getDatastoreContext()).thenReturn(DatastoreContext.newBuilder().build());
+
+ final Field transactionIdField = AbstractClientHandle.class.getDeclaredField("transactionId");
+ transactionIdField.setAccessible(true);
+
+ // set transaction ids to mocked objects
+ transactionIdField.set(clientTransaction, TRANSACTION_IDENTIFIER);
+ transactionIdField.set(clientSnapshot, TRANSACTION_IDENTIFIER);
+
+ Mockito.when(clientActor.getIdentifier()).thenReturn(CLIENT_IDENTIFIER);
+ Mockito.when(clientActor.createTransaction()).thenReturn(clientTransaction);
+ Mockito.when(clientActor.createLocalHistory()).thenReturn(clientLocalHistory);
+ Mockito.when(clientActor.createSnapshot()).thenReturn(clientSnapshot);
+ }
+
+ @Test
+ public void testCreateTransactionChain() throws Exception {
+ try (final ClientBackedDataStore clientBackedDataStore = new ClientBackedDataStore(
+ actorContext, UNKNOWN_ID, clientActor)) {
+ final DOMStoreTransactionChain txChain = clientBackedDataStore.createTransactionChain();
+ Assert.assertNotNull(txChain);
+ Mockito.verify(clientActor, Mockito.times(1)).createLocalHistory();
+ }
+ }
+
+ @Test
+ public void testNewReadOnlyTransaction() throws Exception {
+ try (final ClientBackedDataStore clientBackedDataStore = new ClientBackedDataStore(
+ actorContext, UNKNOWN_ID, clientActor)) {
+ final DOMStoreReadTransaction tx = clientBackedDataStore.newReadOnlyTransaction();
+ Assert.assertNotNull(tx);
+ Mockito.verify(clientActor, Mockito.times(1)).createSnapshot();
+ }
+ }
+
+ @Test
+ public void testNewWriteOnlyTransaction() throws Exception {
+ try (final ClientBackedDataStore clientBackedDataStore = new ClientBackedDataStore(
+ actorContext, UNKNOWN_ID, clientActor)) {
+ final DOMStoreWriteTransaction tx = clientBackedDataStore.newWriteOnlyTransaction();
+ Assert.assertNotNull(tx);
+ Mockito.verify(clientActor, Mockito.times(1)).createTransaction();
+ }
+ }
+
+ @Test
+ public void testNewReadWriteTransaction() throws Exception {
+ try (final ClientBackedDataStore clientBackedDataStore = new ClientBackedDataStore(
+ actorContext, UNKNOWN_ID, clientActor)) {
+ final DOMStoreReadWriteTransaction tx = clientBackedDataStore.newReadWriteTransaction();
+ Assert.assertNotNull(tx);
+ Mockito.verify(clientActor, Mockito.times(1)).createTransaction();
+ }
+ }
+}
\ No newline at end of file