Add tests for activity struct arg (#134)

Author: alexshtin

internal/internal_worker_test.go

@@ -62,6 +62,7 @@ func testInternalWorkerRegister(r *registry) {
 		testActivityMultipleArgs,
 		RegisterActivityOptions{Name: "testActivityMultipleArgs"},
 	)
+	r.RegisterActivity(testActivityMultipleArgsWithStruct)
 	r.RegisterActivity(testActivityReturnString)
 	r.RegisterActivity(testActivityReturnEmptyString)
 	r.RegisterActivity(testActivityReturnEmptyStruct)
@@ -94,6 +95,7 @@ func testInternalWorkerRegisterWithTestEnv(env *TestWorkflowEnvironment) {
 		testActivityMultipleArgs,
 		RegisterActivityOptions{Name: "testActivityMultipleArgs"},
 	)
+	env.RegisterActivity(testActivityMultipleArgsWithStruct)
 	env.RegisterActivity(testActivityReturnString)
 	env.RegisterActivity(testActivityReturnEmptyString)
 	env.RegisterActivity(testActivityReturnEmptyStruct)
@@ -413,6 +415,7 @@ func (s *internalWorkerTestSuite) TestDecisionTaskHandler_WithDataConverter() {
 func sampleWorkflowExecute(ctx Context, input []byte) (result []byte, err error) {
 	ExecuteActivity(ctx, testActivityByteArgs, input)
 	ExecuteActivity(ctx, testActivityMultipleArgs, 2, []string{"test"}, true)
+	ExecuteActivity(ctx, testActivityMultipleArgsWithStruct, -8, newTestActivityArg())
 	return []byte("Done"), nil
 }
 
@@ -428,6 +431,12 @@ func testActivityMultipleArgs(context.Context, int, []string, bool) ([]byte, err
 	return nil, nil
 }
 
+// test testActivityMultipleArgsWithStruct
+func testActivityMultipleArgsWithStruct(_ context.Context, i int, s testActivityArg) ([]byte, error) {
+	fmt.Printf("Executing testActivityMultipleArgsWithStruct: %d, %v\n", i, s)
+	return nil, nil
+}
+
 func (s *internalWorkerTestSuite) TestCreateWorker() {
 	worker := createWorkerWithThrottle(s.service, float64(500.0), nil)
 	err := worker.Start()
@@ -736,6 +745,9 @@ func (w activitiesCallingOptionsWorkflow) Execute(ctx Context, input []byte) (re
 	err = ExecuteActivity(ctx, testActivityMultipleArgs, 2, []string{"test"}, true).Get(ctx, nil)
 	require.NoError(w.t, err, err)
 
+	err = ExecuteActivity(ctx, testActivityMultipleArgsWithStruct, -8, newTestActivityArg()).Get(ctx, nil)
+	require.NoError(w.t, err, err)
+
 	err = ExecuteActivity(ctx, testActivityNoResult, 2, "test").Get(ctx, nil)
 	require.NoError(w.t, err, err)
 
@@ -859,10 +871,34 @@ func testActivityReturnEmptyString() (string, error) {
 	return "", nil
 }
 
+type testActivityArg struct {
+	Index    int
+	Name     string
+	Data     []byte
+	IndexPtr *int
+	NamePtr  *string
+	DataPtr  *[]byte
+}
+
 type testActivityResult struct {
 	Index int
 }
 
+func newTestActivityArg() *testActivityArg {
+	name := "JohnSmith"
+	index := 22
+	data := []byte{22, 8, 78}
+
+	return &testActivityArg{
+		Name:     name,
+		Index:    index,
+		Data:     data,
+		NamePtr:  &name,
+		IndexPtr: &index,
+		DataPtr:  &data,
+	}
+}
+
 // testActivityReturnEmptyStruct
 func testActivityReturnEmptyStruct() (testActivityResult, error) {
 	// Return is mocked to retrun nil from server.