# -*- coding: utf-8 -*-

# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

import mock
import os
import shutil
import tempfile

from dlrn.config import ConfigOptions
from dlrn.drivers.coprdriver import CoprBuildDriver
from dlrn.tests import base
from six.moves import configparser


@mock.patch('sh.restorecon', create=True)
@mock.patch('sh.env', create=True)
class TestDriverCopr(base.TestCase):
    def setUp(self):
        super(TestDriverCopr, self).setUp()
        config = configparser.RawConfigParser()
        config.read("projects.ini")
        config.set("DEFAULT", "build_driver",
                   "dlrn.drivers.coprdriver.CoprBuildDriver")
        config.set('coprbuild_driver', 'coprid', 'foo/repo')
        self.config = ConfigOptions(config)
        self.temp_dir = tempfile.mkdtemp()
        # Create fake src.rpm
        with open('%s/pkg.src.rpm' % self.temp_dir, 'a') as fp:
            fp.write('')
        # Create fake build and download logs
        with open("%s/coprbuild.log" % self.temp_dir, 'a') as fp:
            fp.write("Created builds: 1234")
        with open('%s/coprdownload.log' % self.temp_dir, 'a') as fp:
            fp.write('')
        # Create fake download file structure
        os.mkdir(os.path.join(self.temp_dir, '1234'))
        target_chroot = os.path.join(
            self.temp_dir, '1234', 'fedora-rawhide-i386')
        os.mkdir(target_chroot)
        with open('%s/state.log.gz' % target_chroot, 'a') as fp:
            fp.write('')
        with open('%s/pkg.rpm' % target_chroot, 'a') as fp:
            fp.write('')

    def tearDown(self):
        super(TestDriverCopr, self).tearDown()
        shutil.rmtree(self.temp_dir)

    def test_build_package(self, env_mock, rc_mock):
        driver = CoprBuildDriver(cfg_options=self.config)
        driver.build_package(output_directory=self.temp_dir)

        expected = [mock.call(['copr', 'build',
                               self.config.coprid,
                               '%s/pkg.src.rpm' %
                               self.temp_dir],
                              _err=driver._process_copr_output,
                              _out=driver._process_copr_output,
                              _cwd=self.temp_dir,
                              _env=mock.ANY),
                    mock.call(['copr', 'download-build', '-d',
                               '%s/1234' % self.temp_dir, '1234'],
                              _err=driver._process_copr_output,
                              _out=driver._process_copr_output,
                              _cwd=self.temp_dir,
                              _env=mock.ANY)]
        # 1- copr build (handled by env_mock)
        # 2- copr download-build (handled by env_mock)
        # 3- restorecon (handled by rc_mock)
        self.assertEqual(env_mock.call_count, 2)
        self.assertEqual(rc_mock.call_count, 1)
        self.assertEqual(env_mock.call_args_list, expected)

        # Make sure output_dir is as expected
        content = os.listdir(self.temp_dir)
        self.assertIn('coprdownload.log', content)
        self.assertIn('state.log.gz', content)
        self.assertIn('pkg.rpm', content)
        self.assertIn('pkg.src.rpm', content)
        self.assertNotIn('1234', content)

    def test_driver_config(self, env_mock, rc_mock):
        self.assertEqual(self.config.coprid, 'foo/repo')