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.parser.impl.DefaultYangParserFactory;
21 class ModuleInfoSnapshotBuilderTest {
23 void testModuleRegistration() throws Exception {
24 final var snapshotBuilder = new ModuleInfoSnapshotBuilder(new DefaultYangParserFactory());
25 snapshotBuilder.add(YangModuleInfoImpl.getInstance());
26 snapshotBuilder.addModuleFeatures(Mdsal767Data.class, Set.of(One$F.VALUE));
28 final var snapshot = snapshotBuilder.build();
29 final var modelContext = snapshot.modelContext();
30 final var modules = modelContext.getModuleStatements();
31 final var module = modules.get(QNameModule.of("mdsal767"));
32 assertEquals(1, module.features().size());
33 final var feature = module.features().stream().findAny().orElseThrow();
34 assertEquals(QName.create("mdsal767", "one"), feature.argument());