mirror of
https://github.com/ceph/ceph
synced 2024-12-27 05:53:20 +00:00
148 lines
4.1 KiB
Python
148 lines
4.1 KiB
Python
#!/usr/bin/env python
|
|
|
|
from nose.tools import (
|
|
eq_ as eq,
|
|
assert_equal,
|
|
assert_raises,
|
|
with_setup,
|
|
)
|
|
|
|
import os
|
|
import shutil
|
|
import random
|
|
import yaml
|
|
from .. import suite
|
|
|
|
archive_base_dir = 'test_archive'
|
|
|
|
|
|
def get_random_metadata(name, hung=False):
|
|
"""
|
|
Generate a random info dict for a fake job. If 'hung' is not True, also
|
|
generate a summary dict.
|
|
|
|
:param name: test name e.g. 'test_foo'
|
|
:param hung: simulate a hung job e.g. don't return a summary.yaml
|
|
:return: a dict with keys 'job_id', 'info' and possibly 'summary', with
|
|
corresponding values
|
|
"""
|
|
rand = random.Random()
|
|
|
|
description = 'description for job with name %s' % name
|
|
owner = 'job@owner'
|
|
duration = rand.randint(1, 36000)
|
|
pid = rand.randint(1000, 99999)
|
|
job_id = rand.randint(1, 99999)
|
|
|
|
info = {
|
|
'description': description,
|
|
'job_id': job_id,
|
|
'name': name,
|
|
'owner': owner,
|
|
'pid': pid,
|
|
}
|
|
|
|
metadata = {
|
|
'info': info,
|
|
'job_id': job_id,
|
|
}
|
|
|
|
if not hung:
|
|
success = True if rand.randint(0, 1) != 1 else False
|
|
|
|
summary = {
|
|
'description': description,
|
|
'duration': duration,
|
|
'owner': owner,
|
|
'success': success,
|
|
}
|
|
|
|
if not success:
|
|
summary['failure_reason'] = 'Failure reason!'
|
|
metadata['summary'] = summary
|
|
|
|
return metadata
|
|
|
|
|
|
def make_archive_subdir():
|
|
os.mkdir(archive_base_dir)
|
|
|
|
|
|
def populate_archive(jobs):
|
|
for job in jobs:
|
|
archive_dir = os.path.join(archive_base_dir, str(job['job_id']))
|
|
os.mkdir(archive_dir)
|
|
|
|
with file(os.path.join(archive_dir, 'info.yaml'), 'w') as yfile:
|
|
yaml.safe_dump(job['info'], yfile)
|
|
|
|
if 'summary' in job:
|
|
with file(os.path.join(archive_dir, 'summary.yaml'), 'w') as yfile:
|
|
yaml.safe_dump(job['summary'], yfile)
|
|
|
|
|
|
def teardown_fake_archive():
|
|
shutil.rmtree(archive_base_dir)
|
|
|
|
|
|
reference = {
|
|
'name': 'test_name',
|
|
'jobs': [
|
|
{'info': {'description': 'description for job with name test_name',
|
|
'job_id': 30481, 'name': 'test_name', 'owner': 'job@owner',
|
|
'pid': 80399},
|
|
'job_id': 30481},
|
|
{'info': {'description': 'description for job with name test_name',
|
|
'job_id': 88979, 'name': 'test_name', 'owner': 'job@owner',
|
|
'pid': 3903},
|
|
'job_id': 88979,
|
|
'summary': {
|
|
'description': 'description for job with name test_name',
|
|
'duration': 35190, 'failure_reason': 'Failure reason!',
|
|
'owner': 'job@owner', 'success': False}},
|
|
{'info': {'description': 'description for job with name test_name',
|
|
'job_id': 68369, 'name': 'test_name', 'owner': 'job@owner',
|
|
'pid': 38524},
|
|
'job_id': 68369,
|
|
'summary': {'description': 'description for job with name test_name',
|
|
'duration': 33771, 'owner': 'job@owner', 'success':
|
|
True}},
|
|
],
|
|
'subject': '1 failed, 1 hung, 1 passed in test_name',
|
|
'body': """
|
|
Test Run: test_name
|
|
=================================================================
|
|
logs: http://qa-proxy.ceph.com/teuthology/test_archive/
|
|
failed: 1
|
|
hung: 1
|
|
passed: 1
|
|
|
|
Failed
|
|
=================================================================
|
|
[88979] description for job with name test_name
|
|
-----------------------------------------------------------------
|
|
time: 35190s
|
|
log: http://qa-proxy.ceph.com/teuthology/test_archive/88979/
|
|
|
|
Failure reason!
|
|
|
|
|
|
Hung
|
|
=================================================================
|
|
[30481] description for job with name test_name
|
|
|
|
Passed
|
|
=================================================================
|
|
[68369] description for job with name test_name
|
|
time: 33771s
|
|
""".strip(),
|
|
}
|
|
|
|
|
|
@with_setup(make_archive_subdir, teardown_fake_archive)
|
|
def test_build_email_body():
|
|
populate_archive(reference['jobs'])
|
|
(subject, body) = suite.build_email_body(reference['name'], archive_base_dir, 36000)
|
|
assert_equal(subject, reference['subject'])
|
|
assert_equal(body, reference['body'])
|