yanic/database/socket/database_test.go

package socket

import (
	"net"
	"testing"
	"time"

	"github.com/FreifunkBremen/yanic/runtime"
	"github.com/stretchr/testify/assert"
)

func TestStartup(t *testing.T) {
	assert := assert.New(t)

	config := make(map[string]interface{})

	config["enable"] = false
	conn, err := Connect(config)
	assert.Nil(conn)

	config["enable"] = true
	config["type"] = ""
	config["address"] = ""
	conn, err = Connect(config)
	assert.Error(err, "connection should not work")
	assert.Nil(conn)

	config["type"] = "unix"
	config["address"] = "/tmp/yanic-database.socket"

	conn, err = Connect(config)
	assert.NoError(err, "connection should work")
	assert.NotNil(conn)

	conn.Close()
}

func TestClient(t *testing.T) {
	assert := assert.New(t)

	config := make(map[string]interface{})

	config["enable"] = true
	config["type"] = "unix"
	config["address"] = "/tmp/yanic-database2.socket"

	conn, err := Connect(config)
	assert.NoError(err, "connection should work")
	assert.NotNil(conn)

	client, err := net.Dial("unix", "/tmp/yanic-database2.socket")
	assert.NoError(err, "connection should work")
	assert.NotNil(client)
	time.Sleep(time.Duration(3) * time.Microsecond)

	conn.InsertNode(&runtime.Node{})
	conn.InsertGlobals(&runtime.GlobalStats{}, time.Now())
	time.Sleep(time.Duration(3) * time.Microsecond)

	// to reach in sendJSON removing of disconnection
	err = client.Close()
	assert.NoError(err, "disconnect should work")
	time.Sleep(time.Duration(3) * time.Microsecond)
	conn.InsertNode(&runtime.Node{})
	time.Sleep(time.Duration(3) * time.Microsecond)

	// to reach all parts of conn.Close()
	client, err = net.Dial("unix", "/tmp/yanic-database2.socket")
	time.Sleep(time.Duration(3) * time.Microsecond)

	conn.Close()
}