2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.yangtools.yang.model.repo.util;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertNotNull;
12 import static org.junit.Assert.assertThrows;
13 import static org.mockito.ArgumentMatchers.any;
14 import static org.mockito.Mockito.doNothing;
15 import static org.mockito.Mockito.doReturn;
17 import com.google.common.base.MoreObjects;
18 import com.google.common.util.concurrent.ListenableFuture;
19 import java.io.ByteArrayInputStream;
20 import java.io.IOException;
21 import java.io.InputStream;
22 import java.nio.charset.StandardCharsets;
23 import java.util.concurrent.ExecutionException;
24 import java.util.concurrent.TimeUnit;
25 import org.junit.Before;
26 import org.junit.Test;
27 import org.junit.runner.RunWith;
28 import org.mockito.Mock;
29 import org.mockito.junit.MockitoJUnitRunner;
30 import org.opendaylight.yangtools.yang.common.Revision;
31 import org.opendaylight.yangtools.yang.model.repo.api.RevisionSourceIdentifier;
32 import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceRepresentation;
33 import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
34 import org.opendaylight.yangtools.yang.model.repo.api.YangSchemaSourceRepresentation;
35 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
36 import org.opendaylight.yangtools.yang.model.repo.spi.PotentialSchemaSource;
37 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceProvider;
38 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceRegistration;
39 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceRegistry;
41 @RunWith(MockitoJUnitRunner.StrictStubs.class)
42 public class InMemorySchemaSourceCacheTest {
43 public static final Class<YangSchemaSourceRepresentation> REPRESENTATION = YangSchemaSourceRepresentation.class;
44 public static final long LIFETIME = 1000L;
45 public static final TimeUnit UNITS = TimeUnit.MILLISECONDS;
48 public SchemaSourceRegistry registry;
50 public SchemaSourceRegistration<?> registration;
53 public void setUp() throws Exception {
54 doNothing().when(registration).close();
55 doReturn(registration).when(registry).registerSchemaSource(any(SchemaSourceProvider.class),
56 any(PotentialSchemaSource.class));
60 public void inMemorySchemaSourceCacheTest1() {
61 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache =
62 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION);
63 assertNotNull(inMemorySchemaSourceCache);
64 inMemorySchemaSourceCache.close();
68 public void inMemorySchemaSourceCacheTest2() {
69 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache =
70 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION, LIFETIME, UNITS);
71 assertNotNull(inMemorySchemaSourceCache);
72 inMemorySchemaSourceCache.close();
76 public void inMemorySchemaSourceCacheOfferAndGetSourcestest() throws Exception {
77 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache =
78 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION);
79 final String content = "content";
80 final YangTextSchemaSource source = new TestingYangSource("test", "2012-12-12", content);
81 inMemorySchemaSourceCache.offer(source);
82 final SourceIdentifier sourceIdentifier = RevisionSourceIdentifier.create("test", Revision.of("2012-12-12"));
83 final ListenableFuture<? extends YangSchemaSourceRepresentation> checkedSource = inMemorySchemaSourceCache
84 .getSource(sourceIdentifier);
85 assertNotNull(checkedSource);
86 final YangSchemaSourceRepresentation yangSchemaSourceRepresentation = checkedSource.get();
87 assertNotNull(yangSchemaSourceRepresentation);
88 assertEquals(sourceIdentifier, yangSchemaSourceRepresentation.getIdentifier());
89 inMemorySchemaSourceCache.close();
93 public void inMemorySchemaSourceCacheNullGetSourcestest() throws Exception {
94 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache =
95 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION);
96 final SourceIdentifier sourceIdentifier = RevisionSourceIdentifier.create("test", Revision.of("2012-12-12"));
97 final ListenableFuture<? extends YangSchemaSourceRepresentation> checkedSource =
98 inMemorySchemaSourceCache.getSource(sourceIdentifier);
99 assertNotNull(checkedSource);
102 assertThrows(ExecutionException.class, () -> checkedSource.get());
103 inMemorySchemaSourceCache.close();
107 public void inMemorySchemaSourceCache3test() throws InterruptedException, ExecutionException {
108 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache =
109 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION);
110 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache2 =
111 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION, LIFETIME, UNITS);
113 final String content = "content";
114 final YangTextSchemaSource source = new TestingYangSource("test", "2012-12-12", content);
115 inMemorySchemaSourceCache.offer(source);
116 inMemorySchemaSourceCache2.offer(source);
118 final SourceIdentifier sourceIdentifier = RevisionSourceIdentifier.create("test", Revision.of("2012-12-12"));
119 final ListenableFuture<? extends YangSchemaSourceRepresentation> checkedSource =
120 inMemorySchemaSourceCache.getSource(sourceIdentifier);
121 final ListenableFuture<? extends SchemaSourceRepresentation> checkedSource2 =
122 inMemorySchemaSourceCache2.getSource(sourceIdentifier);
123 assertNotNull(checkedSource);
124 assertNotNull(checkedSource2);
126 assertEquals(checkedSource.get(), checkedSource2.get());
127 inMemorySchemaSourceCache.close();
128 inMemorySchemaSourceCache2.close();
131 private class TestingYangSource extends YangTextSchemaSource {
133 private final String content;
135 protected TestingYangSource(final String name, final String revision, final String content) {
136 super(RevisionSourceIdentifier.create(name, Revision.ofNullable(revision)));
137 this.content = content;
141 protected MoreObjects.ToStringHelper addToStringAttributes(final MoreObjects.ToStringHelper toStringHelper) {
142 return toStringHelper;
146 public InputStream openStream() throws IOException {
147 return new ByteArrayInputStream(this.content.getBytes(StandardCharsets.UTF_8));