# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

from __future__ import unicode_literals

import imp
import os
import sys
import unittest

from StringIO import StringIO

import mach.main


class TestErrorOutput(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        common_path = os.path.join(os.path.dirname(__file__), 'common.py')
        imp.load_source('mach.commands.error_output_test', common_path)

    def _run_mach(self, args):
        m = mach.main.Mach(os.getcwd())

        stdout = StringIO()
        stderr = StringIO()
        stdout.encoding = 'UTF-8'
        stderr.encoding = 'UTF-8'

        result = m.run(args, stdout=stdout, stderr=stderr)

        return (result, stdout.getvalue(), stderr.getvalue())

    def test_command_error(self):
        result, stdout, stderr = self._run_mach(['throw', '--message',
            'Command Error'])

        self.assertEqual(result, 1)

        self.assertIn(mach.main.COMMAND_ERROR, stdout)

    def test_invoked_error(self):
        result, stdout, stderr = self._run_mach(['throw_deep', '--message',
            'Deep stack'])

        self.assertEqual(result, 1)

        self.assertIn(mach.main.MODULE_ERROR, stdout)