Browse Source

test: add test for InitializeActionHandler

Signed-off-by: Mark Sagi-Kazar <mark.sagikazar@gmail.com>
Mark Sagi-Kazar 4 years ago
parent
commit
56bf8364cd
1 changed files with 24 additions and 0 deletions
  1. 24 0
      common/actions_test.go

+ 24 - 0
common/actions_test.go

@@ -179,3 +179,27 @@ func TestPreDeleteAction(t *testing.T) {
 
 
 	Config.Actions = actionsCopy
 	Config.Actions = actionsCopy
 }
 }
+
+type actionHandlerStub struct {
+	called bool
+}
+
+func (h *actionHandlerStub) Handle(notification ActionNotification) error {
+	h.called = true
+
+	return nil
+}
+
+func TestInitializeActionHandler(t *testing.T) {
+	handler := &actionHandlerStub{}
+
+	InitializeActionHandler(handler)
+	t.Cleanup(func() {
+		InitializeActionHandler(defaultActionHandler{})
+	})
+
+	err := actionHandler.Handle(ActionNotification{})
+
+	assert.NoError(t, err)
+	assert.True(t, handler.called)
+}