mirror of
https://github.com/google/nomulus.git
synced 2025-05-03 21:47:51 +02:00
6ff48b7dae
First step of RDE encoding refactoring. Creates a single InputStream (OutputStream) to decode (encode) Ghostryde files. This replaces the 3 InputStreams (OutputStreams) that were needed before. Also removes a lot of classes, and removes the "injection" of the Ghostryde class. It's an encoding, there's no point in injecting it. Finally, removed the buffer-size configuration and replaced with a static final const value. It's just a buffer size - it doesn't actually affect much. There are much more "important" fields that weren't configured (such as the compression algorithm and whether or not to do integrity checks) ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=202319102
115 lines
4.8 KiB
Java
115 lines
4.8 KiB
Java
// Copyright 2017 The Nomulus Authors. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package google.registry.tools;
|
|
|
|
import static com.google.common.truth.Truth.assertThat;
|
|
import static java.nio.charset.StandardCharsets.UTF_8;
|
|
|
|
import google.registry.keyring.api.Keyring;
|
|
import google.registry.rde.Ghostryde;
|
|
import google.registry.testing.BouncyCastleProviderRule;
|
|
import google.registry.testing.FakeKeyringModule;
|
|
import google.registry.testing.InjectRule;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.nio.file.Paths;
|
|
import org.junit.Before;
|
|
import org.junit.Rule;
|
|
import org.junit.Test;
|
|
|
|
/** Unit tests for {@link GhostrydeCommand}. */
|
|
public class GhostrydeCommandTest extends CommandTestCase<GhostrydeCommand> {
|
|
|
|
private static final byte[] SONG_BY_CHRISTINA_ROSSETTI = (""
|
|
+ "When I am dead, my dearest, \n"
|
|
+ " Sing no sad songs for me; \n"
|
|
+ "Plant thou no roses at my head, \n"
|
|
+ " Nor shady cypress tree: \n"
|
|
+ "Be the green grass above me \n"
|
|
+ " With showers and dewdrops wet; \n"
|
|
+ "And if thou wilt, remember, \n"
|
|
+ " And if thou wilt, forget. \n"
|
|
+ " \n"
|
|
+ "I shall not see the shadows, \n"
|
|
+ " I shall not feel the rain; \n"
|
|
+ "I shall not hear the nightingale \n"
|
|
+ " Sing on, as if in pain: \n"
|
|
+ "And dreaming through the twilight \n"
|
|
+ " That doth not rise nor set, \n"
|
|
+ "Haply I may remember, \n"
|
|
+ " And haply may forget. \n").getBytes(UTF_8);
|
|
|
|
@Rule
|
|
public final InjectRule inject = new InjectRule();
|
|
|
|
@Rule
|
|
public final BouncyCastleProviderRule bouncy = new BouncyCastleProviderRule();
|
|
|
|
private Keyring keyring;
|
|
|
|
@Before
|
|
public void before() {
|
|
keyring = new FakeKeyringModule().get();
|
|
command.rdeStagingDecryptionKey = keyring::getRdeStagingDecryptionKey;
|
|
command.rdeStagingEncryptionKey = keyring::getRdeStagingEncryptionKey;
|
|
}
|
|
|
|
@Test
|
|
public void testEncrypt_outputIsAFile_writesToFile() throws Exception {
|
|
Path inFile = Paths.get(tmpDir.newFile("atrain.txt").toString());
|
|
Path outFile = Paths.get(tmpDir.newFile().toString());
|
|
Files.write(inFile, SONG_BY_CHRISTINA_ROSSETTI);
|
|
runCommand("--encrypt", "--input=" + inFile, "--output=" + outFile);
|
|
byte[] decoded =
|
|
Ghostryde.decode(Files.readAllBytes(outFile), keyring.getRdeStagingDecryptionKey());
|
|
assertThat(decoded).isEqualTo(SONG_BY_CHRISTINA_ROSSETTI);
|
|
}
|
|
|
|
@Test
|
|
public void testEncrypt_outputIsADirectory_appendsGhostrydeExtension() throws Exception {
|
|
Path inFile = Paths.get(tmpDir.newFile("atrain.txt").toString());
|
|
Path outDir = Paths.get(tmpDir.newFolder().toString());
|
|
Files.write(inFile, SONG_BY_CHRISTINA_ROSSETTI);
|
|
runCommand("--encrypt", "--input=" + inFile, "--output=" + outDir);
|
|
Path lenOutFile = outDir.resolve("atrain.txt.length");
|
|
assertThat(Ghostryde.readLength(Files.newInputStream(lenOutFile)))
|
|
.isEqualTo(SONG_BY_CHRISTINA_ROSSETTI.length);
|
|
Path outFile = outDir.resolve("atrain.txt.ghostryde");
|
|
byte[] decoded =
|
|
Ghostryde.decode(Files.readAllBytes(outFile), keyring.getRdeStagingDecryptionKey());
|
|
assertThat(decoded).isEqualTo(SONG_BY_CHRISTINA_ROSSETTI);
|
|
}
|
|
|
|
@Test
|
|
public void testDecrypt_outputIsAFile_writesToFile() throws Exception {
|
|
Path inFile = Paths.get(tmpDir.newFile().toString());
|
|
Path outFile = Paths.get(tmpDir.newFile().toString());
|
|
Files.write(
|
|
inFile, Ghostryde.encode(SONG_BY_CHRISTINA_ROSSETTI, keyring.getRdeStagingEncryptionKey()));
|
|
runCommand("--decrypt", "--input=" + inFile, "--output=" + outFile);
|
|
assertThat(Files.readAllBytes(outFile)).isEqualTo(SONG_BY_CHRISTINA_ROSSETTI);
|
|
}
|
|
|
|
@Test
|
|
public void testDecrypt_outputIsADirectory_AppendsDecryptExtension() throws Exception {
|
|
Path inFile = Paths.get(tmpDir.newFolder().toString()).resolve("atrain.ghostryde");
|
|
Path outDir = Paths.get(tmpDir.newFolder().toString());
|
|
Files.write(
|
|
inFile, Ghostryde.encode(SONG_BY_CHRISTINA_ROSSETTI, keyring.getRdeStagingEncryptionKey()));
|
|
runCommand("--decrypt", "--input=" + inFile, "--output=" + outDir);
|
|
Path outFile = outDir.resolve("atrain.ghostryde.decrypt");
|
|
assertThat(Files.readAllBytes(outFile)).isEqualTo(SONG_BY_CHRISTINA_ROSSETTI);
|
|
}
|
|
}
|