import json
import os
import shutil
import tempfile
import unittest

from avocado.core import exit_codes
from avocado.utils import process
from avocado.utils import script

basedir = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..', '..')
basedir = os.path.abspath(basedir)

AVOCADO_TEST_CANCEL = """
import avocado

class AvocadoCancelTest(avocado.Test):

    def test(self):
        self.cancel()
"""

AVOCADO_TEST_CANCEL_ON_SETUP = """
import avocado

class AvocadoCancelTest(avocado.Test):

    def setUp(self):
        self.cancel()

    def test(self):
        pass
"""


class TestCancel(unittest.TestCase):

    def setUp(self):
        os.chdir(basedir)
        self.tmpdir = tempfile.mkdtemp(prefix='avocado_' + __name__)

        test_path = os.path.join(self.tmpdir, 'test_cancel.py')
        self.test_cancel = script.Script(test_path,
                                         AVOCADO_TEST_CANCEL)
        self.test_cancel.save()

        test_path = os.path.join(self.tmpdir, 'test_cancel_on_setup.py')
        self.test_cancel_on_setup = script.Script(test_path,
                                                  AVOCADO_TEST_CANCEL_ON_SETUP)
        self.test_cancel_on_setup.save()

    def test_cancel(self):
        os.chdir(basedir)
        cmd_line = ['./scripts/avocado',
                    'run',
                    '--sysinfo=off',
                    '--job-results-dir',
                    '%s' % self.tmpdir,
                    '%s' % self.test_cancel,
                    '--json -']
        result = process.run(' '.join(cmd_line), ignore_status=True)
        json_results = json.loads(result.stdout)
        self.assertEqual(result.exit_status, exit_codes.AVOCADO_ALL_OK)
        self.assertEqual(json_results['cancel'], 1)

    def test_cancel_on_setup(self):
        os.chdir(basedir)
        cmd_line = ['./scripts/avocado',
                    'run',
                    '--sysinfo=off',
                    '--job-results-dir',
                    '%s' % self.tmpdir,
                    '%s' % self.test_cancel_on_setup,
                    '--json -']
        result = process.run(' '.join(cmd_line), ignore_status=True)
        json_results = json.loads(result.stdout)
        self.assertEqual(result.exit_status, exit_codes.AVOCADO_TESTS_FAIL)
        self.assertEqual(json_results['cancel'], 0)
        self.assertEqual(json_results['errors'], 1)

    def tearDown(self):
        shutil.rmtree(self.tmpdir)


if __name__ == '__main__':
    unittest.main()