Skip to content

Commit a0cf54d

Browse files
committed
Fix tests
1 parent 4483a59 commit a0cf54d

File tree

3 files changed

+12
-6
lines changed

3 files changed

+12
-6
lines changed

internal/internal_schedule_client_test.go

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,11 @@ func (s *scheduleClientTestSuite) TestCreateScheduleWorkflowMemoDataConverter()
250250
Do(func(_ interface{}, req *workflowservice.CreateScheduleRequest, _ ...interface{}) {
251251
startWorkflow := req.Schedule.Action.GetStartWorkflow()
252252
encoding := string(startWorkflow.Memo.Fields["testMemo"].Metadata[converter.MetadataEncoding])
253-
s.Equal("binary/gob", encoding)
253+
if memoUserDCEncode {
254+
s.Equal("binary/gob", encoding)
255+
} else {
256+
s.Equal("json/plain", encoding)
257+
}
254258
})
255259

256260
_, err := s.client.ScheduleClient().Create(context.Background(), options)
@@ -302,7 +306,11 @@ func (s *scheduleClientTestSuite) TestCreateScheduleWorkflowMemoUserAndDefaultCo
302306

303307
_, err := s.client.ScheduleClient().Create(context.Background(), options)
304308
s.Error(err)
305-
s.ErrorContains(err, "failingMemoDataConverter memo encoding failed")
309+
if memoUserDCEncode {
310+
s.ErrorContains(err, "failingMemoDataConverter memo encoding failed")
311+
} else {
312+
s.ErrorContains(err, "unsupported type: chan int")
313+
}
306314
}
307315

308316
s.T().Run("old behavior", func(t *testing.T) {

internal/internal_workflow_testsuite.go

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2857,9 +2857,6 @@ func (env *testWorkflowEnvironmentImpl) UpsertTypedSearchAttributes(attributes S
28572857

28582858
func (env *testWorkflowEnvironmentImpl) UpsertMemo(memoMap map[string]interface{}) error {
28592859
memo, err := validateAndSerializeMemo(memoMap, env.dataConverter, env)
2860-
if err != nil {
2861-
return err
2862-
}
28632860

28642861
env.workflowInfo.Memo = mergeMemo(env.workflowInfo.Memo, memo)
28652862

internal/internal_workflow_testsuite_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1877,7 +1877,8 @@ func (s *WorkflowTestSuiteUnitTest) Test_MockUpsertMemo() {
18771877
s.NotNil(wfInfo.Memo)
18781878
valBytes := wfInfo.Memo.Fields["CustomIntField"]
18791879
var result int
1880-
_ = converter.GetDefaultDataConverter().FromPayload(valBytes, &result)
1880+
err = converter.GetDefaultDataConverter().FromPayload(valBytes, &result)
1881+
s.NoError(err)
18811882
s.Equal(1, result)
18821883

18831884
return nil

0 commit comments

Comments
 (0)