Function TestAuditService

lib/syncthing/auditservice_test.go:19–70

(t *testing.T)

)

func TestAuditService(t *testing.T) {
	buf := new(bytes.Buffer)
	evLogger := events.NewLogger()
	ctx, cancel := context.WithCancel(context.Background())
	go evLogger.Serve(ctx)
	defer cancel()
	sub := evLogger.Subscribe(events.AllEvents)
	defer sub.Unsubscribe()

	// Event sent before start, will not be logged
	evLogger.Log(events.ConfigSaved, "the first event")
	// Make sure the event goes through before creating the service
	<-sub.C()

	auditCtx, auditCancel := context.WithCancel(context.Background())
	service := newAuditService(buf, evLogger)
	done := make(chan struct{})
	go func() {
		service.Serve(auditCtx)
		close(done)
	}()

	// Subscription needs to happen in service.Serve
	time.Sleep(10 * time.Millisecond)

	// Event that should end up in the audit log
	evLogger.Log(events.ConfigSaved, "the second event")

	// We need to give the events time to arrive, since the channels are buffered etc.
	time.Sleep(10 * time.Millisecond)

	auditCancel()
	<-done

	// This event should not be logged, since we have stopped.
	evLogger.Log(events.ConfigSaved, "the third event")

	result := buf.String()
	t.Log(result)

	if strings.Contains(result, "first event") {
		t.Error("Unexpected first event")
	}

	if !strings.Contains(result, "second event") {
		t.Error("Missing second event")
	}

	if strings.Contains(result, "third event") {
		t.Error("Missing third event")
	}
}