"""Tests for the bot_setup module."""
import os
import shutil
import subprocess
import tempfile
import unittest
from catatester.devicemap import bot_setup

try:
  from unittest import mock
except ImportError:
  import mock

_STARTUP_SCRIPTS_PATH = 'startup_scripts'


class BotSetupTest(unittest.TestCase):
  """Tests for the methods in bot_setup module."""

  def setUp(self):
    self._swarm_bots_home = tempfile.mkdtemp()
    self.addCleanup(self.cleanUp)
    self.addCleanup(mock.patch.stopall)
    self._mock_subprocess_call = mock.patch(
        'subprocess.call', autospec=True, spec_set=True).start()
    self._mock_copy_file = mock.patch(
        'shutil.copy', autospec=True, spec_set=True).start()
    self._mock_create_file = mock.patch(
        'catatester.devicemap.bot_setup._CreateFile',
        autospec=True, spec_set=True).start()
    self._mock_update_file_mode = mock.patch(
        'catatester.devicemap.bot_setup._UpdateFilePermissions',
        autospec=True, spec_set=True).start()

  def cleanUp(self):
    shutil.rmtree(self._swarm_bots_home)

  def testSetupBotsCreatesNewDirCopiesFileAndStartsService(self):
    """Validates setup_bot method can successfully start bot service."""
    self._mock_subprocess_call.side_effect = [0, 1, 0]
    bot_num = 2
    bot_setup.SetupBot(
        bot_num=bot_num,
        user='mock_user',
        group='mock_group',
        swarm_bots_home=self._swarm_bots_home,
        swarm_bot_prefix='mock_swarm_bot_prefix',
        swarm_zip_path='mock_swarm_zip_path',
        gsutil_dir='mock_gsutil_dir',
        gsutil_boto='mock_gsutil_boto',
        fragglerock_key='mock_fragglerock_key',
        eurtest_binaries='mock_eurtest_binaries',
        venv='mock_venv',
        devicemap_dir='mock_devicemap_dir',
        depot_tools_dir='mock_depot_tools'
    )
    self.assertEqual(
        1, len(os.listdir(self._swarm_bots_home)), 'Bot directory not created.')
    self.assertTrue(
        os.path.isdir(os.path.join(self._swarm_bots_home, str(bot_num))),
        'Bot directory not created.')
    expected_systemd_file_path = (
        '/etc/systemd/system/mock_swarm_bot_prefix%s.service' %  bot_num)
    expected_sh_path = (
        os.path.join(
            self._swarm_bots_home, _STARTUP_SCRIPTS_PATH,
            'mock_swarm_bot_prefix%s_startup.sh' % bot_num))
    self._mock_create_file.assert_has_calls([
        mock.call(
            file_path=expected_sh_path, content=mock.ANY,
            owner='mock_user', group='mock_group', mode=508),
        mock.call(
            file_path=expected_systemd_file_path, content=mock.ANY,
            owner='root', group='root', mode=420)
    ])
    self._mock_subprocess_call.assert_has_calls([
        mock.call(['systemctl', 'daemon-reload']),
        mock.call([
            'systemctl', 'is-active', '--quiet',
            'mock_swarm_bot_prefix%s' % bot_num
        ]),
        mock.call([
            'systemctl', 'start', 'mock_swarm_bot_prefix%s' % bot_num])
    ])
    self._mock_update_file_mode.assert_has_calls([
        mock.call(mock.ANY, 'mock_user', 'mock_group', 508),
        mock.call('mock_swarm_zip_path', 'mock_user', 'mock_group', 508)
    ])

  def testBotServiceNotRestartedIfItsRunning(self):
    """Validates setup_bot doesn't restart an already running bot service."""
    self._mock_subprocess_call.side_effect = [0, 0]
    bot_setup.SetupBot(
        bot_num=1,
        user='mock_user',
        group='mock_group',
        swarm_bots_home=self._swarm_bots_home,
        swarm_bot_prefix='mock_swarm_bot_prefix',
        swarm_zip_path='mock_swarm_zip_path',
        gsutil_dir='mock_gsutil_dir',
        gsutil_boto='mock_gsutil_boto',
        fragglerock_key='mock_fragglerock_key',
        eurtest_binaries='mock_eurtest_binaries',
        venv='mock_venv',
        devicemap_dir='mock_devicemap_dir',
        depot_tools_dir='mock_depot_tools')
    self._mock_subprocess_call.assert_called_with([
        'systemctl', 'is-active', '--quiet', 'mock_swarm_bot_prefix1'])

if __name__ == '__main__':
  unittest.main()