2 * Copyright (c) 2018 Pantheon Technologies, s.r.o. 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.util;
10 import static com.google.common.collect.ImmutableList.of;
11 import static org.junit.Assert.assertEquals;
12 import static org.mockito.Mockito.doReturn;
14 import com.google.common.collect.ImmutableSet;
15 import java.util.List;
16 import java.util.Optional;
17 import org.junit.Before;
18 import org.junit.Test;
19 import org.junit.runner.RunWith;
20 import org.mockito.Mock;
21 import org.mockito.junit.MockitoJUnitRunner;
22 import org.opendaylight.yangtools.yang.common.QNameModule;
23 import org.opendaylight.yangtools.yang.common.XMLNamespace;
24 import org.opendaylight.yangtools.yang.common.YangVersion;
25 import org.opendaylight.yangtools.yang.model.api.Module;
26 import org.opendaylight.yangtools.yang.model.api.ModuleImport;
27 import org.opendaylight.yangtools.yang.model.api.Submodule;
29 @RunWith(MockitoJUnitRunner.class)
30 public class ModuleDependencySortTest {
31 public static final QNameModule FOO_MODULE = QNameModule.create(XMLNamespace.of("foo"));
32 public static final QNameModule BAR_MODULE = QNameModule.create(XMLNamespace.of("bar"));
35 public Module fooNoRev;
38 public ModuleImport fooNoRevImport;
44 public Submodule barSubmodule;
47 public void before() {
48 doReturn("foo").when(fooNoRev).getName();
49 doReturn(FOO_MODULE.getNamespace()).when(fooNoRev).getNamespace();
50 doReturn(FOO_MODULE.getRevision()).when(fooNoRev).getRevision();
51 doReturn(ImmutableSet.of()).when(fooNoRev).getImports();
52 doReturn(ImmutableSet.of()).when(fooNoRev).getSubmodules();
54 doReturn("foo").when(fooNoRevImport).getModuleName();
55 doReturn(Optional.empty()).when(fooNoRevImport).getRevision();
57 doReturn("bar").when(bar).getName();
58 doReturn(BAR_MODULE.getNamespace()).when(bar).getNamespace();
59 doReturn(BAR_MODULE.getRevision()).when(bar).getRevision();
60 doReturn(YangVersion.VERSION_1).when(bar).getYangVersion();
61 doReturn(ImmutableSet.of()).when(bar).getImports();
62 doReturn(ImmutableSet.of(barSubmodule)).when(bar).getSubmodules();
64 doReturn(ImmutableSet.of(fooNoRevImport)).when(barSubmodule).getImports();
68 public void testSimpleModules() {
69 assertSortedTo(of(fooNoRev), fooNoRev);
73 public void testSubmodules() {
74 assertSortedTo(of(fooNoRev, bar), bar, fooNoRev);
75 assertSortedTo(of(fooNoRev, bar), fooNoRev, bar);
78 private static void assertSortedTo(final List<Module> expected, final Module... modules) {
79 assertEquals(expected, ModuleDependencySort.sort(modules));