logging_initializer_test.go

package cproxy

import (
	"fmt"
	"testing"

	"github.com/smartystreets/assertions/should"
	"github.com/smartystreets/gunit"
)

func TestLoggingInitializerFixture(t *testing.T) {
	gunit.Run(new(LoggingInitializerFixture), t)
}

type LoggingInitializerFixture struct {
	*gunit.Fixture

	client      *TestSocket
	server      *TestSocket
	fakeInner   *TestInitializer
	initializer initializer
	logs        []string
}

func (this *LoggingInitializerFixture) Setup() {
	this.client = NewTestSocket()
	this.server = NewTestSocket()
	this.client.address = "1.2.3.4"
	this.client.port = 4321
	this.server.address = "5.6.7.8"
	this.server.port = 8765

	this.fakeInner = NewTestInitializer(true)
	this.initializer = newLoggingInitializer(&configuration{Logger: this, Initializer: this.fakeInner, LogConnections: true})
}

func (this *LoggingInitializerFixture) TestInnerInitializerCalled() {
	result := this.initializer.Initialize(this.client, this.server)

	this.So(result, should.BeTrue)
	this.So(this.fakeInner.client, should.Equal, this.client)
	this.So(this.fakeInner.server, should.Equal, this.server)
}

func (this *LoggingInitializerFixture) TestLoggingOnFailure() {
	this.fakeInner.success = false

	this.initializer.Initialize(this.client, this.server)

	this.So(this.logs, should.Resemble, []string{"[INFO] Connection failed [1.2.3.4:4321] -> [5.6.7.8:8765]"})
}

func (this *LoggingInitializerFixture) TestLoggingOnSuccess() {
	this.initializer.Initialize(this.client, this.server)

	this.So(this.logs, should.Resemble, []string{"[INFO] Established connection [1.2.3.4:4321] -> [5.6.7.8:8765]"})
}

func (this *LoggingInitializerFixture) Printf(format string, args ...interface{}) {
	this.logs = append(this.logs, fmt.Sprintf(format, args...))
}