lobby/server/identification_test.go

package server

import (
	"os"
	"testing"

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

const tmpPath = "./tmp"
const testLabelPath = tmpPath + "/labels"

func TestGetIdentification(t *testing.T) {
	localHost := LocalHost{
		LabelsPath:       testLabelPath,
		HostnameOverride: "test.example.com",
		InitialLabels:    Labels{Label("service:test"), Label("test:1")},
	}

	discovery, err := localHost.GetIdentification()
	assert.Nil(t, err)

	assert.Equal(t, "test.example.com", discovery.Hostname)
	assert.Equal(t, "service:test", discovery.Labels[0].String())

	err = os.MkdirAll(testLabelPath, os.ModePerm)
	assert.Nil(t, err)

	err = os.WriteFile(testLabelPath+"/test", []byte("service:test2\npublic_ip:1.2.3.4"), 0644)
	assert.Nil(t, err)

	discovery, err = localHost.GetIdentification()
	assert.Nil(t, err)

	assert.Equal(t, Label("public_ip:1.2.3.4"), discovery.Labels[3])

	os.RemoveAll(tmpPath)
}

func TestLoadLocalLabels(t *testing.T) {
	localHost := LocalHost{
		LabelsPath:       testLabelPath,
		HostnameOverride: "test.example.com",
		InitialLabels:    Labels{Label("service:test"), Label("test:1")},
	}

	err := os.MkdirAll(testLabelPath, os.ModePerm)
	assert.Nil(t, err)

	err = os.WriteFile(testLabelPath+"/test", []byte("service:test\npublic_ip:1.2.3.4"), 0644)
	assert.Nil(t, err)

	labels, err := localHost.loadLocalLabels()
	assert.Nil(t, err)

	assert.Equal(t, 1, len(labels))
	assert.Equal(t, "public_ip:1.2.3.4", labels[0].String())

	os.RemoveAll(tmpPath)
}