PGPainless/pgpainless
PGPainless/pgpainless
1
0
Fork 0
mirror of https://github.com/pgpainless/pgpainless.git synced 2024-12-25 12:27:58 +01:00
Code Issues Releases Wiki Activity

Improve image attribute test

Browse source
This commit is contained in:
Paul Schaub 2024-02-02 16:39:19 +01:00
parent 0d3fb446a8
commit 05c203177e
Signed by: vanitasvitae
GPG key ID: 62BEE9264BF17311
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
pgpainless-core/src/test/kotlin/org/pgpainless/key/generation/GenerateOpenPgpKeyTest.kt
View file

@ -4,8 +4,11 @@
package org.pgpainless.key.generation package org.pgpainless.key.generation
import java.io.InputStream
import org.bouncycastle.bcpg.attr.ImageAttribute import org.bouncycastle.bcpg.attr.ImageAttribute
import org.bouncycastle.openpgp.PGPUserAttributeSubpacketVectorGenerator import org.bouncycastle.openpgp.PGPUserAttributeSubpacketVectorGenerator
import org.bouncycastle.util.io.Streams
import org.junit.jupiter.api.Assertions.assertArrayEquals
import org.junit.jupiter.api.Assertions.assertTrue import org.junit.jupiter.api.Assertions.assertTrue
import org.junit.jupiter.api.Test import org.junit.jupiter.api.Test
import org.junit.jupiter.api.assertThrows import org.junit.jupiter.api.assertThrows
@ -20,7 +23,6 @@ import org.pgpainless.key.generation.type.xdh.XDHSpec
import org.pgpainless.key.protection.SecretKeyRingProtector import org.pgpainless.key.protection.SecretKeyRingProtector
import org.pgpainless.policy.Policy import org.pgpainless.policy.Policy
import org.pgpainless.util.DateUtil import org.pgpainless.util.DateUtil
import java.io.InputStream
class GenerateOpenPgpKeyTest { class GenerateOpenPgpKeyTest {
@ -83,12 +85,16 @@ class GenerateOpenPgpKeyTest {
@Test @Test
fun testKeyGenerationWithJPEGAttribute() { fun testKeyGenerationWithJPEGAttribute() {
val key = GenerateOpenPgpKey(Policy.getInstance()) val key =
GenerateOpenPgpKey(Policy.getInstance())
.buildV4Key(KeyType.EDDSA(EdDSACurve._Ed25519)) .buildV4Key(KeyType.EDDSA(EdDSACurve._Ed25519))
.addJpegImage(requireResource("suzanne.jpg")) .addJpegImage(requireResource("suzanne.jpg"))
.build() .build()
assertTrue(key.publicKey.userAttributes.hasNext()) assertTrue(key.publicKey.userAttributes.hasNext())
assertArrayEquals(
Streams.readAll(requireResource("suzanne.jpg")),
key.publicKey.userAttributes.next().imageAttribute.imageData)
} }
private fun requireResource(resourceName: String): InputStream { private fun requireResource(resourceName: String): InputStream {