45 lines
1.9 KiB
Java
45 lines
1.9 KiB
Java
// SPDX-FileCopyrightText: 2021 Paul Schaub <vanitasvitae@fsfe.org>
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package org.pgpainless.key.protection;
|
|
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
import static org.junit.jupiter.api.Assertions.assertNull;
|
|
|
|
import java.io.IOException;
|
|
import java.util.Map;
|
|
import java.util.concurrent.ConcurrentHashMap;
|
|
|
|
import org.bouncycastle.openpgp.PGPException;
|
|
import org.bouncycastle.openpgp.PGPSecretKeyRing;
|
|
import org.junit.jupiter.api.Test;
|
|
import org.pgpainless.key.TestKeys;
|
|
import org.pgpainless.key.protection.passphrase_provider.MapBasedPassphraseProvider;
|
|
import org.pgpainless.s2k.Passphrase;
|
|
|
|
public class MapBasedPassphraseProviderTest {
|
|
|
|
@Test
|
|
public void testMapBasedProvider() throws IOException, PGPException {
|
|
Map<Long, Passphrase> passphraseMap = new ConcurrentHashMap<>();
|
|
passphraseMap.put(1L, Passphrase.fromPassword("tiger"));
|
|
passphraseMap.put(123123123L, Passphrase.fromPassword("snake"));
|
|
passphraseMap.put(69696969L, Passphrase.emptyPassphrase());
|
|
MapBasedPassphraseProvider provider = new MapBasedPassphraseProvider(passphraseMap);
|
|
|
|
assertEquals(Passphrase.fromPassword("tiger"), provider.getPassphraseFor(1L));
|
|
assertEquals(Passphrase.fromPassword("snake"), provider.getPassphraseFor(123123123L));
|
|
assertEquals(Passphrase.emptyPassphrase(), provider.getPassphraseFor(69696969L));
|
|
assertNull(provider.getPassphraseFor(555L));
|
|
|
|
PGPSecretKeyRing secretKeys = TestKeys.getCryptieSecretKeyRing();
|
|
passphraseMap = new ConcurrentHashMap<>();
|
|
passphraseMap.put(secretKeys.getSecretKey().getKeyID(), TestKeys.CRYPTIE_PASSPHRASE);
|
|
provider = new MapBasedPassphraseProvider(passphraseMap);
|
|
|
|
assertEquals(TestKeys.CRYPTIE_PASSPHRASE, provider.getPassphraseFor(secretKeys.getSecretKey()));
|
|
assertNull(provider.getPassphraseFor(TestKeys.getEmilSecretKeyRing().getSecretKey()));
|
|
}
|
|
}
|