From 6de645a8b80463b2100bd997d8537093c51e54b4 Mon Sep 17 00:00:00 2001
From: Frank Schroeder <frank.schroeder@gmail.com>
Date: Thu, 19 Oct 2017 11:12:29 +0200
Subject: [PATCH] ae: add test for ifNotPausedRun

---
 agent/ae/ae_test.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/agent/ae/ae_test.go b/agent/ae/ae_test.go
index 3d3449d8b..500435359 100644
--- a/agent/ae/ae_test.go
+++ b/agent/ae/ae_test.go
@@ -1,7 +1,9 @@
 package ae
 
 import (
+	"errors"
 	"fmt"
+	"reflect"
 	"testing"
 )
 
@@ -56,3 +58,22 @@ func TestAE_nestedPauseResume(t *testing.T) {
 	}()
 	l.Resume()
 }
+
+func TestAE_ifNotPausedRun(t *testing.T) {
+	l := NewStateSyner(nil, 0, nil, nil)
+
+	errCalled := errors.New("f called")
+	f := func() error { return errCalled }
+
+	l.Pause()
+	err := l.ifNotPausedRun(f)
+	if got, want := err, errPaused; !reflect.DeepEqual(got, want) {
+		t.Fatalf("got error %q want %q", got, want)
+	}
+	l.Resume()
+
+	err = l.ifNotPausedRun(f)
+	if got, want := err, errCalled; got != want {
+		t.Fatalf("got error %q want %q", got, want)
+	}
+}