# -*- coding: utf-8 -*-
from __future__ import unicode_literals
import os
import shutil
from django.test import SimpleTestCase

from ..console_scripts.admin import DJANGO_MAIN_MODULE, AdminCommand

from .utils import mkdtemp


class AdminTestCase(SimpleTestCase):
    def setUp(self):
        super(AdminTestCase, self).setUp()
        self.tmp_dir = mkdtemp(prefix='AdminTestCase')

    def tearDown(self):
        super(AdminTestCase, self).tearDown()
        if os.path.isdir(self.tmp_dir):
            shutil.rmtree(self.tmp_dir)

    def test_setup(self):
        path = self.tmp_dir
        cmd = AdminCommand()
        argv = ['setup', path]
        exitval = cmd(argv)
        self.assertEqual(exitval, os.EX_OK)
        self.assertTrue(os.path.isfile(os.path.join(path, 'manage.py')))
        self.assertTrue(os.path.isfile(os.path.join(path, DJANGO_MAIN_MODULE, 'settings.py')))
        self.assertTrue(os.path.isdir(os.path.join(path, 'var', 'log')))
        self.assertTrue(os.path.isdir(os.path.join(path, 'var', 'www', 'static')))