/**
 * Copyright (c) 2008, SnakeYAML
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */
package org.yaml.snakeyaml.representer;

import java.util.Set;
import java.util.TreeSet;
import junit.framework.TestCase;
import org.yaml.snakeyaml.DumperOptions;
import org.yaml.snakeyaml.TypeDescription;
import org.yaml.snakeyaml.Yaml;
import org.yaml.snakeyaml.introspector.Property;

public class FilterPropertyToDumpTest extends TestCase {

  public void testFilterPropertyInJavaBeanDumper() {
    BeanToRemoveProperty bean = new BeanToRemoveProperty();
    bean.setNumber(24);
    bean.setId("ID124");
    Yaml d = new Yaml();
    String dump = d.dumpAsMap(bean);
    // System.out.println(dump);
    assertEquals("id: ID124\nnumber: 24\n", dump);
  }

  public void testFilterPropertyInYaml() {
    BeanToRemoveProperty bean = new BeanToRemoveProperty();
    bean.setNumber(25);
    bean.setId("ID125");
    Yaml yaml = new Yaml(new MyRepresenter());
    String dump = yaml.dumpAsMap(bean);
    // System.out.println(dump);
    assertEquals("number: 25\n", dump);
  }

  public void testDoNotFilterPropertyIncludeReadOnly() {
    BeanToRemoveProperty bean = new BeanToRemoveProperty();
    bean.setNumber(26);
    bean.setId("ID126");
    DumperOptions options = new DumperOptions();
    options.setAllowReadOnlyProperties(true);
    Yaml yaml = new Yaml(options);
    String dump = yaml.dump(bean);
    // System.out.println(dump);
    assertEquals(
        "!!org.yaml.snakeyaml.representer.FilterPropertyToDumpTest$BeanToRemoveProperty {id: ID126,\n  number: 26, something: true}\n",
        dump);
  }

  public void testFilterPropertyWithTypeDesciptionIncludes() {
    BeanToRemoveProperty bean = new BeanToRemoveProperty();
    bean.setNumber(27);
    bean.setId("ID127");
    Yaml yaml = new Yaml();
    TypeDescription td = new TypeDescription(BeanToRemoveProperty.class);
    td.setIncludes("number");
    yaml.addTypeDescription(td);
    String dump = yaml.dump(bean);
    // System.out.println(dump);
    assertEquals(
        "!!org.yaml.snakeyaml.representer.FilterPropertyToDumpTest$BeanToRemoveProperty {number: 27}\n",
        dump);
  }

  public void testFilterPropertyWithTypeDesciptionExcludes() {
    BeanToRemoveProperty bean = new BeanToRemoveProperty();
    bean.setNumber(28);
    bean.setId("ID128");
    Yaml yaml = new Yaml();
    TypeDescription td = new TypeDescription(BeanToRemoveProperty.class);
    td.setExcludes("id");
    yaml.addTypeDescription(td);
    String dump = yaml.dump(bean);
    // System.out.println(dump);
    assertEquals(
        "!!org.yaml.snakeyaml.representer.FilterPropertyToDumpTest$BeanToRemoveProperty {number: 28}\n",
        dump);
  }


  public class BeanToRemoveProperty {

    private int number;
    private String id;

    public boolean isSomething() {
      return true;
    }

    public int getNumber() {
      return number;
    }

    public void setNumber(int number) {
      this.number = number;
    }

    public void setId(String id) {
      this.id = id;
    }

    public String getId() {
      return id;
    }
  }

  private class MyRepresenter extends Representer {

    /**
     * Create
     */
    public MyRepresenter() {
      super(new DumperOptions());
    }

    @Override
    protected Set<Property> getProperties(Class<? extends Object> type) {
      Set<Property> set = super.getProperties(type);
      Set<Property> filtered = new TreeSet<Property>();
      if (type.equals(BeanToRemoveProperty.class)) {
        // filter properties
        for (Property prop : set) {
          String name = prop.getName();
          if (!name.equals("id")) {
            filtered.add(prop);
          }
        }
      }
      return filtered;
    }
  }
}
