Master (#1262)

* clear * feat: Add everything * fix: Commenht
2026-08-04 18:31:41 +00:00 · 2025-10-03 14:45:52 -04:00
parent 43b4a11c06
commit 13e6c3e84d
1935 changed files with 655061 additions and 40058 deletions
@@ -0,0 +1,251 @@
+package keeper_test
+
+import (
+	"crypto/ed25519"
+	"crypto/rand"
+	"encoding/base64"
+	"encoding/hex"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/sonr-io/sonr/x/did/types"
+)
+
+func TestVerifyDIDDocumentSignature(t *testing.T) {
+	f := SetupTest(t)
+
+	// Generate Ed25519 key pair for testing
+	publicKey, privateKey, err := ed25519.GenerateKey(rand.Reader)
+	require.NoError(t, err)
+
+	// Create test DID document
+	did := "did:sonr:test123"
+	didDoc := &types.DIDDocument{
+		Id:                did,
+		PrimaryController: "did:sonr:controller123",
+		VerificationMethod: []*types.VerificationMethod{
+			{
+				Id:                     did + "#key1",
+				VerificationMethodKind: "Ed25519VerificationKey2020",
+				Controller:             did,
+				PublicKeyBase64:        base64.StdEncoding.EncodeToString(publicKey),
+			},
+		},
+		Deactivated: false,
+		Version:     1,
+		CreatedAt:   1234567890,
+		UpdatedAt:   1234567890,
+	}
+
+	// Store the DID document
+	ormDoc := didDoc.ToORM()
+	err = f.k.OrmDB.DIDDocumentTable().Insert(f.ctx, ormDoc)
+	require.NoError(t, err)
+
+	// Test signature verification
+	testCases := []struct {
+		name           string
+		did            string
+		signature      []byte
+		expectedResult bool
+		expectedError  bool
+	}{
+		{
+			name:           "Valid signature",
+			did:            did,
+			signature:      createEd25519Signature(privateKey, []byte("test message")),
+			expectedResult: true,
+			expectedError:  false,
+		},
+		{
+			name:           "Invalid signature",
+			did:            did,
+			signature:      []byte("invalid signature"),
+			expectedResult: false,
+			expectedError:  true,
+		},
+		{
+			name:           "Non-existent DID",
+			did:            "did:sonr:nonexistent",
+			signature:      createEd25519Signature(privateKey, []byte("test message")),
+			expectedResult: false,
+			expectedError:  true,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			result, err := f.k.VerifyDIDDocumentSignature(f.ctx, tc.did, tc.signature)
+
+			if tc.expectedError {
+				require.Error(t, err)
+			} else {
+				require.NoError(t, err)
+				require.Equal(t, tc.expectedResult, result)
+			}
+		})
+	}
+}
+
+func TestVerifyDIDDocumentSignature_DeactivatedDID(t *testing.T) {
+	f := SetupTest(t)
+
+	// Generate Ed25519 key pair for testing
+	publicKey, _, err := ed25519.GenerateKey(rand.Reader)
+	require.NoError(t, err)
+
+	// Create test DID document that is deactivated
+	did := "did:sonr:deactivated123"
+	didDoc := &types.DIDDocument{
+		Id:                did,
+		PrimaryController: "did:sonr:controller123",
+		VerificationMethod: []*types.VerificationMethod{
+			{
+				Id:                     did + "#key1",
+				VerificationMethodKind: "Ed25519VerificationKey2020",
+				Controller:             did,
+				PublicKeyBase64:        base64.StdEncoding.EncodeToString(publicKey),
+			},
+		},
+		Deactivated: true, // This is deactivated
+		Version:     1,
+		CreatedAt:   1234567890,
+		UpdatedAt:   1234567890,
+	}
+
+	// Store the DID document
+	ormDoc := didDoc.ToORM()
+	err = f.k.OrmDB.DIDDocumentTable().Insert(f.ctx, ormDoc)
+	require.NoError(t, err)
+
+	// Test signature verification should fail for deactivated DID
+	result, err := f.k.VerifyDIDDocumentSignature(f.ctx, did, []byte("any signature"))
+	require.Error(t, err)
+	require.False(t, result)
+	require.Contains(t, err.Error(), "deactivated")
+}
+
+func TestVerifyDIDDocumentSignature_MultipleVerificationMethods(t *testing.T) {
+	f := SetupTest(t)
+
+	// Generate Ed25519 key pairs for testing
+	publicKey1, privateKey1, err := ed25519.GenerateKey(rand.Reader)
+	require.NoError(t, err)
+	publicKey2, _, err := ed25519.GenerateKey(rand.Reader)
+	require.NoError(t, err)
+
+	// Create test DID document with multiple verification methods
+	did := "did:sonr:multi123"
+	didDoc := &types.DIDDocument{
+		Id:                did,
+		PrimaryController: "did:sonr:controller123",
+		VerificationMethod: []*types.VerificationMethod{
+			{
+				Id:                     did + "#key1",
+				VerificationMethodKind: "Ed25519VerificationKey2020",
+				Controller:             did,
+				PublicKeyBase64:        base64.StdEncoding.EncodeToString(publicKey1),
+			},
+			{
+				Id:                     did + "#key2",
+				VerificationMethodKind: "Ed25519VerificationKey2020",
+				Controller:             did,
+				PublicKeyHex:           hex.EncodeToString(publicKey2),
+			},
+		},
+		Deactivated: false,
+		Version:     1,
+		CreatedAt:   1234567890,
+		UpdatedAt:   1234567890,
+	}
+
+	// Store the DID document
+	ormDoc := didDoc.ToORM()
+	err = f.k.OrmDB.DIDDocumentTable().Insert(f.ctx, ormDoc)
+	require.NoError(t, err)
+
+	// Test signature verification with first key should succeed
+	signature1 := createEd25519Signature(privateKey1, []byte("test message"))
+	result, err := f.k.VerifyDIDDocumentSignature(f.ctx, did, signature1)
+	require.NoError(t, err)
+	require.True(t, result)
+}
+
+func TestVerifyDIDDocumentSignature_UnsupportedVerificationMethod(t *testing.T) {
+	f := SetupTest(t)
+
+	// Create test DID document with unsupported verification method
+	did := "did:sonr:unsupported123"
+	didDoc := &types.DIDDocument{
+		Id:                did,
+		PrimaryController: "did:sonr:controller123",
+		VerificationMethod: []*types.VerificationMethod{
+			{
+				Id:                     did + "#key1",
+				VerificationMethodKind: "UnsupportedMethod2020",
+				Controller:             did,
+				PublicKeyBase64:        "dummy-key",
+			},
+		},
+		Deactivated: false,
+		Version:     1,
+		CreatedAt:   1234567890,
+		UpdatedAt:   1234567890,
+	}
+
+	// Store the DID document
+	ormDoc := didDoc.ToORM()
+	err := f.k.OrmDB.DIDDocumentTable().Insert(f.ctx, ormDoc)
+	require.NoError(t, err)
+
+	// Test signature verification should fail for unsupported method
+	result, err := f.k.VerifyDIDDocumentSignature(f.ctx, did, []byte("any signature"))
+	require.Error(t, err)
+	require.False(t, result)
+	require.Contains(t, err.Error(), "signature verification failed")
+}
+
+// TestVerifyDIDDocumentSignature_WebAuthnVerificationMethod - REMOVED
+// This test was testing deprecated WebAuthn signature verification functionality
+// that has been replaced with the gasless transaction approach.
+
+func TestVerifyDIDDocumentSignature_JsonWebSignature2020(t *testing.T) {
+	f := SetupTest(t)
+
+	// Create test DID document with JWS verification method
+	did := "did:sonr:jws123"
+	didDoc := &types.DIDDocument{
+		Id:                did,
+		PrimaryController: "did:sonr:controller123",
+		VerificationMethod: []*types.VerificationMethod{
+			{
+				Id:                     did + "#jws1",
+				VerificationMethodKind: "JsonWebSignature2020",
+				Controller:             did,
+				PublicKeyJwk:           `{"kty":"OKP","crv":"Ed25519","x":"dummy-key"}`,
+			},
+		},
+		Deactivated: false,
+		Version:     1,
+		CreatedAt:   1234567890,
+		UpdatedAt:   1234567890,
+	}
+
+	// Store the DID document
+	ormDoc := didDoc.ToORM()
+	err := f.k.OrmDB.DIDDocumentTable().Insert(f.ctx, ormDoc)
+	require.NoError(t, err)
+
+	// Test signature verification with JWS method
+	// Note: This will fail since we don't have a real JWS signature
+	jwsSignature := `{"signature":"dummy-signature","protected":"dummy-protected","header":{}}`
+	result, err := f.k.VerifyDIDDocumentSignature(f.ctx, did, []byte(jwsSignature))
+	require.Error(t, err)
+	require.False(t, result)
+}
+
+// Helper function to create Ed25519 signature
+func createEd25519Signature(privateKey ed25519.PrivateKey, message []byte) []byte {
+	return ed25519.Sign(privateKey, message)
+}