Fix up release old producers to avoid memory leak
[mdsal.git] / dom / mdsal-dom-inmemory-datastore / src / test / java / org / opendaylight / mdsal / dom / store / inmemory / TestUtils.java
index 3af5bfa469da7e0575d5c9d35985802e7d58c79a..768eaba4f4f7c0849e00f44e7a2a0330729a89c0 100644 (file)
@@ -17,8 +17,6 @@ import java.util.Map;
 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
 import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteCursor;
-import org.opendaylight.mdsal.dom.spi.shard.DOMDataTreeShardProducer;
-import org.opendaylight.mdsal.dom.spi.shard.DOMDataTreeShardWriteTransaction;
 import org.opendaylight.mdsal.dom.spi.shard.ForeignShardModificationContext;
 import org.opendaylight.mdsal.dom.spi.shard.WriteCursorStrategy;
 import org.opendaylight.mdsal.dom.spi.shard.WriteableModificationNode;
@@ -54,10 +52,11 @@ final class TestUtils {
 
     static final WriteCursorStrategy WRITE_CURSOR_STRATEGY = mock(WriteCursorStrategy.class);
 
-    static final DOMDataTreeShardProducer DOM_DATA_TREE_SHARD_PRODUCER = mock(DOMDataTreeShardProducer.class);
+    static final InMemoryDOMDataTreeShardProducer DOM_DATA_TREE_SHARD_PRODUCER =
+            mock(InMemoryDOMDataTreeShardProducer.class);
 
-    static final DOMDataTreeShardWriteTransaction DOM_DATA_TREE_SHARD_WRITE_TRANSACTION =
-            mock(DOMDataTreeShardWriteTransaction.class);
+    static final InmemoryDOMDataTreeShardWriteTransaction DOM_DATA_TREE_SHARD_WRITE_TRANSACTION =
+            mock(InmemoryDOMDataTreeShardWriteTransaction.class);
 
     static final DataTree DATA_TREE = mock(DataTree.class);