#!/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'])

Add test for teuthology.suite.build_email_body()