mirror of
				https://source.quilibrium.com/quilibrium/ceremonyclient.git
				synced 2025-11-04 00:27:41 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			60 lines
		
	
	
		
			1.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			60 lines
		
	
	
		
			1.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
//
 | 
						|
// Copyright Coinbase, Inc. All Rights Reserved.
 | 
						|
//
 | 
						|
// SPDX-License-Identifier: Apache-2.0
 | 
						|
//
 | 
						|
 | 
						|
package elgamal
 | 
						|
 | 
						|
import (
 | 
						|
	"testing"
 | 
						|
 | 
						|
	"github.com/stretchr/testify/require"
 | 
						|
 | 
						|
	"source.quilibrium.com/quilibrium/monorepo/nekryptology/pkg/core/curves"
 | 
						|
)
 | 
						|
 | 
						|
func TestDecryptionKeyMarshal(t *testing.T) {
 | 
						|
	k256 := curves.K256()
 | 
						|
	_, dk, err := NewKeys(k256)
 | 
						|
	require.NoError(t, err)
 | 
						|
	bin, err := dk.MarshalBinary()
 | 
						|
	require.NoError(t, err)
 | 
						|
	dup := new(DecryptionKey)
 | 
						|
	err = dup.UnmarshalBinary(bin)
 | 
						|
	require.NoError(t, err)
 | 
						|
	require.Equal(t, 0, dk.x.Cmp(dup.x))
 | 
						|
}
 | 
						|
 | 
						|
func TestDecryptionKeyDecryptBadCiphertext(t *testing.T) {
 | 
						|
	k256 := curves.K256()
 | 
						|
	_, dk, err := NewKeys(k256)
 | 
						|
	require.NoError(t, err)
 | 
						|
 | 
						|
	// nil ciphertext
 | 
						|
	_, _, err = dk.VerifiableDecryptWithDomain([]byte{}, nil)
 | 
						|
	require.Error(t, err)
 | 
						|
 | 
						|
	// empty ciphertext
 | 
						|
	_, _, err = dk.VerifiableDecryptWithDomain([]byte{}, new(CipherText))
 | 
						|
	require.Error(t, err)
 | 
						|
 | 
						|
	cs := new(CipherText)
 | 
						|
	cs.C1 = k256.Point.Generator()
 | 
						|
	cs.C2 = k256.Point.Generator()
 | 
						|
	cs.Nonce = make([]byte, 12)
 | 
						|
	cs.Aead = make([]byte, 16)
 | 
						|
 | 
						|
	// empty data in ciphertext
 | 
						|
	_, _, err = dk.VerifiableDecryptWithDomain([]byte{}, cs)
 | 
						|
	require.Error(t, err)
 | 
						|
 | 
						|
	cs.C1 = k256.Point.Identity()
 | 
						|
	cs.C2 = k256.Point.Identity()
 | 
						|
	cs.Nonce = []byte{}
 | 
						|
	cs.Aead = []byte{}
 | 
						|
	// ensure no panic happens when nonce and aead are invalid lengths
 | 
						|
	_, _, err = dk.VerifiableDecryptWithDomain([]byte{}, cs)
 | 
						|
	require.Error(t, err)
 | 
						|
}
 |