// Copyright 2012-2013 Apcera Inc. All rights reserved.

package test

import (
	"runtime"
	"testing"
	"time"
)

func TestSimpleGoServerShutdown(t *testing.T) {
	base := runtime.NumGoroutine()
	s := RunDefaultServer()
	s.Shutdown()
	time.Sleep(10 * time.Millisecond)
	delta := (runtime.NumGoroutine() - base)
	if delta > 1 {
		t.Fatalf("%d Go routines still exist post Shutdown()", delta)
	}
}

func TestGoServerShutdownWithClients(t *testing.T) {
	base := runtime.NumGoroutine()
	s := RunDefaultServer()
	for i := 0; i < 50; i++ {
		createClientConn(t, "localhost", 4222)
	}
	s.Shutdown()
	// Wait longer for client connections
	time.Sleep(500 * time.Millisecond)
	delta := (runtime.NumGoroutine() - base)
	// There may be some finalizers or IO, but in general more than
	// 2 as a delta represents a problem.
	if delta > 2 {
		t.Fatalf("%d Go routines still exist post Shutdown()", delta)
	}
}

func TestGoServerMultiShutdown(t *testing.T) {
	s := RunDefaultServer()
	s.Shutdown()
	s.Shutdown()
}