Remove use of Guava Charsets
[yangtools.git] / yang / yang-model-util / src / test / java / org / opendaylight / yangtools / yang / model / repo / util / InMemorySchemaSourceCacheTest.java
1 /*
2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
3  *
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
7  */
8 package org.opendaylight.yangtools.yang.model.repo.util;
9
10 import static org.mockito.Matchers.any;
11 import static org.mockito.Mockito.doReturn;
12
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;
38
39 @RunWith(MockitoJUnitRunner.class)
40 public class InMemorySchemaSourceCacheTest {
41
42     private static final Class<YangSchemaSourceRepresentation> representation = YangSchemaSourceRepresentation.class;
43     private static final long lifetime = 1000l;
44     private static final TimeUnit units = TimeUnit.MILLISECONDS;
45
46     @Mock
47     private SchemaSourceRegistry registry;
48     @Mock
49     private SchemaSourceRegistration<?> registration;
50
51     @Before
52     public void setUp() throws Exception {
53         doReturn(this.registration).when(this.registry).registerSchemaSource(any(SchemaSourceProvider.class),
54                 any(PotentialSchemaSource.class));
55     }
56
57     @Test
58     public void inMemorySchemaSourceCacheTest1() {
59         final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache = InMemorySchemaSourceCache
60                 .createSoftCache(this.registry, InMemorySchemaSourceCacheTest.representation);
61         Assert.assertNotNull(inMemorySchemaSourceCache);
62         inMemorySchemaSourceCache.close();
63     }
64
65     @Test
66     public void inMemorySchemaSourceCacheTest2() {
67         final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache = InMemorySchemaSourceCache
68                 .createSoftCache(this.registry, InMemorySchemaSourceCacheTest.representation,
69                         InMemorySchemaSourceCacheTest.lifetime, InMemorySchemaSourceCacheTest.units);
70         Assert.assertNotNull(inMemorySchemaSourceCache);
71         inMemorySchemaSourceCache.close();
72     }
73
74     @Test
75     public void inMemorySchemaSourceCacheOfferAndGetSourcestest() throws Exception {
76         final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache = InMemorySchemaSourceCache
77                 .createSoftCache(this.registry, InMemorySchemaSourceCacheTest.representation);
78         final String content = "content";
79         final YangTextSchemaSource source = new TestingYangSource("test", "2012-12-12", content);
80         inMemorySchemaSourceCache.offer(source);
81         final SourceIdentifier sourceIdentifier = RevisionSourceIdentifier.create("test", "2012-12-12");
82         final CheckedFuture<? extends YangSchemaSourceRepresentation, SchemaSourceException> checkedSource = inMemorySchemaSourceCache
83                 .getSource(sourceIdentifier);
84         Assert.assertNotNull(checkedSource);
85         final YangSchemaSourceRepresentation yangSchemaSourceRepresentation = checkedSource.get();
86         Assert.assertNotNull(yangSchemaSourceRepresentation);
87         Assert.assertEquals(sourceIdentifier, yangSchemaSourceRepresentation.getIdentifier());
88         inMemorySchemaSourceCache.close();
89     }
90
91     @Test(expected = ExecutionException.class)
92     public void inMemorySchemaSourceCacheNullGetSourcestest() throws Exception {
93         final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache = InMemorySchemaSourceCache
94                 .createSoftCache(this.registry, InMemorySchemaSourceCacheTest.representation);
95         final SourceIdentifier sourceIdentifier = RevisionSourceIdentifier.create("test", "2012-12-12");
96         final CheckedFuture<? extends YangSchemaSourceRepresentation, SchemaSourceException> checkedSource = inMemorySchemaSourceCache
97                 .getSource(sourceIdentifier);
98         Assert.assertNotNull(checkedSource);
99         checkedSource.get();
100         inMemorySchemaSourceCache.close();
101     }
102
103     @Test
104     public void inMemorySchemaSourceCache3test() throws Exception {
105         final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache = InMemorySchemaSourceCache
106                 .createSoftCache(this.registry, InMemorySchemaSourceCacheTest.representation);
107
108         final InMemorySchemaSourceCache<YangSchemaSourceRepresentation> inMemorySchemaSourceCache2 = InMemorySchemaSourceCache
109                 .createSoftCache(this.registry, representation, InMemorySchemaSourceCacheTest.lifetime,
110                         InMemorySchemaSourceCacheTest.units);
111
112         final String content = "content";
113         final YangTextSchemaSource source = new TestingYangSource("test", "2012-12-12", content);
114         inMemorySchemaSourceCache.offer(source);
115         inMemorySchemaSourceCache2.offer(source);
116
117         final SourceIdentifier sourceIdentifier = RevisionSourceIdentifier.create("test", "2012-12-12");
118         final CheckedFuture<? extends YangSchemaSourceRepresentation, SchemaSourceException> checkedSource = inMemorySchemaSourceCache
119                 .getSource(sourceIdentifier);
120         final CheckedFuture<? extends SchemaSourceRepresentation, SchemaSourceException> checkedSource2 = inMemorySchemaSourceCache2
121                 .getSource(sourceIdentifier);
122         Assert.assertNotNull(checkedSource);
123         Assert.assertNotNull(checkedSource2);
124
125         Assert.assertEquals(checkedSource.get(), checkedSource2.get());
126         inMemorySchemaSourceCache.close();
127         inMemorySchemaSourceCache2.close();
128     }
129
130     private class TestingYangSource extends YangTextSchemaSource {
131
132         private final String content;
133
134         protected TestingYangSource(final String name, final String revision, final String content) {
135             super(RevisionSourceIdentifier.create(name, Optional.fromNullable(revision)));
136             this.content = content;
137         }
138
139         @Override
140         protected MoreObjects.ToStringHelper addToStringAttributes(final MoreObjects.ToStringHelper toStringHelper) {
141             return toStringHelper;
142         }
143
144         @Override
145         public InputStream openStream() throws IOException {
146             return new ByteArrayInputStream(this.content.getBytes(StandardCharsets.UTF_8));
147         }
148     }
149
150 }