cleanup some test code

Author: Quinn-With-Two-Ns

temporal-sdk/src/test/java/io/temporal/worker/shutdown/StickyWorkflowDrainShutdownTest.java

@@ -10,6 +10,7 @@
 import io.temporal.worker.WorkerOptions;
 import io.temporal.worker.tuning.PollerBehavior;
 import io.temporal.worker.tuning.PollerBehaviorAutoscaling;
+import io.temporal.worker.tuning.PollerBehaviorSimpleMaximum;
 import io.temporal.workflow.Workflow;
 import io.temporal.workflow.shared.TestWorkflows.TestWorkflow1;
 import java.time.Duration;
@@ -27,7 +28,8 @@ public class StickyWorkflowDrainShutdownTest {
 
   @Parameterized.Parameters
   public static Collection<PollerBehavior> data() {
-    return Arrays.asList(new PollerBehaviorAutoscaling(1, 10, 10));
+    return Arrays.asList(
+        new PollerBehaviorSimpleMaximum(10), new PollerBehaviorAutoscaling(1, 10, 10));
   }
 
   @Rule public SDKTestWorkflowRule testWorkflowRule;
@@ -49,7 +51,7 @@ public StickyWorkflowDrainShutdownTest(PollerBehavior pollerBehaviorAutoscaling)
             .build();
   }
 
-  @Test(timeout = 20000)
+  @Test
   public void testShutdown() throws InterruptedException {
     TestWorkflow1 workflow = testWorkflowRule.newWorkflowStub(TestWorkflow1.class);
     WorkflowClient.start(workflow::execute, null);
@@ -86,7 +88,7 @@ public static class TestWorkflowImpl implements TestWorkflow1 {
 
     @Override
     public String execute(String now) {
-      for (int i = 0; i < 3; i++) {
+      for (int i = 0; i < 5; i++) {
         Workflow.sleep(1000);
       }
       return "Success";