import os
import shutil
import unittest
import subprocess
import shlex


def run_cmd(app, cmd):
    """Run a command and return a tuple with (stdout, stderr, exit_code)"""
    os.environ['FLASK_APP'] = app
    process = subprocess.Popen(shlex.split(cmd), stdout=subprocess.PIPE,
                               stderr=subprocess.PIPE)
    (stdout, stderr) = process.communicate()
    print('\n$ ' + cmd)
    print(stdout.decode('utf-8'))
    print(stderr.decode('utf-8'))
    return stdout, stderr, process.wait()


class TestMigrate(unittest.TestCase):
    def setUp(self):
        os.chdir(os.path.split(os.path.abspath(__file__))[0])
        try:
            os.remove('app.db')
        except OSError:
            pass
        try:
            shutil.rmtree('migrations')
        except OSError:
            pass
        try:
            shutil.rmtree('temp_folder')
        except OSError:
            pass

    def tearDown(self):
        try:
            os.remove('app.db')
        except OSError:
            pass
        try:
            shutil.rmtree('migrations')
        except OSError:
            pass
        try:
            shutil.rmtree('temp_folder')
        except OSError:
            pass

    def test_alembic_version(self):
        from flask_migrate import alembic_version
        self.assertEqual(len(alembic_version), 3)
        for v in alembic_version:
            self.assertTrue(isinstance(v, int))

    def test_migrate_upgrade(self):
        (o, e, s) = run_cmd('app.py', 'flask db init')
        self.assertTrue(s == 0)
        (o, e, s) = run_cmd('app.py', 'flask db migrate')
        self.assertTrue(s == 0)
        (o, e, s) = run_cmd('app.py', 'flask db upgrade')
        self.assertTrue(s == 0)

        from .app import db, User
        db.session.add(User(name='test'))
        db.session.commit()

    def test_custom_directory(self):
        (o, e, s) = run_cmd('app_custom_directory.py', 'flask db init')
        self.assertTrue(s == 0)
        (o, e, s) = run_cmd('app_custom_directory.py', 'flask db migrate')
        self.assertTrue(s == 0)
        (o, e, s) = run_cmd('app_custom_directory.py', 'flask db upgrade')
        self.assertTrue(s == 0)

        from .app_custom_directory import db, User
        db.session.add(User(name='test'))
        db.session.commit()

    def test_custom_directory_path(self):
        (o, e, s) = run_cmd('app_custom_directory_path.py', 'flask db init')
        self.assertTrue(s == 0)
        (o, e, s) = run_cmd('app_custom_directory_path.py', 'flask db migrate')
        self.assertTrue(s == 0)
        (o, e, s) = run_cmd('app_custom_directory_path.py', 'flask db upgrade')
        self.assertTrue(s == 0)

        from .app_custom_directory_path import db, User
        db.session.add(User(name='test'))
        db.session.commit()

    def test_compare_type(self):
        (o, e, s) = run_cmd('app_compare_type1.py', 'flask db init')
        self.assertTrue(s == 0)
        (o, e, s) = run_cmd('app_compare_type1.py', 'flask db migrate')
        self.assertTrue(s == 0)
        (o, e, s) = run_cmd('app_compare_type1.py', 'flask db upgrade')
        self.assertTrue(s == 0)
        (o, e, s) = run_cmd('app_compare_type2.py', 'flask db migrate')
        self.assertTrue(s == 0)
        self.assertTrue(b'Detected type change from VARCHAR(length=128) '
                        b'to String(length=10)' in e)
