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;
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.nio.charset.StandardCharsets;
20 import java.util.concurrent.ExecutionException;
21 import java.util.concurrent.TimeUnit;
22 import org.junit.Assert;
23 import org.junit.Before;
24 import org.junit.Test;
25 import org.junit.runner.RunWith;
26 import org.mockito.Mock;
27 import org.mockito.runners.MockitoJUnitRunner;
28 import org.opendaylight.yangtools.yang.model.repo.api.RevisionSourceIdentifier;
29 import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceException;
30 import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceRepresentation;
31 import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
32 import org.opendaylight.yangtools.yang.model.repo.api.YangSchemaSourceRepresentation;
33 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
34 import org.opendaylight.yangtools.yang.model.repo.spi.PotentialSchemaSource;
35 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceProvider;
36 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceRegistration;
37 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceRegistry;
39 @RunWith(MockitoJUnitRunner.class)
40 public class InMemorySchemaSourceCacheTest {
42 private static final Class<YangSchemaSourceRepresentation> REPRESENTATION = YangSchemaSourceRepresentation.class;
43 private static final long LIFETIME = 1000L;
44 private static final TimeUnit UNITS = TimeUnit.MILLISECONDS;
47 private SchemaSourceRegistry registry;
49 private SchemaSourceRegistration<?> registration;
52 public void setUp() throws Exception {
53 doReturn(this.registration).when(this.registry).registerSchemaSource(any(SchemaSourceProvider.class),
54 any(PotentialSchemaSource.class));
58 public void inMemorySchemaSourceCacheTest1() {
59 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache =
60 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION);
61 Assert.assertNotNull(inMemorySchemaSourceCache);
62 inMemorySchemaSourceCache.close();
66 public void inMemorySchemaSourceCacheTest2() {
67 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache =
68 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION, LIFETIME, UNITS);
69 Assert.assertNotNull(inMemorySchemaSourceCache);
70 inMemorySchemaSourceCache.close();
74 public void inMemorySchemaSourceCacheOfferAndGetSourcestest() throws Exception {
75 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache =
76 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION);
77 final String content = "content";
78 final YangTextSchemaSource source = new TestingYangSource("test", "2012-12-12", content);
79 inMemorySchemaSourceCache.offer(source);
80 final SourceIdentifier sourceIdentifier = RevisionSourceIdentifier.create("test", "2012-12-12");
81 final CheckedFuture<? extends YangSchemaSourceRepresentation, SchemaSourceException> checkedSource =
82 inMemorySchemaSourceCache.getSource(sourceIdentifier);
83 Assert.assertNotNull(checkedSource);
84 final YangSchemaSourceRepresentation yangSchemaSourceRepresentation = checkedSource.get();
85 Assert.assertNotNull(yangSchemaSourceRepresentation);
86 Assert.assertEquals(sourceIdentifier, yangSchemaSourceRepresentation.getIdentifier());
87 inMemorySchemaSourceCache.close();
90 @Test(expected = ExecutionException.class)
91 public void inMemorySchemaSourceCacheNullGetSourcestest() throws Exception {
92 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache =
93 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION);
94 final SourceIdentifier sourceIdentifier = RevisionSourceIdentifier.create("test", "2012-12-12");
95 final CheckedFuture<? extends YangSchemaSourceRepresentation, SchemaSourceException> checkedSource =
96 inMemorySchemaSourceCache.getSource(sourceIdentifier);
97 Assert.assertNotNull(checkedSource);
99 inMemorySchemaSourceCache.close();
103 public void inMemorySchemaSourceCache3test() throws Exception {
104 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache =
105 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION);
106 final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache2 =
107 InMemorySchemaSourceCache.createSoftCache(this.registry, REPRESENTATION, LIFETIME, UNITS);
109 final String content = "content";
110 final YangTextSchemaSource source = new TestingYangSource("test", "2012-12-12", content);
111 inMemorySchemaSourceCache.offer(source);
112 inMemorySchemaSourceCache2.offer(source);
114 final SourceIdentifier sourceIdentifier = RevisionSourceIdentifier.create("test", "2012-12-12");
115 final CheckedFuture<? extends YangSchemaSourceRepresentation, SchemaSourceException> checkedSource =
116 inMemorySchemaSourceCache.getSource(sourceIdentifier);
117 final CheckedFuture<? extends SchemaSourceRepresentation, SchemaSourceException> checkedSource2 =
118 inMemorySchemaSourceCache2.getSource(sourceIdentifier);
119 Assert.assertNotNull(checkedSource);
120 Assert.assertNotNull(checkedSource2);
122 Assert.assertEquals(checkedSource.get(), checkedSource2.get());
123 inMemorySchemaSourceCache.close();
124 inMemorySchemaSourceCache2.close();
127 private class TestingYangSource extends YangTextSchemaSource {
129 private final String content;
131 protected TestingYangSource(final String name, final String revision, final String content) {
132 super(RevisionSourceIdentifier.create(name, Optional.fromNullable(revision)));
133 this.content = content;
137 protected MoreObjects.ToStringHelper addToStringAttributes(final MoreObjects.ToStringHelper toStringHelper) {
138 return toStringHelper;
142 public InputStream openStream() throws IOException {
143 return new ByteArrayInputStream(this.content.getBytes(StandardCharsets.UTF_8));