PGPainless/pgpainless
PGPainless/pgpainless
1
0
Fork 0
mirror of https://github.com/pgpainless/pgpainless.git synced 2025-12-10 14:21:09 +01:00
Code Issues Releases Wiki Activity

Add test for CompressionAlgorithmNegotiator

Browse source
This commit is contained in:
Paul Schaub 2025-06-03 12:25:40 +02:00
parent 21439854e3
commit d789d3e0c4
Signed by: vanitasvitae
GPG key ID: 62BEE9264BF17311
2 changed files with 71 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

52
pgpainless-core/src/test/java/org/pgpainless/algorithm/negotiation/CompressionAlgorithmNegotiatorTest.java Normal file
View file

@ -0,0 +1,52 @@
// SPDX-FileCopyrightText: 2025 Paul Schaub <vanitasvitae@fsfe.org>
//
// SPDX-License-Identifier: Apache-2.0
package org.pgpainless.algorithm.negotiation;
import org.junit.jupiter.api.Test;
import org.pgpainless.PGPainless;
import org.pgpainless.algorithm.CompressionAlgorithm;
import org.pgpainless.policy.Policy;
import java.util.Arrays;
import java.util.Collections;
import static org.junit.jupiter.api.Assertions.assertEquals;
public class CompressionAlgorithmNegotiatorTest {
@Test
public void staticNegotiateWithoutOverride() {
Policy policy = PGPainless.getInstance().getAlgorithmPolicy()
.copy()
.withCompressionAlgorithmPolicy(new Policy.CompressionAlgorithmPolicy(
CompressionAlgorithm.BZIP2,
Arrays.asList(CompressionAlgorithm.BZIP2, CompressionAlgorithm.UNCOMPRESSED)
))
.build();
CompressionAlgorithmNegotiator negotiator = CompressionAlgorithmNegotiator.staticNegotiation();
// If the user did not pass an override, return the policy default
assertEquals(
CompressionAlgorithm.BZIP2,
negotiator.negotiate(policy, null, Collections.emptySet()));
}
@Test
public void staticNegotiateWithOverride() {
Policy policy = PGPainless.getInstance().getAlgorithmPolicy()
.copy()
.withCompressionAlgorithmPolicy(new Policy.CompressionAlgorithmPolicy(
CompressionAlgorithm.BZIP2,
Arrays.asList(CompressionAlgorithm.BZIP2, CompressionAlgorithm.UNCOMPRESSED)
))
.build();
CompressionAlgorithmNegotiator negotiator = CompressionAlgorithmNegotiator.staticNegotiation();
// If the user passed an override, return that
assertEquals(
CompressionAlgorithm.ZLIB,
negotiator.negotiate(policy, CompressionAlgorithm.ZLIB, Collections.emptySet()));
}
}