/*
* Copyright (c) MuleSoft, Inc. All rights reserved. http://www.mulesoft.com
* The software in this package is published under the terms of the CPAL v1.0
* license, a copy of which has been included with this distribution in the
* LICENSE.txt file.
*/
package org.mule.transformer.encryption;
import org.mule.api.lifecycle.InitialisationException;
import org.mule.api.security.CryptoFailureException;
import org.mule.api.transformer.Transformer;
import org.mule.security.PasswordBasedEncryptionStrategy;
import org.mule.transformer.AbstractTransformerTestCase;
import java.io.ByteArrayInputStream;
import java.util.Arrays;
import static org.junit.Assert.fail;
public class EncryptionTransformerTestCase extends AbstractTransformerTestCase
{
private static final String TEST_DATA = "the quick brown fox jumped over the lazy dog the quick brown fox jumped over the lazy dog the quick brown fox jumped over the lazy dog";
private PasswordBasedEncryptionStrategy strat;
@Override
protected void doSetUp() throws Exception
{
strat = new PasswordBasedEncryptionStrategy();
strat.setPassword("mule");
strat.initialise();
}
@Override
public Object getResultData()
{
try
{
return new ByteArrayInputStream(strat.encrypt(TEST_DATA.getBytes(), null));
}
catch (CryptoFailureException e)
{
fail(e.getMessage());
return null;
}
}
@Override
public Object getTestData()
{
return new ByteArrayInputStream(TEST_DATA.getBytes());
}
@Override
public Transformer getTransformer()
{
EncryptionTransformer transformer = new EncryptionTransformer();
transformer.setStrategy(strat);
try
{
transformer.initialise();
}
catch (InitialisationException e)
{
fail(e.getMessage());
}
return transformer;
}
@Override
public Transformer getRoundTripTransformer()
{
DecryptionTransformer transformer = new DecryptionTransformer();
transformer.setStrategy(strat);
try
{
transformer.initialise();
}
catch (InitialisationException e)
{
fail(e.getMessage());
}
return transformer;
}
@Override
public boolean compareResults(Object src, Object result)
{
if (src == null && result == null)
{
return true;
}
if (src == null || result == null)
{
return false;
}
if (src instanceof byte[] && result instanceof byte[])
{
return Arrays.equals((byte[]) src, (byte[]) result);
}
else
{
return super.compareResults(src, result);
}
}
}