Function TestRoundTrip

repo/encryption/encryption_test.go:25–95

nolint:gocyclo

(t *testing.T)

//nolint:gocyclo
func TestRoundTrip(t *testing.T) {
	data := make([]byte, 100)
	rand.Read(data)

	masterKey := make([]byte, 32)
	rand.Read(masterKey)

	contentID1 := make([]byte, 16)
	rand.Read(contentID1)

	contentID2 := make([]byte, 16)
	rand.Read(contentID2)

	for _, encryptionAlgo := range encryption.SupportedAlgorithms(true) {
		t.Run(encryptionAlgo, func(t *testing.T) {
			e, err := encryption.CreateEncryptor(parameters{encryptionAlgo, masterKey})
			if err != nil {
				t.Fatal(err)
			}

			var cipherText1 gather.WriteBuffer
			defer cipherText1.Close()

			var cipherText1b gather.WriteBuffer
			defer cipherText1b.Close()

			require.NoError(t, e.Encrypt(gather.FromSlice(data), contentID1, &cipherText1))
			require.NoError(t, e.Encrypt(gather.FromSlice(data), contentID1, &cipherText1b))

			if v := cipherText1.ToByteSlice(); bytes.Equal(v, cipherText1b.ToByteSlice()) {
				t.Errorf("multiple Encrypt returned the same ciphertext: %x", v)
			}

			var plainText1 gather.WriteBuffer
			defer plainText1.Close()

			require.NoError(t, e.Decrypt(cipherText1.Bytes(), contentID1, &plainText1))

			if v := plainText1.ToByteSlice(); !bytes.Equal(v, data) {
				t.Errorf("Encrypt()/Decrypt() does not round-trip: %x %x", v, data)
			}

			var cipherText2 gather.WriteBuffer
			defer cipherText2.Close()

			require.NoError(t, e.Encrypt(gather.FromSlice(data), contentID2, &cipherText2))

			var plainText2 gather.WriteBuffer
			defer plainText2.Close()

			require.NoError(t, e.Decrypt(cipherText2.Bytes(), contentID2, &plainText2))

			if v := plainText2.ToByteSlice(); !bytes.Equal(v, data) {
				t.Errorf("Encrypt()/Decrypt() does not round-trip: %x %x", v, data)
			}

			if v := cipherText1.ToByteSlice(); bytes.Equal(v, cipherText2.ToByteSlice()) {
				t.Errorf("ciphertexts should be different, were %x", v)