yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/test/TestStatementSource.java

   1 /*
   2  * Copyright (c) 2015 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.stmt.test;
   9
  10 import static org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping.IMPORT;
  11 import static org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping.MODULE;
  12 import static org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping.NAMESPACE;
  13 import static org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping.PREFIX;
  14
  15 import java.util.Arrays;
  16 import java.util.List;
  17 import org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping;
  18 import org.opendaylight.yangtools.yang.model.api.meta.StatementSource;
  19 import org.opendaylight.yangtools.yang.parser.spi.source.PrefixToModule;
  20 import org.opendaylight.yangtools.yang.parser.spi.source.QNameToStatementDefinition;
  21 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
  22 import org.opendaylight.yangtools.yang.parser.spi.source.StatementSourceReference;
  23 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
  24 import org.opendaylight.yangtools.yang.parser.spi.source.StatementWriter;
  25
  26 class TestStatementSource implements StatementStreamSource {
  27
  28     private static final String NS_PREFIX = "urn:org:opendaylight:yangtools:test:";
  29
  30     private final String name;
  31     private final List<String> imports;
  32     private StatementWriter writer;
  33     private StatementSourceReference REF = new StatementSourceReference() {
  34
  35         @Override
  36         public StatementSource getStatementSource() {
  37             return StatementSource.DECLARATION;
  38         }
  39     };
  40
  41
  42     public TestStatementSource(String name, String... imports) {
  43         this.name = name;
  44         this.imports = Arrays.asList(imports);
  45     }
  46
  47     @Override
  48     public void writeFull(StatementWriter writer, QNameToStatementDefinition stmtDef, PrefixToModule prefixes)
  49             throws SourceException {
  50         this.writer = writer;
  51         header();
  52         extensions();
  53         body();
  54         end();
  55     }
  56
  57     @Override
  58     public void writeLinkage(StatementWriter writer, QNameToStatementDefinition stmtDef) throws SourceException {
  59         this.writer = writer;
  60         header().end();
  61     }
  62
  63     @Override
  64     public void writeLinkageAndStatementDefinitions(StatementWriter writer, QNameToStatementDefinition stmtDef,
  65             PrefixToModule prefixes) throws SourceException {
  66         this.writer = writer;
  67         header();
  68         extensions();
  69         end();
  70     }
  71
  72     protected void extensions() {
  73         // TODO Auto-generated method stub
  74
  75     }
  76
  77     protected void body() {
  78
  79     }
  80
  81     TestStatementSource header() throws SourceException {
  82         stmt(MODULE).arg(name); {
  83             stmt(NAMESPACE).arg(getNamespace()).end();
  84             stmt(PREFIX).arg(name).end();
  85             for(String imp : imports) {
  86                 stmt(IMPORT).arg(imp);
  87                     stmt(PREFIX).arg(imp).end();
  88                 end();
  89             }
  90         }
  91         return this;
  92     }
  93
  94     private String getNamespace() {
  95         return NS_PREFIX + name;
  96     }
  97
  98     protected TestStatementSource arg(String arg) throws SourceException {
  99         writer.argumentValue(arg, REF);
 100         return this;
 101     }
 102
 103     protected TestStatementSource stmt(Rfc6020Mapping stmt) throws SourceException {
 104         writer.startStatement(stmt.getStatementName(), REF);
 105         return this;
 106     }
 107
 108     protected TestStatementSource end() throws SourceException {
 109         writer.endStatement(REF);
 110         return this;
 111     }
 112 }