#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import shutil
import sys
import types

from scripttest import TestFileEnvironment

from test.fixture.pathed import *


class Shell(Pathed):
    """Base class for command line tests"""

    def setUp(self):
        super(Shell, self).setUp()
        self.env = TestFileEnvironment(os.path.join(self.temp_usable_dir, 'env'))

    def run_version(self, repos_path):
        result = self.env.run('migrate version %s' % repos_path)
        return int(result.stdout.strip())

    def run_db_version(self, url, repos_path):
        result = self.env.run('migrate db_version %s %s' % (url, repos_path))
        return int(result.stdout.strip())