Function TEST_P

src/tests/master_draining_tests.cpp:174–232

When an operator submits a DRAIN_AGENT call, the agent with nothing running should be immediately transitioned to the DRAINED state.

// When an operator submits a DRAIN_AGENT call, the agent with nothing running
// should be immediately transitioned to the DRAINED state.
TEST_P(MasterAlreadyDrainedTest, DrainAgent)
{
  Future<Nothing> registrarApplyDrained;
  EXPECT_CALL(*master->registrar, apply(_))
    .WillOnce(DoDefault())
    .WillOnce(DoAll(
        FutureSatisfy(&registrarApplyDrained),
        Invoke(master->registrar.get(), &MockRegistrar::unmocked_apply)));

  ContentType contentType = GetParam();

  {
    v1::master::Call::DrainAgent drainAgent;
    drainAgent.mutable_agent_id()->CopyFrom(agentId);
    drainAgent.mutable_max_grace_period()->set_seconds(10);

    v1::master::Call call;
    call.set_type(v1::master::Call::DRAIN_AGENT);
    call.mutable_drain_agent()->CopyFrom(drainAgent);

    AWAIT_EXPECT_RESPONSE_STATUS_EQ(
        http::OK().status,
        post(master->pid, call, contentType));
  }

  AWAIT_READY(registrarApplyDrained);

  mesos::v1::DrainInfo drainInfo;
  drainInfo.set_state(mesos::v1::DRAINED);
  drainInfo.mutable_config()->set_mark_gone(false);
  drainInfo.mutable_config()->mutable_max_grace_period()
    ->set_nanoseconds(Seconds(10).ns());

  // Ensure that the agent's drain info is reflected in the master's
  // GET_AGENTS response.
  {
    v1::master::Call call;
    call.set_type(v1::master::Call::GET_AGENTS);

    Future<http::Response> response =
      post(master->pid, call, contentType);
    AWAIT_ASSERT_RESPONSE_STATUS_EQ(http::OK().status, response);

    Try<v1::master::Response> getAgents =
      deserialize<v1::master::Response>(contentType, response->body);
    ASSERT_SOME(getAgents);

    ASSERT_EQ(v1::master::Response::GET_AGENTS, getAgents->type());
    ASSERT_EQ(getAgents->get_agents().agents_size(), 1);

    const v1::master::Response::GetAgents::Agent& agent =
        getAgents->get_agents().agents(0);

    EXPECT_EQ(agent.deactivated(), true);

    EXPECT_EQ(agent.drain_info(), drainInfo);
    EXPECT_LT(0, agent.estimated_drain_start_time().nanoseconds());
  }