def test_temp_dir_parent(self):
with temp_dir() as p:
with temp_dir(parent=p) as d:
self.assertTrue(os.path.isdir(d))
self.assertEqual(p, os.path.dirname(d))
self.assertFalse(os.path.exists(d))
self.assertFalse(os.path.exists(p))
def test_temp_dir_parent(self):
with temp_dir() as p:
with temp_dir(parent=p) as d:
self.assertTrue(os.path.isdir(d))
self.assertEqual(p, os.path.dirname(d))
self.assertFalse(os.path.exists(d))
self.assertFalse(os.path.exists(p))
def test_temp_dir_keep(self):
with temp_dir() as p:
with temp_dir(parent=p, keep=True) as d:
self.assertTrue(os.path.isdir(d))
open(os.path.join(d, "a-file"), "w").close()
self.assertTrue(os.path.exists(d))
self.assertTrue(os.path.exists(os.path.join(d, "a-file")))
self.assertFalse(os.path.exists(p))
def test_temp_dir_keep(self):
with temp_dir() as p:
with temp_dir(parent=p, keep=True) as d:
self.assertTrue(os.path.isdir(d))
open(os.path.join(d, "a-file"), "w").close()
self.assertTrue(os.path.exists(d))
self.assertTrue(os.path.exists(os.path.join(d, "a-file")))
self.assertFalse(os.path.exists(p))
def test_ignore_older_revision_build(self):
with temp_dir() as root:
path_1234 = make_candidate_dir(
root, '1234-artifacts', 'mybranch', '1234')
make_candidate_dir(root, '1233', 'mybranch', '1233')
self.assertEqual(find_latest_branch_candidates(root), [
(path_1234, 1234)])
def test_ignore_older_revision_build(self):
with temp_dir() as root:
path_1234 = make_candidate_dir(root, '1234-artifacts', 'mybranch',
'1234')
make_candidate_dir(root, '1233', 'mybranch', '1233')
self.assertEqual(find_latest_branch_candidates(root),
[(path_1234, 1234)])
def test__find_candidates_artifacts_enabled(self):
with temp_dir() as root:
make_candidate_dir(root, 'master-artifacts')
make_candidate_dir(root, '1.25')
candidate = os.path.join(root, 'candidate', 'master-artifacts')
self.assertEqual(list(_find_candidates(root, artifacts=True)), [
(candidate, os.path.join(candidate, 'buildvars.json'))])
def test_find_candidates_find_all(self):
with temp_dir() as root:
a_week_ago = time() - timedelta(days=7, seconds=1).total_seconds()
master_path = make_candidate_dir(root, '1.23', modified=a_week_ago)
master_path_2 = make_candidate_dir(root, '1.24')
self.assertItemsEqual(list(find_candidates(root)), [master_path_2])
self.assertItemsEqual(list(find_candidates(root, find_all=True)),
[master_path, master_path_2])
def test__find_candidates_artifacts_enabled(self):
with temp_dir() as root:
make_candidate_dir(root, 'master-artifacts')
make_candidate_dir(root, '1.25')
candidate = os.path.join(root, 'candidate', 'master-artifacts')
self.assertEqual(
list(_find_candidates(root, artifacts=True)),
[(candidate, os.path.join(candidate, 'buildvars.json'))])
def test_find_candidates_find_all(self):
with temp_dir() as root:
a_week_ago = time() - timedelta(days=7, seconds=1).total_seconds()
master_path = make_candidate_dir(root, '1.23', modified=a_week_ago)
master_path_2 = make_candidate_dir(root, '1.24')
self.assertItemsEqual(list(find_candidates(root)), [master_path_2])
self.assertItemsEqual(list(find_candidates(root, find_all=True)),
[master_path, master_path_2])
def test_include_older_revision_build_different_branch(self):
with temp_dir() as root:
path_1234 = make_candidate_dir(root, '1234-artifacts',
'branch_foo', '1234')
path_1233 = make_candidate_dir(root, '1233-artifacts',
'branch_bar', '1233')
self.assertItemsEqual(find_latest_branch_candidates(root),
[(path_1233, 1233), (path_1234, 1234)])
def test_include_older_revision_build_different_branch(self):
with temp_dir() as root:
path_1234 = make_candidate_dir(
root, '1234-artifacts', 'branch_foo', '1234')
path_1233 = make_candidate_dir(
root, '1233-artifacts', 'branch_bar', '1233')
self.assertItemsEqual(
find_latest_branch_candidates(root), [
(path_1233, 1233), (path_1234, 1234)])
def client_and_plan():
cur_client = fake_juju_client()
cur_client.bootstrap()
with patch('hammer_time.hammer_time.client_for_existing',
return_value=cur_client,
autospec=True) as cfe_mock:
with temp_dir() as plan_dir:
plan_file = os.path.join(plan_dir, 'asdf.yaml')
yield (cur_client, cfe_mock, plan_file)
def test_no_warn_on_empty_logs(self):
"""Special case a file named 'empty' doesn't make log dir dirty"""
with warnings.catch_warnings(record=True) as warned:
with temp_dir() as log_dir:
open(os.path.join(log_dir, "empty"), "w").close()
cmd_line = ['local', '/a/juju', log_dir, 'testtest']
parser = add_basic_testing_arguments(ArgumentParser())
parser.parse_args(cmd_line)
self.assertEqual(warned, [])
self.assertEqual("", self.log_stream.getvalue())
def test_no_warn_on_empty_logs(self):
"""Special case a file named 'empty' doesn't make log dir dirty"""
with warnings.catch_warnings(record=True) as warned:
with temp_dir() as log_dir:
open(os.path.join(log_dir, "empty"), "w").close()
cmd_line = ['local', '/a/juju', log_dir, 'testtest']
parser = add_basic_testing_arguments(ArgumentParser())
parser.parse_args(cmd_line)
self.assertEqual(warned, [])
self.assertEqual("", self.log_stream.getvalue())
def test_warns_on_dirty_logs(self):
with warnings.catch_warnings(record=True) as warned:
with temp_dir() as log_dir:
open(os.path.join(log_dir, "existing.log"), "w").close()
cmd_line = ['local', '/a/juju', log_dir, 'testtest']
parser = add_basic_testing_arguments(ArgumentParser())
parser.parse_args(cmd_line)
self.assertEqual(len(warned), 1)
self.assertRegexpMatches(
str(warned[0].message),
r"^Directory '.*' has existing contents.$")
self.assertEqual("", self.log_stream.getvalue())
def test_warns_on_dirty_logs(self):
with warnings.catch_warnings(record=True) as warned:
with temp_dir() as log_dir:
open(os.path.join(log_dir, "existing.log"), "w").close()
cmd_line = ['local', '/a/juju', log_dir, 'testtest']
parser = add_basic_testing_arguments(ArgumentParser())
parser.parse_args(cmd_line)
self.assertEqual(len(warned), 1)
self.assertRegexpMatches(
str(warned[0].message),
r"^Directory '.*' has existing contents.$")
self.assertEqual("", self.log_stream.getvalue())
def test_find_candidates_old_buildvars(self):
with temp_dir() as root:
a_week_ago = time() - timedelta(days=7, seconds=1).total_seconds()
make_candidate_dir(root, 'master', modified=a_week_ago)
self.assertEqual(list(find_candidates(root)), [])
def test_find_latest_branch_candidates(self):
with temp_dir() as root:
master_path = make_candidate_dir(root, 'master-artifacts')
self.assertEqual(find_latest_branch_candidates(root),
[(master_path, 1234)])
def test_find_latest_branch_candidates(self):
with temp_dir() as root:
master_path = make_candidate_dir(root, 'master-artifacts')
self.assertEqual(find_latest_branch_candidates(root),
[(master_path, 1234)])
def test_find_candidates_artifacts(self):
with temp_dir() as root:
make_candidate_dir(root, 'master-artifacts')
self.assertEqual(list(find_candidates(root)), [])
def test_find_candidates(self):
with temp_dir() as root:
master_path = make_candidate_dir(root, 'master')
self.assertEqual(list(find_candidates(root)), [master_path])
def test_find_candidates(self):
with temp_dir() as root:
master_path = make_candidate_dir(root, 'master')
self.assertEqual(list(find_candidates(root)), [master_path])
def test_temp_dir_contents(self):
with temp_dir() as d:
self.assertTrue(os.path.isdir(d))
open(os.path.join(d, "a-file"), "w").close()
self.assertFalse(os.path.exists(d))
def test_temp_dir(self):
with temp_dir() as d:
self.assertTrue(os.path.isdir(d))
self.assertFalse(os.path.exists(d))
def test_find_candidates_old_buildvars(self):
with temp_dir() as root:
a_week_ago = time() - timedelta(days=7, seconds=1).total_seconds()
make_candidate_dir(root, 'master', modified=a_week_ago)
self.assertEqual(list(find_candidates(root)), [])
def test_find_candidates_artifacts(self):
with temp_dir() as root:
make_candidate_dir(root, 'master-artifacts')
self.assertEqual(list(find_candidates(root)), [])
def test_temp_dir(self):
with temp_dir() as d:
self.assertTrue(os.path.isdir(d))
self.assertFalse(os.path.exists(d))
def test_temp_dir_contents(self):
with temp_dir() as d:
self.assertTrue(os.path.isdir(d))
open(os.path.join(d, "a-file"), "w").close()
self.assertFalse(os.path.exists(d))
