Python ExecutorDetector Examples

Example #1

File: test_executor_detector.py Project: xuchenw/incubator-aurora

def test_integration():
  SLAVES = ('slave001', 'slave123')
  FRAMEWORKS = ('framework1', 'framework2')
  EXECUTORS = ('executor_a', 'executor_b')
  RUNS = ('001', '002', 'latest')

  with temporary_dir() as td:
    all_groups = set()
    for slave, framework, executor, run in itertools.product(SLAVES, FRAMEWORKS, EXECUTORS, RUNS):
      match = Match(td, slave, framework, executor, run)
      safe_mkdir(ExecutorDetector.path(match))
      all_groups.add(tuple(sorted(match.groups().items())))

    for match in ExecutorDetector(td):
      assert tuple(sorted(match.groups().items())) in all_groups

Example #2

File: test_executor_detector.py Project: betepahos/incubator-aurora

def test_find_root():
  BAD_PATHS = (
    os.path.sep,
    '.',
    os.path.sep * 10,
    '/root/slaves',
    '/root/slaves/S/frameworks/F/executors//runs/R',
    'root/slaves/S/frameworks/F/executors//runs/R',
  )

  GOOD_PATHS = (
    ExecutorDetector.path(DEFAULT_MATCH),
    os.path.join(ExecutorDetector.path(DEFAULT_MATCH), 'some', 'other', 'path')
  )

  for cwd in BAD_PATHS:
    assert ExecutorDetector.find_root(cwd) is None

  for cwd in GOOD_PATHS:
    assert ExecutorDetector.find_root(cwd) == 'abcd'

Example #3

File: test_executor_detector.py Project: xuchenw/incubator-aurora

def test_find_root():
  BAD_PATHS = (
    os.path.sep,
    '.',
    os.path.sep * 10,
    '/root/slaves',
    '/root/slaves/S/frameworks/F/executors//runs/R',
    'root/slaves/S/frameworks/F/executors//runs/R',
  )

  GOOD_PATHS = (
    ExecutorDetector.path(DEFAULT_MATCH),
    os.path.join(ExecutorDetector.path(DEFAULT_MATCH), 'some', 'other', 'path')
  )

  for cwd in BAD_PATHS:
    assert ExecutorDetector.find_root(cwd) is None

  for cwd in GOOD_PATHS:
    assert ExecutorDetector.find_root(cwd) == 'abcd'

Example #4

File: test_executor_detector.py Project: kevinburg/incubator-aurora

def test_find_root():
    BAD_PATHS = (
        os.path.sep,
        ".",
        os.path.sep * 10,
        "/root/slaves",
        "/root/slaves/S/frameworks/F/executors//runs/R",
        "root/slaves/S/frameworks/F/executors//runs/R",
    )

    GOOD_PATHS = (
        ExecutorDetector.path(DEFAULT_MATCH),
        os.path.join(ExecutorDetector.path(DEFAULT_MATCH), "some", "other", "path"),
    )

    for cwd in BAD_PATHS:
        assert ExecutorDetector.find_root(cwd) is None

    for cwd in GOOD_PATHS:
        assert ExecutorDetector.find_root(cwd) == "abcd"

Example #5

File: test_executor_detector.py Project: betepahos/incubator-aurora

def test_integration():
  SLAVES = ('slave001', 'slave123')
  FRAMEWORKS = ('framework1', 'framework2')
  EXECUTORS = ('executor_a', 'executor_b')
  RUNS = ('001', '002', 'latest')

  with temporary_dir() as td:
    all_groups = set()
    for slave, framework, executor, run in itertools.product(SLAVES, FRAMEWORKS, EXECUTORS, RUNS):
      match = Match(td, slave, framework, executor, run)
      safe_mkdir(ExecutorDetector.path(match))
      all_groups.add(tuple(sorted(match.groups().items())))

    for match in ExecutorDetector(td):
      assert tuple(sorted(match.groups().items())) in all_groups

Example #6

File: test_executor_detector.py Project: kevinburg/incubator-aurora

def test_integration():
    SLAVES = ("slave001", "slave123")
    FRAMEWORKS = ("framework1", "framework2")
    EXECUTORS = ("executor_a", "executor_b")
    RUNS = ("001", "002", "latest")

    with temporary_dir() as td:
        all_groups = set()
        for slave, framework, executor, run in itertools.product(SLAVES, FRAMEWORKS, EXECUTORS, RUNS):
            match = Match(td, slave, framework, executor, run)
            safe_mkdir(ExecutorDetector.path(match))
            all_groups.add(tuple(sorted(match.groups().items())))

        for match in ExecutorDetector(td):
            assert tuple(sorted(match.groups().items())) in all_groups

Example #7

File: test_executor_detector.py Project: betepahos/incubator-aurora

def test_bad_match():
  assert ExecutorDetector.match('herpderp') is None

Example #8

File: test_executor_detector.py Project: betepahos/incubator-aurora

def test_match_inverse():
  assert ExecutorDetector.match(ExecutorDetector.path(DEFAULT_MATCH)).groups() == (
      DEFAULT_MATCH.groups())

Example #9

File: test_executor_detector.py Project: xuchenw/incubator-aurora

def test_bad_match():
  assert ExecutorDetector.match('herpderp') is None

Example #10

File: test_executor_detector.py Project: xuchenw/incubator-aurora

def test_match_inverse():
  assert ExecutorDetector.match(ExecutorDetector.path(DEFAULT_MATCH)).groups() == (
      DEFAULT_MATCH.groups())