2 * Copyright (c) 2021 PANTHEON.tech, s.r.o. and others. 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.generator.impl.reactor;
10 import static org.junit.Assert.assertEquals;
11 import static org.mockito.Mockito.doReturn;
13 import java.util.List;
14 import java.util.Optional;
15 import org.junit.Test;
16 import org.junit.runner.RunWith;
17 import org.mockito.Mock;
18 import org.mockito.junit.MockitoJUnitRunner;
19 import org.opendaylight.mdsal.binding.model.api.GeneratedTransferObject;
20 import org.opendaylight.mdsal.binding.model.api.JavaTypeName;
21 import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedTOBuilder;
22 import org.opendaylight.mdsal.binding.model.ri.generated.type.builder.CodegenGeneratedTOBuilder;
23 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
25 @RunWith(MockitoJUnitRunner.StrictStubs.class)
26 public class GeneratorTest {
28 public TypeDefinition<?> typedef;
30 public final GeneratedTOBuilder builder =
31 new CodegenGeneratedTOBuilder(JavaTypeName.create("test.package", "TestBuilder"));
34 public void testAddUnits() {
35 doReturn(Optional.of("125")).when(typedef).getUnits();
37 Generator.addUnits(builder, typedef);
38 final GeneratedTransferObject genTO = builder.build();
39 assertEquals(1, genTO.getConstantDefinitions().size());
40 assertEquals("_UNITS", genTO.getConstantDefinitions().get(0).getName());
41 assertEquals(genTO.getConstantDefinitions().get(0).getValue(), "\"125\"");
45 public void testAddUnitsNonExistent() {
46 doReturn(Optional.empty()).when(typedef).getUnits();
48 Generator.addUnits(builder, typedef);
49 GeneratedTransferObject genTO = builder.build();
50 assertEquals(List.of(), genTO.getConstantDefinitions());
55 public void testAddUnitsEmpty() {
56 doReturn(Optional.of("")).when(typedef).getUnits();
58 Generator.addUnits(builder, typedef);
59 GeneratedTransferObject genTO = builder.build();
60 assertEquals(List.of(), genTO.getConstantDefinitions());