BUG-4688: Make SourceIdentifier use Revision

[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / IncludeStatementImpl.java

diff --git a/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/IncludeStatementImpl.java b/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/IncludeStatementImpl.java
index cc66a1f4c9799895a26b122b1e45f41bdd8e1e6d..74419055c88f105ba8159d555229010110a1cbf5 100644 (file)
--- a/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/IncludeStatementImpl.java
+++ b/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/IncludeStatementImpl.java
@@ -75,7 +75,7 @@ public class IncludeStatementImpl extends AbstractDeclaredStatement<String> impl
             final StmtContext<Revision, ?, ?> revision = findFirstDeclaredSubstatement(stmt,
                 RevisionDateStatement.class);
             stmt.addRequiredSource(revision == null ? RevisionSourceIdentifier.create(stmt.getStatementArgument())
-                : RevisionSourceIdentifier.create(stmt.getStatementArgument(), revision.rawStatementArgument()));
+                : RevisionSourceIdentifier.create(stmt.getStatementArgument(), revision.getStatementArgument()));
         }
 
         @Override
@@ -102,7 +102,7 @@ public class IncludeStatementImpl extends AbstractDeclaredStatement<String> impl
                     final StmtContext<?, ?, ?> includedSubModuleContext = requiresCtxPrerequisite.resolve(ctx);
 
                     stmt.addToNs(IncludedModuleContext.class, revision != null
-                            ? RevisionSourceIdentifier.create(submoduleName, revision.rawStatementArgument())
+                            ? RevisionSourceIdentifier.create(submoduleName, revision.getStatementArgument())
                                     : RevisionSourceIdentifier.create(submoduleName), includedSubModuleContext);
                     stmt.addToNs(IncludedSubmoduleNameToModuleCtx.class, stmt.getStatementArgument(),
                         includedSubModuleContext);