From a40a9337653613ae46215bc0717e639bbaea997c Mon Sep 17 00:00:00 2001
From: Emmi Rehn <emmi.rehn@csc.fi>
Date: Wed, 2 Aug 2023 12:46:59 +0300
Subject: [PATCH] Add test

---
 streaming/streaming_test.go | 61 +++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/streaming/streaming_test.go b/streaming/streaming_test.go
index 02669f9..16ac29f 100644
--- a/streaming/streaming_test.go
+++ b/streaming/streaming_test.go
@@ -540,3 +540,64 @@ func TestFileReEncryption(t *testing.T) {
 		t.Fail()
 	}
 }
+
+func TestNewCrypt4GHWriterWithNonces(t *testing.T) {
+	inFile, err := os.Open("../test/sample.txt")
+	if err != nil {
+		t.Error(err)
+	}
+	readerPublicKey, err := keys.ReadPublicKey(strings.NewReader(crypt4ghX25519Pub))
+	if err != nil {
+		t.Error(err)
+	}
+	buffer := bytes.Buffer{}
+	readerPublicKeyList := [][chacha20poly1305.KeySize]byte{}
+	readerPublicKeyList = append(readerPublicKeyList, readerPublicKey)
+	readerPublicKeyList = append(readerPublicKeyList, readerPublicKey)
+	readerPublicKeyList = append(readerPublicKeyList, readerPublicKey)
+	if len(readerPublicKeyList) != 3 {
+		t.Errorf("expected %d public keys in list but got %d", 3, len(readerPublicKeyList))
+	}
+	_, privateKey, err := keys.GenerateKeyPair()
+	if err != nil {
+		return
+	}
+	writer, err := NewCrypt4GHWriter(&buffer, privateKey, readerPublicKeyList, nil, nil)
+	if err != nil {
+		t.Error(err)
+	}
+	_, err = io.Copy(writer, inFile)
+	if err != nil {
+		t.Error(err)
+	}
+	_, err = inFile.Seek(0, io.SeekStart)
+	if err != nil {
+		t.Error(err)
+	}
+	err = writer.Close()
+	if err != nil {
+		t.Error(err)
+	}
+
+	buffer2 := bytes.Buffer{}
+	writer2, err := NewCrypt4GHWriter(&buffer2, privateKey, readerPublicKeyList, nil, writer.Rands)
+	if err != nil {
+		t.Error(err)
+	}
+	_, err = io.Copy(writer2, inFile)
+	if err != nil {
+		t.Error(err)
+	}
+	err = inFile.Close()
+	if err != nil {
+		t.Error(err)
+	}
+	err = writer2.Close()
+	if err != nil {
+		t.Error(err)
+	}
+
+	if bytes.Compare(buffer.Bytes(), buffer2.Bytes()) != 0 {
+		t.Fail()
+	}
+}