opendaylight/md-sal/sal-clustering-commons/src/test/java/org/opendaylight/controller/cluster/schema/provider/impl/RemoteYangTextSourceProviderImplTest.java

   1 /*
   2  * Copyright (c) 2015 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.controller.cluster.schema.provider.impl;
   9
  10 import static org.junit.Assert.assertArrayEquals;
  11 import static org.junit.Assert.assertEquals;
  12 import static org.junit.Assert.assertTrue;
  13
  14 import com.google.common.io.ByteSource;
  15 import com.google.common.util.concurrent.Futures;
  16 import java.util.Collections;
  17 import java.util.Set;
  18 import org.junit.Before;
  19 import org.junit.Test;
  20 import org.mockito.Mockito;
  21 import org.opendaylight.yangtools.yang.common.Revision;
  22 import org.opendaylight.yangtools.yang.model.repo.api.RevisionSourceIdentifier;
  23 import org.opendaylight.yangtools.yang.model.repo.api.SchemaRepository;
  24 import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceException;
  25 import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
  26 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
  27 import scala.concurrent.Await;
  28 import scala.concurrent.Future;
  29 import scala.concurrent.duration.FiniteDuration;
  30
  31 public class RemoteYangTextSourceProviderImplTest {
  32
  33     private static final SourceIdentifier ID = RevisionSourceIdentifier.create("Test", Revision.of("2015-10-30"));
  34
  35     private RemoteYangTextSourceProviderImpl remoteRepository;
  36     private SchemaRepository mockedLocalRepository;
  37     private final Set<SourceIdentifier> providedSources = Collections.singleton(ID);
  38
  39     @Before
  40     public void setUp() {
  41         mockedLocalRepository = Mockito.mock(SchemaRepository.class);
  42
  43         remoteRepository = new RemoteYangTextSourceProviderImpl(mockedLocalRepository, providedSources);
  44     }
  45
  46     @Test
  47     public void testGetExistingYangTextSchemaSource() throws Exception {
  48         String source = "Test source.";
  49         YangTextSchemaSource schemaSource = YangTextSchemaSource.delegateForByteSource(
  50                 ID, ByteSource.wrap(source.getBytes()));
  51         Mockito.when(mockedLocalRepository.getSchemaSource(ID, YangTextSchemaSource.class)).thenReturn(
  52                 Futures.immediateFuture(schemaSource));
  53
  54         Future<YangTextSchemaSourceSerializationProxy> retrievedSourceFuture =
  55                 remoteRepository.getYangTextSchemaSource(ID);
  56         assertTrue(retrievedSourceFuture.isCompleted());
  57         YangTextSchemaSource resultSchemaSource = Await.result(retrievedSourceFuture,
  58                 FiniteDuration.Zero()).getRepresentation();
  59         assertEquals(resultSchemaSource.getIdentifier(), schemaSource.getIdentifier());
  60         assertArrayEquals(resultSchemaSource.read(), schemaSource.read());
  61     }
  62
  63     @Test(expected = SchemaSourceException.class)
  64     public void testGetNonExistentYangTextSchemaSource() throws Exception {
  65         Mockito.when(mockedLocalRepository.getSchemaSource(ID, YangTextSchemaSource.class)).thenReturn(
  66                 Futures.immediateFailedFuture(new SchemaSourceException("Source is not provided")));
  67
  68         Future<YangTextSchemaSourceSerializationProxy> retrievedSourceFuture =
  69                 remoteRepository.getYangTextSchemaSource(ID);
  70         assertTrue(retrievedSourceFuture.isCompleted());
  71         Await.result(retrievedSourceFuture, FiniteDuration.Zero());
  72     }
  73
  74     @Test
  75     public void testGetProvidedSources() throws Exception {
  76         Set<SourceIdentifier> remoteProvidedSources = Await.result(remoteRepository
  77                 .getProvidedSources(), FiniteDuration.Zero());
  78         assertEquals(providedSources, remoteProvidedSources);
  79     }
  80
  81 }