From 5269681c26a8b276b4b309e332592c3f1e226a84 Mon Sep 17 00:00:00 2001
From: Naoki Kosaka <naoki_kosaka@yukimochi.jp>
Date: Fri, 17 Sep 2021 09:49:54 +0900
Subject: [PATCH] Add sender_test.

---
 deliver/sender_test.go | 58 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 deliver/sender_test.go

diff --git a/deliver/sender_test.go b/deliver/sender_test.go
new file mode 100644
index 0000000..de761fa
--- /dev/null
+++ b/deliver/sender_test.go
@@ -0,0 +1,58 @@
+package deliver
+
+import (
+	"bytes"
+	"crypto/rsa"
+	"crypto/sha256"
+	"crypto/x509"
+	"encoding/base64"
+	"encoding/pem"
+	"github.com/Songmu/go-httpdate"
+	"github.com/go-fed/httpsig"
+	"io/ioutil"
+	"net/http"
+	"os"
+	"testing"
+	"time"
+)
+
+func generatePublicKeyPEMString(publicKey *rsa.PublicKey) string {
+	publicKeyByte := x509.MarshalPKCS1PublicKey(publicKey)
+	publicKeyPem := pem.EncodeToMemory(
+		&pem.Block{
+			Type:  "RSA PUBLIC KEY",
+			Bytes: publicKeyByte,
+		},
+	)
+	return string(publicKeyPem)
+}
+
+func TestAppendSignature(t *testing.T) {
+	file, _ := os.Open("../misc/test/create.json")
+	body, _ := ioutil.ReadAll(file)
+	req, _ := http.NewRequest("POST", "https://localhost", bytes.NewBuffer(body))
+	req.Header.Set("Content-Type", "application/activity+json")
+	req.Header.Set("Date", httpdate.Time2Str(time.Now()))
+	appendSignature(req, &body, "https://innocent.yukimochi.io/users/YUKIMOCHI#main-key", globalConfig.ActorKey())
+
+	// Verify HTTPSignature
+	verifier, err := httpsig.NewVerifier(req)
+	if err != nil {
+		t.Fatalf("Failed - " + err.Error())
+	}
+	err = verifier.Verify(globalConfig.ActorKey().Public(), httpsig.RSA_SHA256)
+	if err != nil {
+		t.Fatalf("Failed - " + err.Error())
+	}
+
+	// Verify Digest
+	givenDigest := req.Header.Get("Digest")
+	hash := sha256.New()
+	hash.Write(body)
+	b := hash.Sum(nil)
+	calculatedDigest := "SHA-256=" + base64.StdEncoding.EncodeToString(b)
+
+	if givenDigest != calculatedDigest {
+		t.Fatalf("Failed - " + err.Error())
+	}
+}