@@ -974,6 +974,29 @@ func TestActivityMockingByNameWithoutRegistrationFails(t *testing.T) {
974974 assert .Panics (t , func () { env .OnActivity ("SayHello" , mock .Anything , mock .Anything ) }, "The code did not panic" )
975975}
976976
977+ func TestMockCallWrapperInOrder (t * testing.T ) {
978+ testSuite := & WorkflowTestSuite {}
979+ env := testSuite .NewTestWorkflowEnvironment ()
980+ env .RegisterActivity (namedActivity )
981+
982+ env .InOrderMockCalls (
983+ env .OnActivity (namedActivity , mock .Anything , "call1" ).Return ("result1" , nil ),
984+ env .OnActivity (namedActivity , mock .Anything , "call2" ).Return ("result2" , nil ),
985+ )
986+
987+ env .ExecuteWorkflow (func (ctx Context ) error {
988+ ctx = WithLocalActivityOptions (ctx , LocalActivityOptions {
989+ ScheduleToCloseTimeout : time .Hour ,
990+ StartToCloseTimeout : time .Hour ,
991+ })
992+ var result string
993+ return ExecuteLocalActivity (ctx , "namedActivity" , "call2" ).Get (ctx , & result )
994+ })
995+ var expectedErr * PanicError
996+ require .ErrorAs (t , env .GetWorkflowError (), & expectedErr )
997+ require .ErrorContains (t , expectedErr , "Must not be called before" )
998+ }
999+
9771000func TestMockCallWrapperNotBefore (t * testing.T ) {
9781001 testSuite := & WorkflowTestSuite {}
9791002 env := testSuite .NewTestWorkflowEnvironment ()
0 commit comments