Update tests

Author: AlCutter

note/note_cosigv1_test.go

@@ -87,6 +87,8 @@ func TestMLDSASignerVerifierRoundtrip(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	t.Logf("s.KeyHash(): %08x, v.KeyHash(): %08x", s.KeyHash(), v.KeyHash())
+
 	if _, err := note.Open(n, note.VerifierList(v)); err != nil {
 		t.Fatal(err)
 	}
@@ -286,6 +288,74 @@ func TestVKeyToCosignatureV1(t *testing.T) {
 	if _, err = note.Open(n, note.VerifierList(v)); err == nil {
 			t.Errorf("Expected error trying to open cosigned note with standard vkey, but got success")
 	}
+
+	// Check that VKeyToCosignatureV1 fails for MLDSA keys.
+	_, mlVkey := mustGenerateMLDSAKey(t, "mldsa")
+	if _, err := VKeyToCosignatureV1(mlVkey); err == nil {
+		t.Errorf("Expected error for MLDSA key in VKeyToCosignatureV1, got success")
+	}
+}
+
+func TestSubtreeRoundtrip(t *testing.T) {
+	skey, vkey := mustGenerateMLDSAKey(t, "mldsa")
+	
+	signer, err := NewMLDSASigner(skey)
+	if err != nil {
+		t.Fatal(err)
+	}
+	
+	verifier, err := NewMLDSAVerifier(vkey)
+	if err != nil {
+		t.Fatal(err)
+	}
+	
+	origin := "test-log"
+	var start uint64 = 0
+	var end uint64 = 10
+	root := make([]byte, 32)
+	if _, err := rand.Read(root); err != nil {
+		t.Fatal(err)
+	}
+	timestamp := uint64(time.Now().Unix())
+	
+	sig, err := signer.SignSubtree(timestamp, origin, start, end, root)
+	if err != nil {
+		t.Fatal(err)
+	}
+	
+	if !verifier.VerifySubtree(timestamp, origin, start, end, root, sig) {
+		t.Error("Failed to verify valid subtree signature")
+	}
+	
+	// Test failure cases
+	wrongRoot := make([]byte, 32)
+	wrongRoot[0] = 1
+	if verifier.VerifySubtree(timestamp, origin, start, end, wrongRoot, sig) {
+		t.Error("VerifySubtree succeeded with wrong root")
+	}
+	
+	if verifier.VerifySubtree(timestamp, "wrong origin", start, end, root, sig) {
+		t.Error("VerifySubtree succeeded with wrong origin")
+	}
+}
+
+func TestMLDSAInvalidTimestamp(t *testing.T) {
+	skey, _ := mustGenerateMLDSAKey(t, "mldsa")
+	signer, err := NewMLDSASigner(skey)
+	if err != nil {
+		t.Fatal(err)
+	}
+	
+	origin := "test-log"
+	var start uint64 = 10 // > 0
+	var end uint64 = 20
+	root := make([]byte, 32)
+	timestamp := uint64(time.Now().Unix()) // > 0
+	
+	_, err = signer.SignSubtree(timestamp, origin, start, end, root)
+	if err == nil {
+		t.Error("Expected error for invalid timestamp (start > 0 && timestamp > 0), got nil")
+	}
 }
 
 func mustGenerateEd25519Key(t *testing.T, name string) (string, string) {