e2e-tests/load-tests/src/test/scala/com/google/gerrit/scenarios/ReplayRecordsFromFeeder.scala

// Copyright (C) 2019 The Android Open Source Project
//
// 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.gerrit.scenarios

import com.github.barbasa.gatling.git.protocol.GitProtocol
import com.github.barbasa.gatling.git.request.builder.GitRequestBuilder
import io.gatling.core.Predef._
import io.gatling.core.structure.ScenarioBuilder
import java.io._

import com.github.barbasa.gatling.git.{
  GatlingGitConfiguration,
  GitRequestSession
}
import org.apache.commons.io.FileUtils

import scala.concurrent.duration._
import org.eclipse.jgit.hooks._

class ReplayRecordsFromFeederScenario extends Simulation {

  val gitProtocol = GitProtocol()
  implicit val conf = GatlingGitConfiguration()
  implicit val postMessageHook: Option[String] = Some(
    s"hooks/${CommitMsgHook.NAME}")

  val feeder = jsonFile("data/requests.json").circular

  val replayCallsScenario: ScenarioBuilder =
    scenario("Git commands")
      .repeat(10000) {
        feed(feeder)
          .exec(new GitRequestBuilder(GitRequestSession("${cmd}", "${url}")))
      }

  setUp(
    replayCallsScenario.inject(
      nothingFor(4 seconds),
      atOnceUsers(10),
      rampUsers(10) during (5 seconds),
      constantUsersPerSec(20) during (15 seconds),
      constantUsersPerSec(20) during (15 seconds) randomized
    ))
    .protocols(gitProtocol)
    .maxDuration(60 seconds)

  after {
    try {
      //After is often called too early. Some retries should be implemented.
      Thread.sleep(5000)
      FileUtils.deleteDirectory(new File(conf.tmpBasePath))
    } catch {
      case e: IOException => {
        System.err.println(
          "Unable to delete temporary directory: " + conf.tmpBasePath)
        e.printStackTrace
      }
    }
  }
}