2 * Copyright (c) 2023 PANTHEON.tech 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.mdsal.binding.runtime.spi;
10 import static org.junit.jupiter.api.Assertions.assertEquals;
13 import org.junit.jupiter.api.Test;
14 import org.opendaylight.yang.gen.v1.mdsal767.norev.Mdsal767Data;
15 import org.opendaylight.yang.gen.v1.mdsal767.norev.One$F;
16 import org.opendaylight.yang.svc.v1.mdsal767.norev.YangModuleInfoImpl;
17 import org.opendaylight.yangtools.yang.common.QName;
18 import org.opendaylight.yangtools.yang.common.QNameModule;
19 import org.opendaylight.yangtools.yang.common.XMLNamespace;
20 import org.opendaylight.yangtools.yang.parser.impl.DefaultYangParserFactory;
22 class ModuleInfoSnapshotBuilderTest {
24 void testModuleRegistration() throws Exception {
25 final var snapshotBuilder = new ModuleInfoSnapshotBuilder(new DefaultYangParserFactory());
26 snapshotBuilder.add(YangModuleInfoImpl.getInstance());
27 snapshotBuilder.addModuleFeatures(Mdsal767Data.class, Set.of(One$F.VALUE));
29 final var snapshot = snapshotBuilder.build();
30 final var modelContext = snapshot.modelContext();
31 final var modules = modelContext.getModuleStatements();
32 final var module = modules.get(QNameModule.create(XMLNamespace.of("mdsal767")));
33 assertEquals(1, module.features().size());
34 final var feature = module.features().stream().findAny().orElseThrow();
35 assertEquals(QName.create("mdsal767", "one"), feature.argument());