aboutsummaryrefslogtreecommitdiff
path: root/src/test/java/org/yaml/snakeyaml/issues/issue480/AnchorUnicodeTest.java
blob: 5c195f662c6723aa8e433f323553473bf46bda72 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
/**
 * 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.issues.issue480;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import junit.framework.TestCase;
import org.yaml.snakeyaml.DumperOptions;
import org.yaml.snakeyaml.Yaml;
import org.yaml.snakeyaml.emitter.EmitterException;
import org.yaml.snakeyaml.nodes.Node;
import org.yaml.snakeyaml.serializer.AnchorGenerator;

public class AnchorUnicodeTest extends TestCase {

  private static final Set<Character> INVALID_ANCHOR = new HashSet();

  static {
    INVALID_ANCHOR.add('[');
    INVALID_ANCHOR.add(']');
    INVALID_ANCHOR.add('{');
    INVALID_ANCHOR.add('}');
    INVALID_ANCHOR.add(',');
    INVALID_ANCHOR.add('*');
    INVALID_ANCHOR.add('&');
  }

  public void testUnicodeAnchor() {
    DumperOptions options = new DumperOptions();
    options.setAnchorGenerator(new AnchorGenerator() {
      int id = 0;

      @Override
      public String nextAnchor(Node node) {
        return "タスク" + id++;
      }
    });

    Yaml yaml = new Yaml(options);

    List<String> list = new ArrayList<>();
    list.add("abc");

    List<List<String>> toExport = new ArrayList<>();
    toExport.add(list);
    toExport.add(list);

    String output = yaml.dump(toExport);
    assertEquals("- &タスク0 [abc]\n- *タスク0\n", output);
  }

  public void testInvalidAnchor() {
    for (Character ch : INVALID_ANCHOR) {
      Yaml yaml = new Yaml(createSettings(ch));
      List<String> list = new ArrayList<>();
      list.add("abc");
      List<List<String>> toExport = new ArrayList<>();
      toExport.add(list);
      toExport.add(list);
      try {
        yaml.dump(toExport);
        fail();
      } catch (EmitterException e) {
        assertTrue(e.getMessage().startsWith("Invalid character"));
      }
    }
  }

  private DumperOptions createSettings(final Character invalid) {
    DumperOptions options = new DumperOptions();
    options.setAnchorGenerator(new AnchorGenerator() {
      @Override
      public String nextAnchor(Node node) {
        return "anchor" + invalid;
      }
    });
    return options;
  }
}