lobby/server/discovery_test.go

package server

import (
	"encoding/json"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestDiscovery(t *testing.T) {
	now := time.Now().Unix()
	now90 := now - 90

	discovery := Discovery{
		Hostname: "test.rosti.cz",
		Labels: Labels{
			Label("service:test"),
			Label("test:123"),
			Label("public_ip:1.2.3.4"),
		},
		LastCheck: now,
	}

	assert.True(t, discovery.IsAlive(), "discovery suppose to be alive")
	discovery.LastCheck = now90
	assert.False(t, discovery.IsAlive(), "discovery not suppose to be alive")
	discovery.LastCheck = now

	assert.Equal(t, Labels{Label("service:test")}, discovery.FindLabelsByPrefix("service"))
	assert.Equal(t, nil, discovery.Validate()) // TODO: This needs more love

	content, err := json.Marshal(&discovery)
	assert.Nil(t, err)
	content2, err := discovery.Bytes()
	assert.Nil(t, err)
	assert.Equal(t, content, content2)
}