package vote

import (
	"os"
	"testing"
	"time"
)

func TestVote(t *testing.T) {
	v, _ := New([]string{"foo", "bar", "baz", "bay"})
	hostname, _ := os.Hostname()

	if v.FromID != hostname {
		t.Errorf("Expected vote to come from earth but came from %s", v.FromID)
	}

	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] != "bar" {
		t.Errorf("Expected vote 2 to be bar but is %s", v.IDs[1])
	}
}

func TestVoteExpiry(t *testing.T) {
	v, _ := New([]string{"foo", "bar", "baz", "bay"})

	// Set expiry 1h into the future
	v.ExpiresAt = time.Now().Add(1 * time.Hour)
	if v.Expired() {
		t.Errorf("Didn't expect vote to be expired")
	}

	// Set expiry to now
	v.ExpiresAt = time.Now()
	if !v.Expired() {
		t.Errorf("Expected vote to be expired")
	}
}

func TestMarshalling(t *testing.T) {
	v, _ := New([]string{"foo", "bar", "baz", "bay"})
	bytes, err := v.ToJSON()
	if err != nil {
		t.Errorf("unable to serialize vote to json: %v", err)
	}

	v2, err := NewFromJSON(bytes)
	if err != nil {
		t.Errorf("unable to deserialize json to vote: %v", err)
	}

	if !v.Equal(v2) {
		t.Errorf("serialized %v and deserialized %v votes differ", v, v2)
	}
}