@@ -44,7 +44,6 @@ import (
4444 "go.uber.org/cadence/client"
4545 "go.uber.org/cadence/interceptors"
4646 "go.uber.org/cadence/internal"
47- "go.uber.org/cadence/test/replaytests"
4847 "go.uber.org/cadence/worker"
4948 "go.uber.org/cadence/workflow"
5049)
@@ -561,15 +560,15 @@ func (ts *IntegrationTestSuite) TestOverrideSpanContext() {
561560// but not on VersionedWorkflowV4, VersionedWorkflowV5, VersionedWorkflowV6.
562561func (ts * IntegrationTestSuite ) TestVersionedWorkflowV1 () {
563562 ts .testVersionedWorkflow (testVersionedWorkflowTestCase {
564- version : replaytests . VersionWorkflowVersionV1 ,
565- compatibleVersions : []replaytests. VersionWorkflowVersion {
566- replaytests . VersionWorkflowVersionV2 ,
567- replaytests . VersionWorkflowVersionV3 ,
563+ version : VersionedWorkflowVersionV1 ,
564+ compatibleVersions : []VersionedWorkflowVersion {
565+ VersionedWorkflowVersionV2 ,
566+ VersionedWorkflowVersionV3 ,
568567 },
569- inCompatibleVersions : []replaytests. VersionWorkflowVersion {
570- replaytests . VersionWorkflowVersionV4 ,
571- replaytests . VersionWorkflowVersionV5 ,
572- replaytests . VersionWorkflowVersionV6 ,
568+ inCompatibleVersions : []VersionedWorkflowVersion {
569+ VersionedWorkflowVersionV4 ,
570+ VersionedWorkflowVersionV5 ,
571+ VersionedWorkflowVersionV6 ,
573572 },
574573 })
575574}
@@ -579,15 +578,15 @@ func (ts *IntegrationTestSuite) TestVersionedWorkflowV1() {
579578// but not on VersionedWorkflowV4, VersionedWorkflowV5, VersionedWorkflowV6.
580579func (ts * IntegrationTestSuite ) TestVersionedWorkflowV2 () {
581580 ts .testVersionedWorkflow (testVersionedWorkflowTestCase {
582- version : replaytests . VersionWorkflowVersionV2 ,
583- compatibleVersions : []replaytests. VersionWorkflowVersion {
584- replaytests . VersionWorkflowVersionV1 ,
585- replaytests . VersionWorkflowVersionV3 ,
581+ version : VersionedWorkflowVersionV2 ,
582+ compatibleVersions : []VersionedWorkflowVersion {
583+ VersionedWorkflowVersionV1 ,
584+ VersionedWorkflowVersionV3 ,
586585 },
587- inCompatibleVersions : []replaytests. VersionWorkflowVersion {
588- replaytests . VersionWorkflowVersionV4 ,
589- replaytests . VersionWorkflowVersionV5 ,
590- replaytests . VersionWorkflowVersionV6 ,
586+ inCompatibleVersions : []VersionedWorkflowVersion {
587+ VersionedWorkflowVersionV4 ,
588+ VersionedWorkflowVersionV5 ,
589+ VersionedWorkflowVersionV6 ,
591590 },
592591 })
593592}
@@ -597,15 +596,15 @@ func (ts *IntegrationTestSuite) TestVersionedWorkflowV2() {
597596// but not on VersionedWorkflowV1
598597func (ts * IntegrationTestSuite ) TestVersionedWorkflowV3 () {
599598 ts .testVersionedWorkflow (testVersionedWorkflowTestCase {
600- version : replaytests . VersionWorkflowVersionV3 ,
601- compatibleVersions : []replaytests. VersionWorkflowVersion {
602- replaytests . VersionWorkflowVersionV2 ,
603- replaytests . VersionWorkflowVersionV4 ,
604- replaytests . VersionWorkflowVersionV5 ,
605- replaytests . VersionWorkflowVersionV6 ,
599+ version : VersionedWorkflowVersionV3 ,
600+ compatibleVersions : []VersionedWorkflowVersion {
601+ VersionedWorkflowVersionV2 ,
602+ VersionedWorkflowVersionV4 ,
603+ VersionedWorkflowVersionV5 ,
604+ VersionedWorkflowVersionV6 ,
606605 },
607- inCompatibleVersions : []replaytests. VersionWorkflowVersion {
608- replaytests . VersionWorkflowVersionV1 ,
606+ inCompatibleVersions : []VersionedWorkflowVersion {
607+ VersionedWorkflowVersionV1 ,
609608 },
610609 })
611610}
@@ -615,15 +614,15 @@ func (ts *IntegrationTestSuite) TestVersionedWorkflowV3() {
615614// but not on VersionedWorkflowV1
616615func (ts * IntegrationTestSuite ) TestVersionedWorkflowV4 () {
617616 ts .testVersionedWorkflow (testVersionedWorkflowTestCase {
618- version : replaytests . VersionWorkflowVersionV4 ,
619- compatibleVersions : []replaytests. VersionWorkflowVersion {
620- replaytests . VersionWorkflowVersionV2 ,
621- replaytests . VersionWorkflowVersionV3 ,
622- replaytests . VersionWorkflowVersionV5 ,
623- replaytests . VersionWorkflowVersionV6 ,
617+ version : VersionedWorkflowVersionV4 ,
618+ compatibleVersions : []VersionedWorkflowVersion {
619+ VersionedWorkflowVersionV2 ,
620+ VersionedWorkflowVersionV3 ,
621+ VersionedWorkflowVersionV5 ,
622+ VersionedWorkflowVersionV6 ,
624623 },
625- inCompatibleVersions : []replaytests. VersionWorkflowVersion {
626- replaytests . VersionWorkflowVersionV1 ,
624+ inCompatibleVersions : []VersionedWorkflowVersion {
625+ VersionedWorkflowVersionV1 ,
627626 },
628627 })
629628}
@@ -633,15 +632,15 @@ func (ts *IntegrationTestSuite) TestVersionedWorkflowV4() {
633632// but not on VersionedWorkflowV1.
634633func (ts * IntegrationTestSuite ) TestVersionedWorkflowV5 () {
635634 ts .testVersionedWorkflow (testVersionedWorkflowTestCase {
636- version : replaytests . VersionWorkflowVersionV5 ,
637- compatibleVersions : []replaytests. VersionWorkflowVersion {
638- replaytests . VersionWorkflowVersionV2 ,
639- replaytests . VersionWorkflowVersionV3 ,
640- replaytests . VersionWorkflowVersionV4 ,
641- replaytests . VersionWorkflowVersionV6 ,
635+ version : VersionedWorkflowVersionV5 ,
636+ compatibleVersions : []VersionedWorkflowVersion {
637+ VersionedWorkflowVersionV2 ,
638+ VersionedWorkflowVersionV3 ,
639+ VersionedWorkflowVersionV4 ,
640+ VersionedWorkflowVersionV6 ,
642641 },
643- inCompatibleVersions : []replaytests. VersionWorkflowVersion {
644- replaytests . VersionWorkflowVersionV1 ,
642+ inCompatibleVersions : []VersionedWorkflowVersion {
643+ VersionedWorkflowVersionV1 ,
645644 },
646645 })
647646}
@@ -651,41 +650,41 @@ func (ts *IntegrationTestSuite) TestVersionedWorkflowV5() {
651650// but not on VersionedWorkflowV1, VersionedWorkflowV2, VersionedWorkflowV3, VersionedWorkflowV4.
652651func (ts * IntegrationTestSuite ) TestVersionedWorkflowV6 () {
653652 ts .testVersionedWorkflow (testVersionedWorkflowTestCase {
654- version : replaytests . VersionWorkflowVersionV6 ,
655- compatibleVersions : []replaytests. VersionWorkflowVersion {
656- replaytests . VersionWorkflowVersionV5 ,
653+ version : VersionedWorkflowVersionV6 ,
654+ compatibleVersions : []VersionedWorkflowVersion {
655+ VersionedWorkflowVersionV5 ,
657656 },
658- inCompatibleVersions : []replaytests. VersionWorkflowVersion {
659- replaytests . VersionWorkflowVersionV1 ,
660- replaytests . VersionWorkflowVersionV2 ,
661- replaytests . VersionWorkflowVersionV3 ,
662- replaytests . VersionWorkflowVersionV4 ,
657+ inCompatibleVersions : []VersionedWorkflowVersion {
658+ VersionedWorkflowVersionV1 ,
659+ VersionedWorkflowVersionV2 ,
660+ VersionedWorkflowVersionV3 ,
661+ VersionedWorkflowVersionV4 ,
663662 },
664663 })
665664}
666665
667666type testVersionedWorkflowTestCase struct {
668- version replaytests. VersionWorkflowVersion
669- compatibleVersions []replaytests. VersionWorkflowVersion
670- inCompatibleVersions []replaytests. VersionWorkflowVersion
667+ version VersionedWorkflowVersion
668+ compatibleVersions []VersionedWorkflowVersion
669+ inCompatibleVersions []VersionedWorkflowVersion
671670}
672671
673672// testVersionedWorkflow tests that a workflow started on the worker with version
674673// can be replayed on worker with compatibleVersions
675674// but not on worker with inCompatibleVersions
676675func (ts * IntegrationTestSuite ) testVersionedWorkflow (c testVersionedWorkflowTestCase ) {
677- replaytests . SetupWorkerForVersionedWorkflow (c .version , ts .worker )
676+ SetupWorkerForVersionedWorkflow (c .version , ts .worker )
678677 wfId := fmt .Sprintf ("test-versioned-workflow-v%d" , c .version )
679- execution , err := ts .executeWorkflow (wfId , replaytests . VersionedWorkflowName , nil , "arg" )
678+ execution , err := ts .executeWorkflow (wfId , VersionedWorkflowName , nil , "arg" )
680679 ts .NoError (err )
681680
682681 c .compatibleVersions = append (c .compatibleVersions , c .version )
683682
684- ts .Require ().Equalf (len (c .compatibleVersions )+ len (c .inCompatibleVersions ), int (replaytests . MaxVersionWorkflowVersion ),
683+ ts .Require ().Equalf (len (c .compatibleVersions )+ len (c .inCompatibleVersions ), int (MaxVersionedWorkflowVersion ),
685684 "Test case should cover all versions, but got %d compatible (one of them the testing version itself) and %d incompatible versions, that not equal to %d" ,
686685 len (c .compatibleVersions ),
687686 len (c .inCompatibleVersions ),
688- replaytests . MaxVersionWorkflowVersion )
687+ MaxVersionedWorkflowVersion )
689688
690689 for _ , replayedVersion := range c .compatibleVersions {
691690 err := ts .replayVersionedWorkflow (replayedVersion , execution )
@@ -698,9 +697,9 @@ func (ts *IntegrationTestSuite) testVersionedWorkflow(c testVersionedWorkflowTes
698697 }
699698}
700699
701- func (ts * IntegrationTestSuite ) replayVersionedWorkflow (version replaytests. VersionWorkflowVersion , execution * workflow.Execution ) error {
700+ func (ts * IntegrationTestSuite ) replayVersionedWorkflow (version VersionedWorkflowVersion , execution * workflow.Execution ) error {
702701 replayer := worker .NewWorkflowReplayer ()
703- replaytests . SetupWorkerForVersionedWorkflow (version , replayer )
702+ SetupWorkerForVersionedWorkflow (version , replayer )
704703 return replayer .ReplayWorkflowExecution (context .Background (), ts .rpcClient , zaptest .NewLogger (ts .T ()), domainName , * execution )
705704}
706705
0 commit comments