/*
 * Copyright 2017 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.google.cloud.spanner;

import static org.mockito.Mockito.any;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.when;

import com.google.cloud.grpc.GrpcTransportOptions.ExecutorFactory;
import com.google.cloud.spanner.SessionPool.Clock;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import org.threeten.bp.Instant;

abstract class BaseSessionPoolTest {
  ScheduledExecutorService mockExecutor;
  int sessionIndex;

  final class TestExecutorFactory implements ExecutorFactory<ScheduledExecutorService> {

    @Override
    public ScheduledExecutorService get() {
      ScheduledExecutorService realExecutor = new ScheduledThreadPoolExecutor(2);
      mockExecutor = spy(realExecutor);
      @SuppressWarnings("rawtypes")
      ScheduledFuture mockFuture = mock(ScheduledFuture.class);
      // To prevent maintenance loop from running.
      doReturn(mockFuture)
          .when(mockExecutor)
          .scheduleAtFixedRate(
              any(Runnable.class), any(Long.class), any(Long.class), any(TimeUnit.class));
      return mockExecutor;
    }

    @Override
    public void release(ScheduledExecutorService executor) {
      // do nothing
    }
  }

  Session mockSession() {
    Session session = mock(Session.class);
    when(session.getName())
        .thenReturn(
            "projects/dummy/instances/dummy/database/dummy/sessions/session" + sessionIndex);
    sessionIndex++;
    return session;
  }

  void runMaintainanceLoop(FakeClock clock, SessionPool pool, long numCycles) {
    for (int i = 0; i < numCycles; i++) {
      pool.poolMaintainer.maintainPool();
      clock.currentTimeMillis += SessionPool.PoolMaintainer.LOOP_FREQUENCY;
    }
  }

  static class FakeClock extends Clock {
    volatile long currentTimeMillis;

    @Override
    public Instant instant() {
      return Instant.ofEpochMilli(currentTimeMillis);
    }
  }
}

/*
 * Copyright 2017 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.google.cloud.spanner;

import static org.mockito.Mockito.any;
import static org.mockito.Mockito.anyLong;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.when;

import com.google.cloud.grpc.GrpcTransportOptions.ExecutorFactory;
import com.google.cloud.spanner.SessionPool.Clock;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import org.threeten.bp.Instant;

abstract class BaseSessionPoolTest {
  ScheduledExecutorService mockExecutor;
  int sessionIndex;

  final class TestExecutorFactory implements ExecutorFactory<ScheduledExecutorService> {

    @Override
    public ScheduledExecutorService get() {
      ScheduledExecutorService realExecutor = new ScheduledThreadPoolExecutor(2);
      mockExecutor = spy(realExecutor);
      @SuppressWarnings("rawtypes")
      ScheduledFuture mockFuture = mock(ScheduledFuture.class);
      // To prevent maintenance loop from running.
      doReturn(mockFuture)
          .when(mockExecutor)
          .scheduleAtFixedRate(any(Runnable.class), anyLong(), anyLong(), any(TimeUnit.class));
      return mockExecutor;
    }

    @Override
    public void release(ScheduledExecutorService executor) {
      // do nothing
    }
  }

  Session mockSession() {
    Session session = mock(Session.class);
    when(session.getName())
        .thenReturn(
            "projects/dummy/instances/dummy/database/dummy/sessions/session" + sessionIndex);
    sessionIndex++;
    return session;
  }

  void runMaintainanceLoop(FakeClock clock, SessionPool pool, long numCycles) {
    for (int i = 0; i < numCycles; i++) {
      pool.poolMaintainer.maintainPool();
      clock.currentTimeMillis += SessionPool.PoolMaintainer.LOOP_FREQUENCY;
    }
  }

  static class FakeClock extends Clock {
    volatile long currentTimeMillis;

    @Override
    public Instant instant() {
      return Instant.ofEpochMilli(currentTimeMillis);
    }
  }
}

Fix usages of any(<Primitive>.class) matchers (#4453)