Function TEST_P

src/tests/disk_quota_tests.cpp:210–308

This test verifies that the container will be killed if the disk usage exceeds its quota.

// This test verifies that the container will be killed if the disk
// usage exceeds its quota.
TEST_P(DiskQuotaEnforcement, DiskUsageExceedsQuota)
{
  // NOTE: We don't use the "ROOT_" tag on the test because that
  // would require the SANDBOX tests to be run as root as well.
  // If we did that, then we would lose disk isolator test coverage
  // in the default CI configuration.
  if (GetParam() == ParamDiskQuota::ROOTFS && ::geteuid() != 0) {
    return;
  }

  Try<Owned<cluster::Master>> master = StartMaster();
  ASSERT_SOME(master);

  slave::Flags flags = CreateSlaveFlags();
  flags.isolation = "posix/cpu,posix/mem,disk/du";

  // NOTE: We can't pause the clock because we need the reaper to reap
  // the 'du' subprocess.
  flags.container_disk_watch_interval = Milliseconds(1);
  flags.enforce_container_disk_quota = true;

  if (GetParam() == ParamDiskQuota::ROOTFS) {
    flags.isolation += ",filesystem/linux,docker/runtime";

    flags.docker_registry = path::join(sandbox.get(), "registry");
    flags.docker_store_dir = path::join(sandbox.get(), "store");
    flags.image_providers = "docker";
    flags.image_provisioner_backend = "overlay";

    AWAIT_READY(CreateDockerImage(flags.docker_registry, "test_image"));
  }

  Owned<MasterDetector> detector = master.get()->createDetector();
  Try<Owned<cluster::Slave>> slave = StartSlave(detector.get(), flags);
  ASSERT_SOME(slave);

  MockScheduler sched;
  MesosSchedulerDriver driver(
      &sched, DEFAULT_FRAMEWORK_INFO, master.get()->pid, DEFAULT_CREDENTIAL);

  EXPECT_CALL(sched, registered(&driver, _, _));

  Future<vector<Offer>> offers;
  EXPECT_CALL(sched, resourceOffers(&driver, _))
    .WillOnce(FutureArg<1>(&offers))
    .WillRepeatedly(Return());        // Ignore subsequent offers.

  driver.start();

  AWAIT_READY(offers);
  ASSERT_FALSE(offers->empty());

  const Offer& offer = offers.get()[0];

  TaskInfo task;

  // Create a task which requests 1MB disk, but actually uses more
  // than 2MB disk.