@@ -3,11 +3,11 @@ package runner
33import (
44 "context"
55 "errors"
6- "github.com/project-flogo/core/data/metadata"
76 "testing"
87
8+ "github.com/project-flogo/core/data/metadata"
9+
910 "github.com/project-flogo/core/action"
10- "github.com/project-flogo/core/data"
1111 "github.com/stretchr/testify/assert"
1212 "github.com/stretchr/testify/mock"
1313)
@@ -28,36 +28,33 @@ func (m *MockAsyncAction) Metadata() *action.Metadata {
2828 return nil
2929}
3030
31- func (m * MockAsyncAction ) Run (context context.Context , inputs map [string ]* data. Attribute , handler action.ResultHandler ) error {
31+ func (m * MockAsyncAction ) Run (context context.Context , inputs map [string ]interface {} , handler action.ResultHandler ) error {
3232 args := m .Called (context , inputs , handler )
33+
3334 if handler != nil {
34- dataAttr , _ := data .NewAttribute ("data" , data .TypeString , "mock" )
35- codeAttr , _ := data .NewAttribute ("code" , data .TypeInteger , 200 )
36- resultData := map [string ]* data.Attribute {
37- "data" : dataAttr ,
38- "code" : codeAttr ,
39- }
35+ resultData := make (map [string ]interface {})
36+ resultData ["data" ] = "mock"
37+ resultData ["code" ] = 200
4038 handler .HandleResult (resultData , nil )
4139 handler .Done ()
4240 }
41+
4342 return args .Error (0 )
4443}
4544
4645//Test that Result returns the expected values
4746func TestResultOk (t * testing.T ) {
4847
4948 //mockData,_ :=data.CoerceToObject("{\"data\":\"mock data \"}")
50- dataAttr , _ := data .NewAttribute ("data" , data .TypeString , "mock data" )
51- codeAttr , _ := data .NewAttribute ("code" , data .TypeInteger , 1 )
52- resultData := map [string ]* data.Attribute {
53- "data" : dataAttr ,
54- "code" : codeAttr ,
55- }
49+
50+ resultData := make (map [string ]interface {})
51+ resultData ["data" ] = "mock data"
52+ resultData ["code" ] = 1
5653
5754 rh := & SyncResultHandler {resultData : resultData , err : errors .New ("New Error" )}
5855 data , err := rh .Result ()
59- assert .Equal (t , 1 , data ["code" ]. Value () )
60- assert .Equal (t , "mock data" , data ["data" ]. Value () )
56+ assert .Equal (t , 1 , data ["code" ])
57+ assert .Equal (t , "mock data" , data ["data" ])
6158 assert .NotNil (t , err )
6259}
6360
@@ -81,7 +78,7 @@ func TestDirectStopOk(t *testing.T) {
8178func TestDirectRunNilAction (t * testing.T ) {
8279 runner := NewDirect ()
8380 assert .NotNil (t , runner )
84- _ , err := runner .Execute (nil , nil , nil )
81+ _ , err := runner .RunAction (nil , nil , nil )
8582 assert .NotNil (t , err )
8683}
8784
@@ -91,8 +88,9 @@ func TestDirectRunErr(t *testing.T) {
9188 assert .NotNil (t , runner )
9289 // Mock Action
9390 mockAction := new (MockAsyncAction )
94- mockAction .On ("Run" , nil , mock .AnythingOfType ("map[string]*data.Attribute" ), mock .AnythingOfType ("*runner.SyncResultHandler" )).Return (errors .New ("Action Error" ))
95- _ , err := runner .Execute (nil , mockAction , nil )
91+
92+ mockAction .On ("Run" , nil , mock .AnythingOfType ("map[string]interface {}" ), mock .AnythingOfType ("*runner.SyncResultHandler" )).Return (errors .New ("Action Error" ))
93+ _ , err := runner .RunAction (nil , mockAction , nil )
9694 assert .NotNil (t , err )
9795}
9896
@@ -103,23 +101,23 @@ func TestDirectRunOk(t *testing.T) {
103101 // Mock Action
104102 mockAction := new (MockAsyncAction )
105103
106- mockAction .On ("Run" , nil , mock .AnythingOfType ("map[string]*data.Attribute " ), mock .AnythingOfType ("*runner.SyncResultHandler" )).Return (nil )
107- results , err := runner .Execute ( nil , mockAction , nil )
104+ mockAction .On ("Run" , context . Background () , mock .AnythingOfType ("map[string]interface {} " ), mock .AnythingOfType ("*runner.SyncResultHandler" )).Return (nil )
105+ results , err := runner .RunAction ( context . Background () , mockAction , nil )
108106 assert .Nil (t , err )
109107 assert .NotNil (t , results )
110108 code , ok := results ["code" ]
111109 assert .True (t , ok )
112110 data , ok := results ["data" ]
113111 assert .True (t , ok )
114- assert .Equal (t , 200 , code . Value () )
115- assert .Equal (t , "mock" , data . Value () )
112+ assert .Equal (t , 200 , code )
113+ assert .Equal (t , "mock" , data )
116114}
117115
118116//Test Run method with a nil action
119117func TestDirectRunNilActionOld (t * testing.T ) {
120118 runner := NewDirect ()
121119 assert .NotNil (t , runner )
122- _ , err := runner .Execute ( nil , nil , nil )
120+ _ , err := runner .RunAction ( context . Background () , nil , nil )
123121 assert .NotNil (t , err )
124122}
125123
@@ -129,8 +127,8 @@ func TestDirectRunErrOld(t *testing.T) {
129127 assert .NotNil (t , runner )
130128 // Mock Action
131129 mockAction := new (MockAsyncAction )
132- mockAction .On ("Run" , nil , mock .AnythingOfType ("map[string]*data.Attribute " ), mock .AnythingOfType ("*runner.SyncResultHandler" )).Return (errors .New ("Action Error" ))
133- _ , err := runner .Execute ( nil , mockAction , nil )
130+ mockAction .On ("Run" , context . Background () , mock .AnythingOfType ("map[string]interface {} " ), mock .AnythingOfType ("*runner.SyncResultHandler" )).Return (errors .New ("Action Error" ))
131+ _ , err := runner .RunAction ( context . Background () , mockAction , nil )
134132 assert .NotNil (t , err )
135133}
136134
0 commit comments