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.mockito.Matchers.any;
11 import static org.mockito.Mockito.doReturn;
12 import com.google.common.base.Charsets;
13 import com.google.common.base.MoreObjects;
14 import com.google.common.base.Optional;
15 import com.google.common.util.concurrent.CheckedFuture;
16 import java.io.ByteArrayInputStream;
17 import java.io.IOException;
18 import java.io.InputStream;
19 import java.util.concurrent.ExecutionException;
20 import java.util.concurrent.TimeUnit;
21 import org.junit.Assert;
22 import org.junit.Before;
23 import org.junit.Test;
24 import org.junit.runner.RunWith;
25 import org.mockito.Mock;
26 import org.mockito.runners.MockitoJUnitRunner;
27 import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceException;
28 import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceRepresentation;
29 import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
30 import org.opendaylight.yangtools.yang.model.repo.api.YangSchemaSourceRepresentation;
31 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
32 import org.opendaylight.yangtools.yang.model.repo.spi.PotentialSchemaSource;
33 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceProvider;
34 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceRegistration;
35 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceRegistry;
37 @RunWith(MockitoJUnitRunner.class)
38 public class InMemorySchemaSourceCacheTest {
40 private static final Class<YangSchemaSourceRepresentation> representation = YangSchemaSourceRepresentation.class;
41 private static final long lifetime = 1000l;
42 private static final TimeUnit units = TimeUnit.MILLISECONDS;
45 private SchemaSourceRegistry registry;
47 private SchemaSourceRegistration<?> registration;
50 public void setUp() throws Exception {
51 doReturn(this.registration).when(this.registry).registerSchemaSource(any(SchemaSourceProvider.class),
52 any(PotentialSchemaSource.class));
56 public void inMemorySchemaSourceCacheTest1() {
57 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache = InMemorySchemaSourceCache
58 .createSoftCache(this.registry, InMemorySchemaSourceCacheTest.representation);
59 Assert.assertNotNull(inMemorySchemaSourceCache);
60 inMemorySchemaSourceCache.close();
64 public void inMemorySchemaSourceCacheTest2() {
65 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache = InMemorySchemaSourceCache
66 .createSoftCache(this.registry, InMemorySchemaSourceCacheTest.representation,
67 InMemorySchemaSourceCacheTest.lifetime, InMemorySchemaSourceCacheTest.units);
68 Assert.assertNotNull(inMemorySchemaSourceCache);
69 inMemorySchemaSourceCache.close();
73 public void inMemorySchemaSourceCacheOfferAndGetSourcestest() throws Exception {
74 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache = InMemorySchemaSourceCache
75 .createSoftCache(this.registry, InMemorySchemaSourceCacheTest.representation);
76 final String content = "content";
77 final YangTextSchemaSource source = new TestingYangSource("test", "2012-12-12", content);
78 inMemorySchemaSourceCache.offer(source);
79 final SourceIdentifier sourceIdentifier = new SourceIdentifier("test", "2012-12-12");
80 final CheckedFuture<? extends YangSchemaSourceRepresentation, SchemaSourceException> checkedSource = inMemorySchemaSourceCache
81 .getSource(sourceIdentifier);
82 Assert.assertNotNull(checkedSource);
83 final YangSchemaSourceRepresentation yangSchemaSourceRepresentation = checkedSource.get();
84 Assert.assertNotNull(yangSchemaSourceRepresentation);
85 Assert.assertEquals(sourceIdentifier, yangSchemaSourceRepresentation.getIdentifier());
86 inMemorySchemaSourceCache.close();
89 @Test(expected = ExecutionException.class)
90 public void inMemorySchemaSourceCacheNullGetSourcestest() throws Exception {
91 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache = InMemorySchemaSourceCache
92 .createSoftCache(this.registry, InMemorySchemaSourceCacheTest.representation);
93 final SourceIdentifier sourceIdentifier = new SourceIdentifier("test", "2012-12-12");
94 final CheckedFuture<? extends YangSchemaSourceRepresentation, SchemaSourceException> checkedSource = inMemorySchemaSourceCache
95 .getSource(sourceIdentifier);
96 Assert.assertNotNull(checkedSource);
98 inMemorySchemaSourceCache.close();
102 public void inMemorySchemaSourceCache3test() throws Exception {
103 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache = InMemorySchemaSourceCache
104 .createSoftCache(this.registry, InMemorySchemaSourceCacheTest.representation);
106 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache2 = InMemorySchemaSourceCache
107 .createSoftCache(this.registry, representation, InMemorySchemaSourceCacheTest.lifetime,
108 InMemorySchemaSourceCacheTest.units);
110 final String content = "content";
111 final YangTextSchemaSource source = new TestingYangSource("test", "2012-12-12", content);
112 inMemorySchemaSourceCache.offer(source);
113 inMemorySchemaSourceCache2.offer(source);
115 final SourceIdentifier sourceIdentifier = new SourceIdentifier("test", "2012-12-12");
116 final CheckedFuture<? extends YangSchemaSourceRepresentation, SchemaSourceException> checkedSource = inMemorySchemaSourceCache
117 .getSource(sourceIdentifier);
118 final CheckedFuture<? extends SchemaSourceRepresentation, SchemaSourceException> checkedSource2 = inMemorySchemaSourceCache2
119 .getSource(sourceIdentifier);
120 Assert.assertNotNull(checkedSource);
121 Assert.assertNotNull(checkedSource2);
123 Assert.assertEquals(checkedSource.get(), checkedSource2.get());
124 inMemorySchemaSourceCache.close();
125 inMemorySchemaSourceCache2.close();
128 private class TestingYangSource extends YangTextSchemaSource {
130 private final String content;
132 protected TestingYangSource(final String name, final String revision, final String content) {
133 super(new SourceIdentifier(name, Optional.fromNullable(revision)));
134 this.content = content;
138 protected MoreObjects.ToStringHelper addToStringAttributes(final MoreObjects.ToStringHelper toStringHelper) {
139 return toStringHelper;
143 public InputStream openStream() throws IOException {
144 return new ByteArrayInputStream(this.content.getBytes(Charsets.UTF_8));