1 files changed, 24 insertions, 20 deletions
diff --git a/internal/vote/vote_test.go b/internal/vote/vote_test.go
index 8bc6647..04f0623 100644
--- a/internal/vote/vote_test.go
+++ b/internal/vote/vote_test.go
@@ -1,40 +1,44 @@
 package vote
 
 import (
+	"os"
 	"testing"
 	"time"
-
-	"codeberg.org/snonux/gorum/internal/config"
 )
 
 func TestVote(t *testing.T) {
-	var (
-		conf = config.Config{Nodes: []string{"earth:1234", "foo:1234", "bay:4321"}}
-		v    = New(conf, "earth foo bar baz bay\n")
-	)
+	v, err := New([]string{"foo", "bar", "baz", "bay"})
+	if err != nil {
+		t.Errorf(err.Error())
+	}
+
+	hostname, err := os.Hostname()
+	if err != nil {
+		t.Errorf(err.Error())
+	}
 
-	if v.FromID != "earth" {
+	if v.FromID != hostname {
 		t.Errorf("Expected vote to come from earth but came from %s", v.FromID)
 	}
 
-	if len(v.IDs) != 2 {
-		t.Errorf("Expected vote length to be 2 but is %d", len(v.IDs))
+	if len(v.IDs) != 4 {
+		t.Errorf("Expected vote length to be 4 but is %d", len(v.IDs))
 	}
 
 	if v.IDs[0] != "foo" {
 		t.Errorf("Expected vote 1 to be foo but is %s", v.IDs[0])
 	}
 
-	if v.IDs[1] != "bay" {
-		t.Errorf("Expected vote 2 to be bay but is %s", v.IDs[1])
+	if v.IDs[1] != "bar" {
+		t.Errorf("Expected vote 2 to be bar but is %s", v.IDs[1])
 	}
 }
 
 func TestVoteExpiry(t *testing.T) {
-	var (
-		conf = config.Config{Nodes: []string{"foo:1234", "bay:4321"}}
-		v    = New(conf, "earth foo bar baz bay\n")
-	)
+	v, err := New([]string{"foo", "bar", "baz", "bay"})
+	if err != nil {
+		t.Errorf(err.Error())
+	}
 
 	// Set expiry 1h into the future
 	v.ExpiresAt = time.Now().Add(1 * time.Hour)
@@ -50,10 +54,10 @@ func TestVoteExpiry(t *testing.T) {
 }
 
 func TestMarshalling(t *testing.T) {
-	var (
-		conf = config.Config{Nodes: []string{"foo:1234", "bay:4321"}}
-		v    = New(conf, "earth foo bar baz bay\n")
-	)
+	v, err := New([]string{"foo", "bar", "baz", "bay"})
+	if err != nil {
+		t.Errorf(err.Error())
+	}
 
 	bytes, err := v.ToJSON()
 	if err != nil {
@@ -65,7 +69,7 @@ func TestMarshalling(t *testing.T) {
 		t.Errorf("unable to deserialize json to vote: %v", err)
 	}
 
-	if !v.equals(v2) {
+	if !v.Equal(v2) {
 		t.Errorf("serialized %v and deserialized %v votes differ", v, v2)
 	}
 }