Join the 1-day Testing & QA Summit featuring 15+ Expert Speakers.Register for FREE! Join TestMu Conference

How to use CitrusRuntimeException method of com.consol.citrus.ssh.server.SinglePublicKeyAuthenticatorTest class

Best Citrus code snippet using com.consol.citrus.ssh.server.SinglePublicKeyAuthenticatorTest.CitrusRuntimeException

Source:SinglePublicKeyAuthenticatorTest.java

...13 * See the License for the specific language governing permissions and14 * limitations under the License.15 */16package com.consol.citrus.ssh.server;17import com.consol.citrus.exceptions.CitrusRuntimeException;18import org.apache.sshd.common.util.io.IoUtils;19import org.apache.sshd.common.util.security.SecurityUtils;20import org.bouncycastle.asn1.x509.SubjectPublicKeyInfo;21import org.bouncycastle.jce.provider.BouncyCastleProvider;22import org.bouncycastle.openssl.PEMKeyPair;23import org.bouncycastle.openssl.PEMParser;24import org.springframework.util.FileCopyUtils;25import org.testng.annotations.Test;26import java.io.*;27import java.security.PublicKey;28import static org.testng.Assert.assertFalse;29import static org.testng.Assert.assertTrue;30/**31 * @author Roland Huss32 * @since 05.09.1233 */34public class SinglePublicKeyAuthenticatorTest {35    /**36     * Default constructor.37     */38    public SinglePublicKeyAuthenticatorTest() {39        assertTrue(SecurityUtils.isBouncyCastleRegistered());40    }41    @Test42    public void withClassPath() throws IOException {43        SinglePublicKeyAuthenticator auth = new SinglePublicKeyAuthenticator("roland","classpath:com/consol/citrus/ssh/allowed_test_key.pem");44        PublicKey pKey = getPublicKey("/com/consol/citrus/ssh/allowed_test_key.pem");45        assertTrue(auth.authenticate("roland", pKey, null));46        assertFalse(auth.authenticate("guenther", pKey, null));47        pKey = getPublicKey("/com/consol/citrus/ssh/forbidden_test_key.pem");48        assertFalse(auth.authenticate("roland", pKey, null));49        pKey = getPublicKey("/com/consol/citrus/ssh/citrus.pem");50        assertFalse(auth.authenticate("citrus", pKey, null));51    }52    @Test53    public void withFile() throws IOException {54        File temp = copyToTempFile("/com/consol/citrus/ssh/allowed_test_key.pem");55        SinglePublicKeyAuthenticator auth = new SinglePublicKeyAuthenticator("roland",temp.getAbsolutePath());56        PublicKey pKey = getPublicKeyFromStream(new FileInputStream(temp));57        assertTrue(auth.authenticate("roland", pKey, null));58        assertFalse(auth.authenticate("guenther",pKey,null));59        temp = copyToTempFile("/com/consol/citrus/ssh/forbidden_test_key.pem");60        pKey = getPublicKeyFromStream(new FileInputStream(temp));61        assertFalse(auth.authenticate("roland", pKey, null));62    }63    @Test(expectedExceptions = CitrusRuntimeException.class, expectedExceptionsMessageRegExp = ".*com/consol/citrus/ssh/private.key.*")64    public void invalidKeyFormat() {65        new SinglePublicKeyAuthenticator("roland", "classpath:com/consol/citrus/ssh/private.key");66    }67    @Test(expectedExceptions = CitrusRuntimeException.class, expectedExceptionsMessageRegExp = ".*blubber\\.bla.*")68    public void notInClasspath() {69        new SinglePublicKeyAuthenticator("roland", "classpath:com/consol/citrus/ssh/blubber.bla");70    }71    @Test(expectedExceptions = CitrusRuntimeException.class,expectedExceptionsMessageRegExp = ".*/no/valid/path.*")72    public void invalidFilePath() {73        new SinglePublicKeyAuthenticator("roland","/no/valid/path");74    }75    76    /**77     * Gets public key instance from resource.78     * @param pResource79     * @return80     * @throws IOException81     */82    private PublicKey getPublicKey(String pResource) throws IOException {83        return getPublicKeyFromStream(getClass().getResourceAsStream(pResource));84    }85    /**86     * Creates new temporary file from resource.87     * @param pResource88     * @return89     * @throws IOException90     */91    private File copyToTempFile(String pResource) throws IOException {92        File temp = File.createTempFile("citrus-ssh", "pem");93        FileCopyUtils.copy(getClass().getResourceAsStream(pResource),94                           new FileOutputStream(temp));95        return temp;96    }97    98    /**99     * Create public key instance from file input stream.100     * @param is101     * @return102     * @throws IOException103     */104    private PublicKey getPublicKeyFromStream(InputStream is) throws IOException {105        Reader reader = new InputStreamReader(is);106        try {107            Object o = new PEMParser(reader).readObject();108            if (o instanceof PEMKeyPair) {109                return new BouncyCastleProvider().getPublicKey(((PEMKeyPair) o).getPublicKeyInfo());110            } else if (o instanceof SubjectPublicKeyInfo) {111                return new BouncyCastleProvider().getPublicKey((SubjectPublicKeyInfo) o);112            } else {113                throw new CitrusRuntimeException("Unable to read public key");114            }115        } catch (IOException e) {116            throw new CitrusRuntimeException("Failed to read public key", e);117        } finally {118            IoUtils.closeQuietly(is, reader);119        }120    }121}...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.