service_test.go

// Copyright 2015 Daniel Theophanes.
// Use of this source code is governed by a zlib-style
// license that can be found in the LICENSE file.

package baobab

import (
	"context"
	"sync"
	"testing"
	"time"
)

func TestRunInterrupt(t *testing.T) {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	p := &program{}
	sc := &Config{
		Name: "go_baobab_test",
	}
	s, err := New(p, sc)
	if err != nil {
		t.Fatalf("New err: %s", err)
	}

	go func() {
		<-time.After(1 * time.Second)
		interruptProcess(t)
	}()

	stopped := make(chan struct{})
	go func() {
		timer := time.NewTimer(200 * time.Millisecond)
		defer timer.Stop()
		for {
			select {
			case <-timer.C:
				p.mu.RLock()
				if p.numStopped != 0 {
					p.mu.RUnlock()
					close(stopped)
					return
				}
				p.mu.RUnlock()
				timer.Reset(200 * time.Millisecond)
			case <-ctx.Done():
				return
			}
		}
	}()

	if err = s.Run(); err != nil {
		t.Fatalf("Run() err: %s", err)
	}

	select {
	case <-ctx.Done():
		t.Fatalf("Run() hasn't been stopped")
	case <-stopped:
		// Success case - the program stopped.
	}
}

type program struct {
	numStopped int
	mu         sync.RWMutex
}

func (p *program) Start(s Service) error {
	go p.run()
	return nil
}
func (p *program) run() {
	// Do work here
}
func (p *program) Stop(s Service) error {
	p.mu.Lock()
	defer p.mu.Unlock()
	p.numStopped++
	return nil
}