recipe_modules/file/examples/flatten_single_directories.py

# Copyright 2018 The LUCI Authors. All rights reserved.
# Use of this source code is governed under the Apache License, Version 2.0
# that can be found in the LICENSE file.

DEPS = [
  'file',
  'path',
]


def RunSteps(api):
  base = api.path['start_dir'].join('dir')
  long_dir = base.join('which_has', 'some', 'singlular', 'subdirs')

  api.file.ensure_directory('make chain of single dirs', long_dir)

  filenames = ['bunch', 'of', 'files']
  for n in filenames:
    api.file.truncate('touch %s' % n, long_dir.join(n), 1)

  api.file.flatten_single_directories('remove single dirs', base)
  # to satisfy simulation; run this example for real to get the useful
  # assertions below.
  for n in filenames:
    api.path.mock_add_paths(base.join(n))

  for n in filenames:
    path = base.join(n)
    assert api.path.exists(path), path


def GenTests(api):
  yield api.test('basic')