diff --git a/internal/cli/base_test.go b/internal/cli/base_test.go index 32c6a7c358b..c1456bf45ed 100644 --- a/internal/cli/base_test.go +++ b/internal/cli/base_test.go @@ -260,7 +260,9 @@ func Test_remoteOpPreferred(t *testing.T) { } // Register a remote runner - _, remoteRunnerClose := singleprocess.TestRunner(t, client, &pb.Runner{Odr: false}) + _, remoteRunnerClose := singleprocess.TestRunner(t, client, &pb.Runner{ + Kind: &pb.Runner_Remote_{Remote: &pb.Runner_Remote{}}, + }) defer remoteRunnerClose() // Register a non-default runner profile diff --git a/internal/client/job.go b/internal/client/job.go index 2e46bb09714..b7a2b795704 100644 --- a/internal/client/job.go +++ b/internal/client/job.go @@ -188,6 +188,7 @@ func (c *Project) queueAndStreamJob( } switch event := resp.Event.(type) { + case *pb.GetJobStreamResponse_Complete_: completed = true