public void testMillisProviderSecurity() {
    if (OLD_JDK) {
        return;
    }
    try {
        try {
            Policy.setPolicy(RESTRICT);
            System.setSecurityManager(new SecurityManager());
            DateTimeUtils.setCurrentMillisProvider(new MillisProvider() {
                public long getMillis() {
                    return 0L;
                }
            });
            fail();
        } catch (SecurityException ex) {
            // ok
        } finally {
            System.setSecurityManager(null);
            Policy.setPolicy(ALLOW);
        }
    } finally {
        DateTimeUtils.setCurrentMillisSystem();
    }
}
 

public void testMillisProviderSecurity() {
    if (OLD_JDK) {
        return;
    }
    try {
        try {
            Policy.setPolicy(RESTRICT);
            System.setSecurityManager(new SecurityManager());
            DateTimeUtils.setCurrentMillisProvider(new MillisProvider() {
                public long getMillis() {
                    return 0L;
                }
            });
            fail();
        } catch (SecurityException ex) {
            // ok
        } finally {
            System.setSecurityManager(null);
            Policy.setPolicy(ALLOW);
        }
    } finally {
        DateTimeUtils.setCurrentMillisSystem();
    }
}
 

@Before
public void setUp() {
  options = PipelineOptionsFactory.create();
  clock = mock(MillisProvider.class);
  sampler = ExecutionStateSampler.newForTest(clock);
  counterSet = new CounterSet();
}
 

public void testMillisProvider() {
    try {
        DateTimeUtils.setCurrentMillisProvider(new MillisProvider() {
            public long getMillis() {
                return 1L;
            }
        });
        assertEquals(1L, DateTimeUtils.currentTimeMillis());
    } finally {
        DateTimeUtils.setCurrentMillisSystem();
    }
}
 

public void testMillisProvider() {
    try {
        DateTimeUtils.setCurrentMillisProvider(new MillisProvider() {
            public long getMillis() {
                return 1L;
            }
        });
        assertEquals(1L, DateTimeUtils.currentTimeMillis());
    } finally {
        DateTimeUtils.setCurrentMillisSystem();
    }
}
 

private ExecutionStateSampler(MillisProvider clock) {
  this.clock = clock;
}
 

@VisibleForTesting
public static ExecutionStateSampler newForTest(MillisProvider clock) {
  return new ExecutionStateSampler(checkNotNull(clock));
}
 

@Before
public void setUp() {
  clock = mock(MillisProvider.class);
  sampler = ExecutionStateSampler.newForTest(clock);
}
 

@Before
public void setUp() {
  clock = mock(MillisProvider.class);
  sampler = ExecutionStateSampler.newForTest(clock);
}