libcore/luni/src/test/java/tests/api/org/xml/sax/support/MockResolver.java

   1 /*
   2  * Copyright (C) 2007 The Android Open Source Project
   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 tests.api.org.xml.sax.support;
  18
  19 import java.io.IOException;
  20 import java.util.HashMap;
  21 import java.util.Map;
  22
  23 import org.xml.sax.EntityResolver;
  24 import org.xml.sax.InputSource;
  25 import org.xml.sax.SAXException;
  26
  27 /**
  28  * A helper class for resolving entities.
  29  */
  30 public class MockResolver implements EntityResolver {
  31
  32     private Map<String, InputSource> entities = new HashMap<String, InputSource>();
  33
  34     public void addEntity(String publicId, String systemId, InputSource source) {
  35         entities.put("[" + publicId + ":" + systemId + "]", source);
  36     }
  37
  38     public void removeEntity(String publicId, String systemId) {
  39         entities.remove("[" + publicId + ":" + systemId + "]");
  40     }
  41
  42     public InputSource resolveEntity(String publicId, String systemId)
  43             throws SAXException, IOException {
  44         return entities.get("[" + publicId + ":" + systemId + "]");
  45     }
  46
  47 }