worker/terminationworker: fix test hang

worker/terminationworker/worker.go

@@ -30,30 +30,30 @@ type terminationWorker struct {
 // TerminationSignal signal, and then exits
 // with worker.ErrTerminateAgent.
 func NewWorker() worker.Worker {
-	u := &terminationWorker{}
+	var w terminationWorker
+	c := make(chan os.Signal, 1)
+	signal.Notify(c, TerminationSignal)
 	go func() {
-		defer u.tomb.Done()
-		u.tomb.Kill(u.loop())
+		defer w.tomb.Done()
+		defer signal.Stop(c)
+		w.tomb.Kill(w.loop(c))
 	}()
-	return u
+	return &w
 }
 
-func (u *terminationWorker) Kill() {
-	u.tomb.Kill(nil)
+func (w *terminationWorker) Kill() {
+	w.tomb.Kill(nil)
 }
 
-func (u *terminationWorker) Wait() error {
-	return u.tomb.Wait()
+func (w *terminationWorker) Wait() error {
+	return w.tomb.Wait()
 }
 
-func (u *terminationWorker) loop() (err error) {
-	c := make(chan os.Signal, 1)
-	signal.Notify(c, TerminationSignal)
-	defer signal.Stop(c)
+func (w *terminationWorker) loop(c <-chan os.Signal) (err error) {
 	select {
 	case <-c:
 		return worker.ErrTerminateAgent
-	case <-u.tomb.Dying():
+	case <-w.tomb.Dying():
 		return tomb.ErrDying
 	}
 }

worker/terminationworker/worker_test.go

@@ -5,43 +5,28 @@ package terminationworker_test
 
 import (
 	"os"
-	"os/signal"
 	"runtime"
-	stdtesting "testing"
+	"testing"
 
 	jc "github.com/juju/testing/checkers"
 	gc "gopkg.in/check.v1"
 
-	"github.com/juju/juju/testing"
 	"github.com/juju/juju/worker"
 	"github.com/juju/juju/worker/terminationworker"
 )
 
-func TestPackage(t *stdtesting.T) {
+func TestPackage(t *testing.T) {
 	gc.TestingT(t)
 }
 
 var _ = gc.Suite(&TerminationWorkerSuite{})
 
 type TerminationWorkerSuite struct {
-	testing.BaseSuite
 	// c is a channel that will wait for the termination
 	// signal, to prevent signals terminating the process.
 	c chan os.Signal
 }
 
-func (s *TerminationWorkerSuite) SetUpTest(c *gc.C) {
-	s.BaseSuite.SetUpTest(c)
-	s.c = make(chan os.Signal, 1)
-	signal.Notify(s.c, terminationworker.TerminationSignal)
-}
-
-func (s *TerminationWorkerSuite) TearDownTest(c *gc.C) {
-	signal.Stop(s.c)
-	close(s.c)
-	s.BaseSuite.TearDownTest(c)
-}
-
 func (s *TerminationWorkerSuite) TestStartStop(c *gc.C) {
 	w := terminationworker.NewWorker()
 	w.Kill()