View Javadoc

    1   /**
    2    * Copyright (c) 2008-2011, http://www.snakeyaml.org
    3    *
    4    * Licensed under the Apache License, Version 2.0 (the "License");
    5    * you may not use this file except in compliance with the License.
    6    * You may obtain a copy of the License at
    7    *
    8    *     http://www.apache.org/licenses/LICENSE-2.0
    9    *
    10   * Unless required by applicable law or agreed to in writing, software
    11   * distributed under the License is distributed on an "AS IS" BASIS,
    12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    13   * See the License for the specific language governing permissions and
    14   * limitations under the License.
    15   */
    16
    17  package org.yaml.snakeyaml.constructor;
    18
    19  import junit.framework.TestCase;
    20
    21  import org.yaml.snakeyaml.Yaml;
    22
    23  public class FilterClassesConstructorTest extends TestCase {
    24
    25      public void testGetClassForName() {
    26          Yaml yaml = new Yaml(new FilterConstructor(true));
    27          String input = "!!org.yaml.snakeyaml.constructor.FilterClassesConstructorTest$FilteredBean {name: Andrey, number: 543}";
    28          try {
    29              yaml.load(input);
    30              fail("Filter is expected.");
    31          } catch (Exception e) {
    32              assertTrue(e.getMessage().contains("Filter is applied."));
    33          }
    34          yaml = new Yaml(new FilterConstructor(false));
    35          FilteredBean s = (FilteredBean) yaml.load(input);
    36          assertEquals("Andrey", s.getName());
    37      }
    38
    39      class FilterConstructor extends Constructor {
    40          private boolean filter;
    41
    42          public FilterConstructor(boolean f) {
    43              filter = f;
    44          }
    45
    46          @Override
    47          protected Class<?> getClassForName(String name) throws ClassNotFoundException {
    48              if (filter && name.startsWith("org.yaml")) {
    49                  throw new RuntimeException("Filter is applied.");
    50              }
    51              return super.getClassForName(name);
    52          }
    53      }
    54
    55      public static class FilteredBean {
    56          private String name;
    57          private int number;
    58
    59          public String getName() {
    60              return name;
    61          }
    62
    63          public void setName(String name) {
    64              this.name = name;
    65          }
    66
    67          public int getNumber() {
    68              return number;
    69          }
    70
    71          public void setNumber(int number) {
    72              this.number = number;
    73          }
    74      }
    75  }