tcplistener_test.go

package buffstreams

import (
	"strconv"
	"testing"
)

func TestListenTCPUsesDefaultMessageSize(t *testing.T) {
	cfg := TCPListenerConfig{
		Address:  FormatAddress("", strconv.Itoa(5031)),
		Callback: func([]byte) error { return nil },
	}
	buffM, err := ListenTCP(cfg)
	if err != nil {
		t.Errorf("Could not Listen on port %d: %s", cfg.MaxMessageSize, err.Error())
	}
	if buffM.connConfig.MaxMessageSize != DefaultMaxMessageSize {
		t.Errorf("Expected Max Message Size to be %d, actually got %d", DefaultMaxMessageSize, buffM.connConfig.MaxMessageSize)
	}
}

func TestListenTCPUsesSpecifiedMaxMessageSize(t *testing.T) {
	cfg := TCPListenerConfig{
		MaxMessageSize: 8196,
		Address:        FormatAddress("", strconv.Itoa(5032)),
		Callback:       func([]byte) error { return nil },
	}
	buffM, err := ListenTCP(cfg)
	if err != nil {
		t.Errorf("Could not Listen on port %d: %s", cfg.MaxMessageSize, err.Error())
	}
	if buffM.connConfig.MaxMessageSize != cfg.MaxMessageSize {
		t.Errorf("Expected Max Message Size to be %d, actually got %d", cfg.MaxMessageSize, buffM.connConfig.MaxMessageSize)
	}
}