106 lines
3.4 KiB
Go
106 lines
3.4 KiB
Go
package signature
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// goldenExpiresUnix is the fixed expiration timestamp used by all golden
|
|
// vectors: 2024-01-01T00:00:00Z.
|
|
const goldenExpiresUnix int64 = 1704067200
|
|
|
|
// goldenSigningKey is the fixed signing key used by all golden vectors.
|
|
const goldenSigningKey = "golden-test-key"
|
|
|
|
// TestSigner_GoldenVectors pins the exact HMAC-SHA256 signature output and
|
|
// the exact generated signed URL path for fully-specified requests with a
|
|
// hardcoded signing key. The expected values were computed once and are
|
|
// hardcoded here as known answers.
|
|
//
|
|
// If any of these assertions fail, the signed byte format
|
|
// ("host:path:query:width:height:format:expiration"), the base64url
|
|
// encoding, or the signed URL layout has changed. Such a change breaks
|
|
// every signature already issued to clients, so it must be made
|
|
// deliberately: update these constants only as part of an intentional,
|
|
// documented signature format migration.
|
|
func TestSigner_GoldenVectors(t *testing.T) {
|
|
signer := New(goldenSigningKey)
|
|
|
|
vectors := []struct {
|
|
name string
|
|
req Request
|
|
// wantSignature is the exact base64url (RFC 4648 URL-safe,
|
|
// padded) HMAC-SHA256 signature for the request with Expires
|
|
// set to goldenExpiresUnix.
|
|
wantSignature string
|
|
// wantSignedPath is the exact path returned by
|
|
// GenerateSignedURL for the request. The signature and
|
|
// expiration are returned separately by GenerateSignedURL and
|
|
// are not embedded in the path.
|
|
wantSignedPath string
|
|
}{
|
|
{
|
|
name: "resized without query",
|
|
req: Request{
|
|
SourceHost: "cdn.example.com",
|
|
SourcePath: "/photos/cat.jpg",
|
|
SourceQuery: "",
|
|
Width: 800,
|
|
Height: 600,
|
|
Format: "webp",
|
|
},
|
|
// Signed data: "cdn.example.com:/photos/cat.jpg::800:600:webp:1704067200"
|
|
wantSignature: "x5PfPp8QSDo0cJT96od-AEgrQyOVLfqifH5sst61_-w=",
|
|
wantSignedPath: "/v1/image/cdn.example.com/photos/cat.jpg/800x600.webp",
|
|
},
|
|
{
|
|
name: "resized with query string",
|
|
req: Request{
|
|
SourceHost: "cdn.example.com",
|
|
SourcePath: "/photos/cat.jpg",
|
|
SourceQuery: "token=abc&v=2",
|
|
Width: 800,
|
|
Height: 600,
|
|
Format: "webp",
|
|
},
|
|
// Signed data: "cdn.example.com:/photos/cat.jpg:token=abc&v=2:800:600:webp:1704067200"
|
|
wantSignature: "394_Vf9TdQFkpQ3XKFDQSyxgqKq8N7mApf2S4QaHqyo=",
|
|
wantSignedPath: "/v1/image/cdn.example.com/photos/cat.jpg%3Ftoken=abc&v=2/800x600.webp",
|
|
},
|
|
{
|
|
name: "original size without query",
|
|
req: Request{
|
|
SourceHost: "cdn.example.com",
|
|
SourcePath: "/photos/cat.jpg",
|
|
SourceQuery: "",
|
|
Width: 0,
|
|
Height: 0,
|
|
Format: "png",
|
|
},
|
|
// Signed data: "cdn.example.com:/photos/cat.jpg::0:0:png:1704067200"
|
|
wantSignature: "7Be7oteeQwvnSPU4bchyQ4ZGYGsAGBKpeEtuQ02ox60=",
|
|
wantSignedPath: "/v1/image/cdn.example.com/photos/cat.jpg/orig.png",
|
|
},
|
|
}
|
|
|
|
for _, tt := range vectors {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
signReq := tt.req
|
|
signReq.Expires = time.Unix(goldenExpiresUnix, 0)
|
|
|
|
gotSignature := signer.Sign(&signReq)
|
|
if gotSignature != tt.wantSignature {
|
|
t.Errorf("Sign() = %q, want %q (signed byte format changed?)",
|
|
gotSignature, tt.wantSignature)
|
|
}
|
|
|
|
urlReq := tt.req
|
|
gotPath, _, _ := signer.GenerateSignedURL(&urlReq, time.Hour)
|
|
if gotPath != tt.wantSignedPath {
|
|
t.Errorf("GenerateSignedURL() path = %q, want %q (signed URL layout changed?)",
|
|
gotPath, tt.wantSignedPath)
|
|
}
|
|
})
|
|
}
|
|
}
|