Migrate test asserts
[yangtools.git] / yang / yang-parser-impl / src / test / java / org / opendaylight / yangtools / yang / parser / impl / YangModelValidationModuleTest.java
1 /*
2  * Copyright (c) 2013 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.yangtools.yang.parser.impl;
9
10 import static org.hamcrest.CoreMatchers.containsString;
11 import static org.junit.Assert.assertThat;
12 import static org.mockito.Mockito.doReturn;
13 import static org.mockito.Mockito.mock;
14
15 import org.junit.Before;
16 import org.junit.Test;
17 import org.opendaylight.yangtools.antlrv4.code.gen.YangParser.Module_header_stmtsContext;
18 import org.opendaylight.yangtools.antlrv4.code.gen.YangParser.Module_stmtContext;
19 import org.opendaylight.yangtools.antlrv4.code.gen.YangParser.Namespace_stmtContext;
20 import org.opendaylight.yangtools.antlrv4.code.gen.YangParser.Revision_stmtContext;
21 import org.opendaylight.yangtools.antlrv4.code.gen.YangParser.Revision_stmtsContext;
22 import org.opendaylight.yangtools.antlrv4.code.gen.YangParser.Yang_version_stmtContext;
23 import org.opendaylight.yangtools.yang.parser.util.YangValidationException;
24
25 public class YangModelValidationModuleTest {
26
27     private YangModelBasicValidationListener valid;
28
29     @Before
30     public void setUp() {
31         valid = new YangModelBasicValidationListener();
32     }
33
34     @Test(expected = YangValidationException.class)
35     public void testRevisionInvalidDateFormat() {
36         Revision_stmtContext mockedRev = mockModuleWithRevision("badFormat",
37                 "module1");
38
39         try {
40             valid.enterRevision_stmt(mockedRev);
41         } catch (YangValidationException e) {
42             assertThat(
43                     e.getMessage(),
44                     containsString("Revision:badFormat, invalid date format expected date format is:"));
45             throw e;
46         }
47     }
48
49     @Test
50     public void testRevisionValidDateFormat() {
51         Revision_stmtContext mockedRev = mockModuleWithRevision(
52                 YangModelValidationTest.getFormattedDate(), "module1");
53
54         valid.enterRevision_stmt(mockedRev);
55     }
56
57     @Test(expected = YangValidationException.class)
58     public void testNoHeaderStmts() {
59         Revision_stmtContext rev = mockModuleWithRevision("1999-4-5", "module1");
60
61         try {
62             valid.enterModule_stmt((Module_stmtContext) rev.getParent()
63                     .getParent());
64         } catch (YangValidationException e) {
65             assertThat(
66                     e.getMessage(),
67                     containsString("Missing Module-header statement in Module:module1"));
68             throw e;
69         }
70     }
71
72     @Test(expected = YangValidationException.class)
73     public void testMultipleModulesPerSession() {
74         Module_stmtContext module1 = (Module_stmtContext) mockModuleWithRevision(
75                 "1999-09-10", "m1").getParent().getParent();
76         YangModelValidationTest.addChild(module1, YangModelValidationTest
77                 .mockStatement(Namespace_stmtContext.class, ""));
78
79         Module_stmtContext module2 = (Module_stmtContext) mockModuleWithRevision(
80                 "1999-09-10", "m2").getParent().getParent();
81         YangModelValidationTest.addChild(module1, YangModelValidationTest
82                 .mockStatement(Namespace_stmtContext.class, ""));
83         valid.enterModule_stmt(module1);
84
85         try {
86             valid.enterModule_stmt(module2);
87         } catch (YangValidationException e) {
88             assertThat(e.getMessage(),
89                     containsString("Multiple (sub)modules per file"));
90             throw e;
91         }
92     }
93
94     @Test(expected = YangValidationException.class)
95     public void testNoNamespace() {
96         Module_header_stmtsContext header = YangModelValidationTest
97                 .mockStatement(Module_header_stmtsContext.class, null);
98         Module_stmtContext mod = YangModelValidationTest.mockStatement(
99                 Module_stmtContext.class, "module1");
100         YangModelValidationTest.addChild(mod, header);
101
102         try {
103             valid.enterModule_header_stmts(header);
104         } catch (YangValidationException e) {
105             assertThat(
106                     e.getMessage(),
107                     containsString("Missing Namespace statement in Module-header:"));
108             throw e;
109         }
110     }
111
112     @Test(expected = YangValidationException.class)
113     public void testNoPrefix() {
114         Module_header_stmtsContext header = YangModelValidationTest
115                 .mockStatement(Module_header_stmtsContext.class, null);
116         Namespace_stmtContext nmspc = YangModelValidationTest.mockStatement(
117                 Namespace_stmtContext.class, "http://test");
118         Module_stmtContext mod = YangModelValidationTest.mockStatement(
119                 Module_stmtContext.class, "module1");
120         YangModelValidationTest.addChild(mod, header);
121         YangModelValidationTest.addChild(header, nmspc);
122
123         try {
124             valid.enterModule_header_stmts(header);
125         } catch (YangValidationException e) {
126             assertThat(
127                     e.getMessage(),
128                     containsString("Missing Prefix statement in Module-header:"));
129             throw e;
130         }
131     }
132
133     @Test(expected = YangValidationException.class)
134     public void testInvalidYangVersion() {
135
136         Yang_version_stmtContext yangVersion = YangModelValidationTest
137                 .mockStatement(Yang_version_stmtContext.class, "55Unsup");
138
139         Module_stmtContext mod = YangModelValidationTest.mockStatement(
140                 Module_stmtContext.class, "module1");
141         YangModelValidationTest.addChild(mod, yangVersion);
142
143         try {
144             valid.enterYang_version_stmt(yangVersion);
145         } catch (YangValidationException e) {
146             assertThat(
147                     e.getMessage(),
148                     containsString("Unsupported yang version:55Unsup, supported version:"
149                             + BasicValidations.SUPPORTED_YANG_VERSION));
150             throw e;
151         }
152     }
153
154     @Test
155     public void testValidYangVersion() {
156
157         Yang_version_stmtContext ctx = mock(Yang_version_stmtContext.class);
158         doReturn(1).when(ctx).getChildCount();
159         YangModelValidationTest.mockName(ctx, "1");
160
161         valid.enterYang_version_stmt(ctx);
162     }
163
164     private static Revision_stmtContext mockModuleWithRevision(final String date,
165             final String moduleName) {
166         Revision_stmtContext mockedRev = YangModelValidationTest.mockStatement(
167                 Revision_stmtContext.class, date);
168         Revision_stmtsContext revs = YangModelValidationTest.mockStatement(
169                 Revision_stmtsContext.class, null);
170         Module_stmtContext mod = YangModelValidationTest.mockStatement(
171                 Module_stmtContext.class, moduleName);
172
173         YangModelValidationTest.addChild(revs, mockedRev);
174         YangModelValidationTest.addChild(mod, revs);
175         return mockedRev;
176     }
177 }