# copyright 2003-2014 LOGILAB S.A. (Paris, FRANCE), all rights reserved.

# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr

#

# This file is part of CubicWeb.

#

# CubicWeb is free software: you can redistribute it and/or modify it under the

# terms of the GNU Lesser General Public License as published by the Free

# Software Foundation, either version 2.1 of the License, or (at your option)

# any later version.

#

# CubicWeb is distributed in the hope that it will be useful, but WITHOUT

# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more

# details.

#

# You should have received a copy of the GNU Lesser General Public License along

# with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.

import sys

import os

from os.path import join

from io import StringIO, BytesIO

from six import PY2

from logilab.common.testlib import TestCase, unittest_main

from cubicweb.cwconfig import CubicWebConfiguration

from cubicweb.devtools.testlib import CubicWebTC

from cubicweb.server.migractions import ServerMigrationHelper

CubicWebConfiguration.load_cwctl_plugins() # XXX necessary?

class CubicWebCtlTC(TestCase):

    def setUp(self):

        self.stream = BytesIO() if PY2 else StringIO()

        sys.stdout = self.stream

    def tearDown(self):

        sys.stdout = sys.__stdout__

    def test_list(self):

        from cubicweb.cwctl import ListCommand

        ListCommand(None).run([])

class CubicWebShellTC(CubicWebTC):

    def test_process_script_args_context(self):

        repo = self.repo

        with self.admin_access.repo_cnx() as cnx:

            mih = ServerMigrationHelper(None, repo=repo, cnx=cnx,

                                        interactive=False,

                                        # hack so it don't try to load fs schema

                                        schema=1)

            scripts = {'script1.py': list(),

                       'script2.py': ['-v'],

                       'script3.py': ['-vd', '-f', 'FILE.TXT'],

                      }

            mih.cmd_process_script(join(self.datadir, 'scripts', 'script1.py'),

                                   funcname=None)

            for script, args in scripts.items():

                scriptname = os.path.join(self.datadir, 'scripts', script)

                self.assertTrue(os.path.exists(scriptname))

                mih.cmd_process_script(scriptname, None, scriptargs=args)

if __name__ == '__main__':

    unittest_main()