Adobe-Consulting-Services/acs-aem-commons

/*
 * ACS AEM Commons
 *
 * Copyright (C) 2013 - 2023 Adobe
 *
 * 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 com.adobe.acs.commons.rewriter.impl;

import static org.mockito.Mockito.*;

import java.io.PrintWriter;

import javax.xml.parsers.ParserConfigurationException;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;

import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Captor;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;
import org.xml.sax.Attributes;
import org.xml.sax.ContentHandler;
import org.xml.sax.Locator;
import org.xml.sax.SAXException;
import org.xml.sax.SAXNotRecognizedException;
import org.xml.sax.SAXNotSupportedException;

@RunWith(MockitoJUnitRunner.class)
public class XMLParserGeneratorTest {

    @Mock
    private ContentHandler contentHandler;
    
    @Captor ArgumentCaptor<Attributes> attributesCaptor;

    @Test
    public void test() throws Exception {
        
        XMLParserGeneratorFactory factory = new XMLParserGeneratorFactory();
        XMLParserGenerator generator = (XMLParserGenerator) factory.createGenerator();
        // nothing to do, nothing to mock...
        generator.init(null, null);
        generator.setContentHandler(contentHandler);
        PrintWriter printWriter = generator.getWriter();
        printWriter.println("<?xml version=\"1.0\" encoding=\"utf-8\"?>");
        printWriter.println("<fo:root xmlns:fo=\"http://www.w3.org/1999/XSL/Format\"></fo:root>");
        
        generator.finished();
        generator.dispose();
        
        verify(contentHandler).startDocument();
        verify(contentHandler).setDocumentLocator((Locator) any());
        verify(contentHandler).startPrefixMapping("fo", "http://www.w3.org/1999/XSL/Format");

        verify(contentHandler).startElement(eq("http://www.w3.org/1999/XSL/Format"), eq("root"), eq("fo:root"), attributesCaptor.capture());
        
        verify(contentHandler).endElement("http://www.w3.org/1999/XSL/Format", "root", "fo:root");
        
        verify(contentHandler).endPrefixMapping("fo");
        verify(contentHandler).endDocument();
        verifyNoMoreInteractions(contentHandler);
    }

    @Test
    public void testParserException() {
        SAXParserFactory fakeParserFactory = new SAXParserFactory() {
            @Override
            public SAXParser newSAXParser() throws ParserConfigurationException, SAXException {
                throw new ParserConfigurationException("failers gonna fail.");
            }

            @Override
            public void setFeature(final String name, final boolean value)
                    throws ParserConfigurationException, SAXNotRecognizedException, SAXNotSupportedException {
            }

            @Override
            public boolean getFeature(final String name)
                    throws ParserConfigurationException, SAXNotRecognizedException, SAXNotSupportedException {
                return false;
            }
        };

        XMLParserGeneratorFactory factory = new XMLParserGeneratorFactory();
        XMLParserGenerator generator = (XMLParserGenerator) factory.createGenerator(fakeParserFactory);
        Assert.assertNull("generator is null when parser factory throws config exception.", generator);
    }

    @Test
    public void testOwnParser() {
        XMLParserGeneratorFactory factory = new XMLParserGeneratorFactory();
        XMLParserGenerator generator = (XMLParserGenerator) factory.createGenerator(SAXParserFactory.newInstance());
        Assert.assertNotNull("generator is not null with own default sax parser.", generator);
    }
}