Update tests to better reflect real sigs

transparency-dev · Jan 31, 2024 · 612194f · 612194f
1 parent 826ad2f
commit 612194f
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/internal/distribute/rest/distribute_test.go b/internal/distribute/rest/distribute_test.go
@@ -23,6 +23,7 @@ import (
 	"testing"
 
 	"github.com/gorilla/mux"
+	f_note "github.com/transparency-dev/formats/note"
 	"github.com/transparency-dev/witness/internal/config"
 	"github.com/transparency-dev/witness/internal/distribute/rest"
 	"github.com/transparency-dev/witness/monitoring"
@@ -50,7 +51,7 @@ func TestDistributeOnce(t *testing.T) {
 		t.Fatal(err)
 	}
 	logs := []config.Log{log}
-	wV, err := note.NewVerifier(wPK)
+	wV, err := f_note.NewVerifierForCosignatureV1(wPK)
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -59,7 +60,7 @@ func TestDistributeOnce(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	wSign, err := note.NewSigner(wSK)
+	wSign, err := f_note.NewSignerForCosignatureV1(wSK)
 	if err != nil {
 		t.Fatal(err)
 	}

diff --git a/internal/http/server_test.go b/internal/http/server_test.go
@@ -28,6 +28,7 @@ import (
 	"testing"
 
 	"github.com/gorilla/mux"
+	f_note "github.com/transparency-dev/formats/note"
 	"github.com/transparency-dev/merkle/rfc6962"
 	"github.com/transparency-dev/witness/api"
 	"github.com/transparency-dev/witness/internal/persistence/inmemory"
@@ -74,7 +75,7 @@ type logOpts struct {
 
 func newWitness(t *testing.T, logs []logOpts) *witness.Witness {
 	// Set up Opts for the witness.
-	ns, err := note.NewSigner(wSK)
+	ns, err := f_note.NewSignerForCosignatureV1(wSK)
 	if err != nil {
 		t.Fatalf("couldn't create a witness signer: %v", err)
 	}

diff --git a/internal/witness/witness_test.go b/internal/witness/witness_test.go
@@ -22,6 +22,7 @@ import (
 	"testing"
 
 	_ "github.com/mattn/go-sqlite3" // Load drivers for sqlite3
+	f_note "github.com/transparency-dev/formats/note"
 	"github.com/transparency-dev/merkle/rfc6962"
 	"github.com/transparency-dev/witness/internal/persistence/inmemory"
 	"golang.org/x/mod/sumdb/note"
@@ -71,7 +72,7 @@ type logOpts struct {
 
 func newWitness(t *testing.T, logs []logOpts) *Witness {
 	// Set up Opts for the witness.
-	ns, err := note.NewSigner(wSK)
+	ns, err := f_note.NewSignerForCosignatureV1(wSK)
 	if err != nil {
 		t.Fatalf("couldn't create a witness signer: %v", err)
 	}
@@ -235,7 +236,7 @@ func TestGetChkpt(t *testing.T) {
 				if err != nil {
 					t.Errorf("failed to get latest: %v", err)
 				}
-				wV, err := note.NewVerifier(wPK)
+				wV, err := f_note.NewVerifierForCosignatureV1(wPK)
 				if err != nil {
 					t.Fatalf("couldn't create a witness verifier: %v", err)
 				}