| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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.apache.hadoop.hdfs.util; |
| |
| import org.apache.hadoop.hdfs.util.XMLUtils.UnmanglingError; |
| import org.junit.Assert; |
| import org.junit.Test; |
| |
| public class TestXMLUtils { |
| private static void testRoundTripImpl(String str, String expectedMangled, |
| boolean encodeEntityRefs) { |
| String mangled = XMLUtils.mangleXmlString(str, encodeEntityRefs); |
| Assert.assertEquals(expectedMangled, mangled); |
| String unmangled = XMLUtils.unmangleXmlString(mangled, encodeEntityRefs); |
| Assert.assertEquals(str, unmangled); |
| } |
| |
| private static void testRoundTrip(String str, String expectedMangled) { |
| testRoundTripImpl(str, expectedMangled, false); |
| } |
| |
| private static void testRoundTripWithEntityRefs(String str, |
| String expectedMangled) { |
| testRoundTripImpl(str, expectedMangled, true); |
| } |
| |
| @Test |
| public void testMangleEmptyString() throws Exception { |
| testRoundTrip("", ""); |
| } |
| |
| @Test |
| public void testMangleVanillaString() throws Exception { |
| testRoundTrip("abcdef", "abcdef"); |
| } |
| |
| @Test |
| public void testMangleStringWithBackSlash() throws Exception { |
| testRoundTrip("a\\bcdef", "a\\005c;bcdef"); |
| testRoundTrip("\\\\", "\\005c;\\005c;"); |
| } |
| |
| @Test |
| public void testMangleStringWithForbiddenCodePoint() throws Exception { |
| testRoundTrip("a\u0001bcdef", "a\\0001;bcdef"); |
| testRoundTrip("a\u0002\ud800bcdef", "a\\0002;\\d800;bcdef"); |
| } |
| |
| @Test |
| public void testInvalidSequence() throws Exception { |
| try { |
| XMLUtils.unmangleXmlString("\\000g;foo", false); |
| Assert.fail("expected an unmangling error"); |
| } catch (UnmanglingError e) { |
| // pass through |
| } |
| try { |
| XMLUtils.unmangleXmlString("\\0", false); |
| Assert.fail("expected an unmangling error"); |
| } catch (UnmanglingError e) { |
| // pass through |
| } |
| } |
| |
| @Test |
| public void testAddEntityRefs() throws Exception { |
| testRoundTripWithEntityRefs("The Itchy & Scratchy Show", |
| "The Itchy & Scratchy Show"); |
| testRoundTripWithEntityRefs("\"He said '1 < 2, but 2 > 1'\"", |
| ""He said '1 < 2, but 2 > 1'""); |
| testRoundTripWithEntityRefs("\u0001 < \u0002", "\\0001; < \\0002;"); |
| } |
| } |