/** * DataCleaner (community edition) * Copyright (C) 2014 Neopost - Customer Information Management * * This copyrighted material is made available to anyone wishing to use, modify, * copy, or redistribute it subject to the terms and conditions of the GNU * Lesser General Public License, as published by the Free Software Foundation. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License * for more details. * * You should have received a copy of the GNU Lesser General Public License * along with this distribution; if not, write to: * Free Software Foundation, Inc. * 51 Franklin Street, Fifth Floor * Boston, MA 02110-1301 USA */ package org.datacleaner.beans.writers; import java.util.Collection; import java.util.TreeSet; import org.datacleaner.api.Renderer; import org.datacleaner.configuration.DataCleanerConfiguration; import org.datacleaner.configuration.DataCleanerConfigurationImpl; import org.datacleaner.configuration.DataCleanerEnvironmentImpl; import org.datacleaner.descriptors.ClasspathScanDescriptorProvider; import org.datacleaner.descriptors.DescriptorProvider; import org.datacleaner.descriptors.RendererBeanDescriptor; import org.datacleaner.result.html.DefaultHtmlRenderingContext; import org.datacleaner.result.html.HtmlFragment; import org.datacleaner.result.html.HtmlRenderingContext; import org.datacleaner.result.renderer.HtmlRenderingFormat; import org.datacleaner.result.renderer.RendererFactory; import org.junit.Assert; import org.junit.Test; public class WriteDataResultHtmlRendererTest { @Test public void testRendering() { final WriteDataResult result = new WriteDataResultImpl(2, 3, "datastore", "schema", "table"); final WriteDataResultHtmlRenderer renderer = new WriteDataResultHtmlRenderer(); final HtmlFragment htmlFragment = renderer.render(result); Assert.assertEquals(0, htmlFragment.getHeadElements().size()); Assert.assertEquals(1, htmlFragment.getBodyElements().size()); final HtmlRenderingContext context = new DefaultHtmlRenderingContext(); Assert.assertEquals("<div>\n <p>Executed 2 inserts</p>\n <p>Executed 3 updates</p>\n</div>", htmlFragment.getBodyElements().get(0).toHtml(context).replaceAll("\r\n", "\n")); } @Test public void testClasspathDiscovery() { final DescriptorProvider descriptorProvider = new ClasspathScanDescriptorProvider().scanPackage("org.datacleaner.beans", true); final Collection<RendererBeanDescriptor<?>> htmlRenderers = descriptorProvider.getRendererBeanDescriptorsForRenderingFormat(HtmlRenderingFormat.class); final TreeSet<RendererBeanDescriptor<?>> sorted = new TreeSet<>(htmlRenderers); Assert.assertEquals( "[AnnotationBasedRendererBeanDescriptor[org.datacleaner.beans.DefaultAnalyzerResultHtmlRenderer], " + "AnnotationBasedRendererBeanDescriptor[org.datacleaner.beans.writers.WriteDataResultHtmlRenderer]]", sorted.toString()); final DataCleanerConfiguration conf = new DataCleanerConfigurationImpl() .withEnvironment(new DataCleanerEnvironmentImpl().withDescriptorProvider(descriptorProvider)); final RendererFactory rendererFactory = new RendererFactory(conf); final Renderer<? super WriteDataResultImpl, ? extends HtmlFragment> renderer = rendererFactory .getRenderer(new WriteDataResultImpl(2, 3, "datastore", "schema", "table"), HtmlRenderingFormat.class); Assert.assertEquals(WriteDataResultHtmlRenderer.class, renderer.getClass()); } }