Python TaskRunnerHelper.this_is_really_our_pid示例

示例#1

文件： test_helper.py 项目： retrack/incubator-aurora

def test_this_is_really_our_pid():
    process = mock.Mock(spec=psutil.Process, username=USER1, pid=PID, create_time=CREATE_TIME)
    assert TRH.this_is_really_our_pid(process, process.username, process.create_time)
    assert TRH.this_is_really_our_pid(
        process, process.username, process.create_time + TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) - 1
    )
    assert not TRH.this_is_really_our_pid(process, "user2", process.create_time)
    assert not TRH.this_is_really_our_pid(
        process, process.username, process.create_time + TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) + 1
    )
    assert not TRH.this_is_really_our_pid(
        process, process.username, process.create_time - (TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) + 1)
    )

示例#2

文件： test_helper.py 项目： bhuvan/incubator-aurora

def test_this_is_really_our_pid():
    process = mock.Mock(spec=psutil.Process, pid=PID)
    process.username.return_value = USER1
    process.create_time.return_value = CREATE_TIME
    assert TRH.this_is_really_our_pid(process, process.username(),
                                      process.create_time())
    assert TRH.this_is_really_our_pid(
        process, process.username(),
        process.create_time() + TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) - 1)
    assert not TRH.this_is_really_our_pid(process, 'user2',
                                          process.create_time())
    assert not TRH.this_is_really_our_pid(
        process, process.username(),
        process.create_time() + TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) + 1)
    assert not TRH.this_is_really_our_pid(
        process, process.username(),
        process.create_time() -
        (TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) + 1))

示例#3

文件： test_helper.py 项目： bhuvan/incubator-aurora

def test_this_is_really_our_pid():
  process = mock.Mock(spec=psutil.Process, pid=PID)
  process.username.return_value = USER1
  process.create_time.return_value = CREATE_TIME
  assert TRH.this_is_really_our_pid(
      process,
      process.username(),
      process.create_time())
  assert TRH.this_is_really_our_pid(
      process,
      process.username(),
      process.create_time() + TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) - 1)
  assert not TRH.this_is_really_our_pid(
      process,
      'user2',
      process.create_time())
  assert not TRH.this_is_really_our_pid(
      process,
      process.username(),
      process.create_time() + TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) + 1)
  assert not TRH.this_is_really_our_pid(
      process,
      process.username(),
      process.create_time() - (TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) + 1))

示例#4

def test_this_is_really_our_pid():
    process = mock_process(PID, USER1, uid=UID)
    assert TRH.this_is_really_our_pid(process, UID, USER1,
                                      process.create_time())
    assert TRH.this_is_really_our_pid(
        process, UID, USER1,
        process.create_time() + TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) - 1)
    assert TRH.this_is_really_our_pid(
        process, UID, 'user2', process.create_time()), (
            'An equivalent UID is considered the same user.')
    assert not TRH.this_is_really_our_pid(
        process, UID + 1, USER1,
        process.create_time()), ('UIDs should not match.')
    assert not TRH.this_is_really_our_pid(
        process, UID, USER1,
        process.create_time() + TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) +
        1), ('Process is outside of start time drift.')
    assert not TRH.this_is_really_our_pid(
        process, UID, USER1,
        process.create_time() -
        (TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) + 1)), (
            'Process is outside of start time drift.')
    assert not TRH.this_is_really_our_pid(
        process, None, 'user2', process.create_time()
    ), ("If no uid is checkpointed but the username is different, we can't know it's ours."
        )

    process = mock_process(PID, USER1)
    assert not TRH.this_is_really_our_pid(
        process, UID, USER1, process.create_time()
    ), ('We cannot validate whether this is our process without a process UID.'
        )
    assert TRH.this_is_really_our_pid(
        process, None, USER1,
        process.create_time()), ('Previous behavior is preserved.')

    process = mock_process(PID, username=KeyError('Unknown user'), uid=UID)
    assert TRH.this_is_really_our_pid(process, UID, USER1,
                                      process.create_time())
    assert not TRH.this_is_really_our_pid(process, None, USER1,
                                          process.create_time())
    assert TRH.this_is_really_our_pid(
        process, UID, 'user2', process.create_time()
    ), ('If the user has been renamed but the UID is the same, this is still our process.'
        )

示例#5

文件： test_helper.py 项目： zmyer/aurora

def test_this_is_really_our_pid_root():
    # Test the case where a process has a non root uid but the uid in the header
    # was root (meaning that it `setuid` itself)
    process = mock_process(PID, USER1, uid=UID)
    assert TRH.this_is_really_our_pid(process, ROOT_UID, USER1,
                                      process.create_time())

示例#6

文件： test_helper.py 项目： AltanAlpay/aurora

def test_this_is_really_our_pid():
  process = mock_process(PID, USER1, uid=UID)
  assert TRH.this_is_really_our_pid(process, UID, USER1, process.create_time())
  assert TRH.this_is_really_our_pid(process, UID, USER1,
      process.create_time() + TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) - 1)
  assert TRH.this_is_really_our_pid(process, UID, 'user2', process.create_time()), (
      'An equivalent UID is considered the same user.')
  assert not TRH.this_is_really_our_pid(process, UID + 1, USER1, process.create_time()), (
      'UIDs should not match.')
  assert not TRH.this_is_really_our_pid(process, UID, USER1,
      process.create_time() + TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) + 1), (
          'Process is outside of start time drift.')
  assert not TRH.this_is_really_our_pid(process, UID, USER1,
      process.create_time() - (TRH.MAX_START_TIME_DRIFT.as_(Time.SECONDS) + 1)), (
          'Process is outside of start time drift.')
  assert not TRH.this_is_really_our_pid(process, None, 'user2', process.create_time()), (
      "If no uid is checkpointed but the username is different, we can't know it's ours.")

  process = mock_process(PID, USER1)
  assert not TRH.this_is_really_our_pid(process, UID, USER1, process.create_time()), (
      'We cannot validate whether this is our process without a process UID.')
  assert TRH.this_is_really_our_pid(process, None, USER1, process.create_time()), (
      'Previous behavior is preserved.')

  process = mock_process(PID, username=KeyError('Unknown user'), uid=UID)
  assert TRH.this_is_really_our_pid(process, UID, USER1, process.create_time())
  assert not TRH.this_is_really_our_pid(process, None, USER1, process.create_time())
  assert TRH.this_is_really_our_pid(process, UID, 'user2', process.create_time()), (
      'If the user has been renamed but the UID is the same, this is still our process.')

示例#7

文件： test_helper.py 项目： apache/aurora

def test_this_is_really_our_pid_root():
  # Test the case where a process has a non root uid but the uid in the header
  # was root (meaning that it `setuid` itself)
  process = mock_process(PID, USER1, uid=UID)
  assert TRH.this_is_really_our_pid(process, ROOT_UID, USER1, process.create_time())