tests/testthat/test-crew_definition_aws_batch.R

In crew.aws.batch: A Crew Launcher Plugin for AWS Batch

test_that("crew_definition_aws_batch()", {
  out <- crew_definition_aws_batch(
    job_queue = "job_queue_name",
    job_definition = "job_definition_name",
    log_group = "/aws/batch/job",
    config = list(config = TRUE),
    credentials = list(credentials = TRUE),
    endpoint = "endpoint_url",
    region = "us-east-2"
  )
  expect_silent(out$validate())
  expect_equal(out$job_queue, "job_queue_name")
  expect_equal(out$job_definition, "job_definition_name")
  expect_equal(out$log_group, "/aws/batch/job")
  expect_equal(out$config, list(config = TRUE))
  expect_equal(out$credentials, list(credentials = TRUE))
  expect_equal(out$endpoint, "endpoint_url")
  expect_equal(out$region, "us-east-2")
})

test_that("crew_definition_aws_batch() private$.client()", {
  skip_on_cran()
  x <- crew_definition_aws_batch(job_queue = "x", region = "us-east-2")
  out <- x$.__enclos_env__$private$.client()
  expect_true(is.function(out$list_jobs))
})

test_that("crew_definition_aws_batch() private$.args_register()", {
  skip_on_cran()
  x <- crew_definition_aws_batch(
    job_definition = "job-definition-name",
    job_queue = "crew-aws-batch-job-queue",
    region = "us-east-2"
  )
  out <- x$.__enclos_env__$private$.args_register(
    image = "alpine:latest",
    platform_capabilities = "EC2",
    memory_units = "gigabytes",
    memory = 1,
    cpus = 1,
    gpus = 1,
    seconds_timeout = 600,
    scheduling_priority = 3,
    tags = c("crew_aws_batch_1", "crew_aws_batch_2"),
    propagate_tags = TRUE,
    parameters = c("a=1", "b=2"),
    job_role_arn = "job_role_arn_id",
    execution_role_arn = "execution_role_arn_id"
  )
  exp <- list(
    jobDefinitionName = "job-definition-name",
    type = "container",
    schedulingPriority = 3,
    tags = list("crew_aws_batch_1", "crew_aws_batch_2"),
    propagateTags = TRUE,
    parameters = list("a=1", "b=2"),
    platformCapabilities = "EC2",
    timeout = list(attemptDurationSeconds = 600),
    containerProperties = list(
      image = "alpine:latest",
      resourceRequirements = list(
        memory = list(value = "954", type = "MEMORY"),
        cpus = list(
          value = "1",
          type = "VCPU"
        ),
        gpus = list(
          value = "1",
          type = "GPU"
        )
      ),
      logConfiguration = list(
        logDriver = "awslogs",
        options = list(
          "awslogs-group" = "/aws/batch/job",
          "awslogs-region" = "us-east-2",
          "awslogs-stream-prefix" = "job-definition-name"
        )
      )
    )
  )
  expect_equal(out, exp)
})

test_that("crew_definition_aws_batch() private$.args_submit()", {
  skip_on_cran()
  x <- crew_definition_aws_batch(
    job_definition = "job-definition-name",
    job_queue = "crew-aws-batch-job-queue",
    region = "us-east-2"
  )
  out <- x$.__enclos_env__$private$.args_submit(
    command = c("sleep", "256"),
    name = "crew-aws-batch-job-test",
    memory_units = "gigabytes",
    memory = 234,
    cpus = 8,
    gpus = 3,
    seconds_timeout = 345,
    share_identifier = "identifier",
    scheduling_priority_override = 6,
    tags = c("tag1", "tag2"),
    propagate_tags = FALSE,
    parameters = c("key1=value1", "key2=value2")
  )
  exp <- list(
    jobName = "crew-aws-batch-job-test",
    jobQueue = "crew-aws-batch-job-queue",
    shareIdentifier = "identifier",
    schedulingPriorityOverride = 6,
    jobDefinition = "job-definition-name",
    parameters = c("key1=value1", "key2=value2"),
    tags = c("tag1", "tag2"),
    propagateTags = FALSE,
    timeout = list(attemptDurationSeconds = 345),
    containerOverrides = list(
      command = list("sleep", "256"),
      resourceRequirements = list(
        memory = list(value = "223160", type = "MEMORY"),
        cpus = list(value = "8", type = "VCPU"),
        gpus = list(value = "3", type = "GPU")
      )
    )
  )
  expect_equal(out, exp)
})