feat: Provide Signal method (#40)

GuanceCloud · Aug 5, 2023 · 470af95 · 470af95
1 parent c610f88
commit 470af95
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/pkg/engine/runtime/context.go b/pkg/engine/runtime/context.go
@@ -81,6 +81,10 @@ func (ctx *Context) InData() any {
 	}
 }
 
+func (ctx *Context) Signal() Signal {
+	return ctx.signal
+}
+
 func InitCtxWithoutMap(ctx *Context, inWithoutMap InputWithoutMap, funcs map[string]FuncCall,
 	callRef []*ast.CallExpr, signal Signal, name, content string,
 ) *Context {

diff --git a/pkg/engine/runtime/runtime_test.go b/pkg/engine/runtime/runtime_test.go
@@ -347,6 +347,26 @@ func TestInExpr(t *testing.T) {
 	}, inData.data)
 }
 
+type sign struct {
+}
+
+func (s *sign) ExitSignal() bool {
+	return false
+}
+
+func TestSignal(t *testing.T) {
+	s := func() Signal {
+		return (*sign)(nil)
+	}()
+
+	ctx := &Context{
+		signal: s,
+	}
+
+	sig := ctx.Signal()
+	assert.Equal(t, false, sig == nil)
+}
+
 func TestUnaryAndAssignOP(t *testing.T) {
 	cases := []struct {
 		name string