# (c) 2013, AnsibleWorks
#
# This file is part of Ansible Commander
#
# Ansible Commander is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible Commander 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible Commander.  If not, see <http://www.gnu.org/licenses/>.


import os
import tempfile
from django.conf import settings
from django.test.utils import override_settings
from lib.main.models import *
from lib.main.tests.base import BaseTransactionTest

TEST_PLAYBOOK = '''- hosts: test-group
  gather_facts: False
  tasks:
  - name: should pass
    command: test 1 = 1
'''

@override_settings(CELERY_ALWAYS_EAGER=True,
                   CELERY_EAGER_PROPAGATES_EXCEPTIONS=True)
class BaseCeleryTest(BaseTransactionTest):
    '''
    Base class for celery task tests.
    '''

@override_settings(ANSIBLE_TRANSPORT='local')
class RunLaunchJobTest(BaseCeleryTest):
    '''
    Test cases for run_launch_job celery task.
    '''

    def setUp(self):
        super(RunLaunchJobTest, self).setUp()
        self.setup_users()
        self.organization = self.make_organizations(self.super_django_user, 1)[0]
        self.project = self.make_projects(self.normal_django_user, 1)[0]
        handle, self.test_playbook = tempfile.mkstemp(suffix='.yml', prefix='playbook-')
        test_playbook_file = os.fdopen(handle, 'w')
        test_playbook_file.write(TEST_PLAYBOOK)
        test_playbook_file.close()
        self.project.default_playbook = self.test_playbook
        self.project.save()
        self.organization.projects.add(self.project)
        self.inventory = Inventory.objects.create(name='test-inventory',
                                                  description='description for test-inventory',
                                                  organization=self.organization)
        self.host = self.inventory.hosts.create(name='host.example.com',
                                                inventory=self.inventory)
        self.group = self.inventory.groups.create(name='test-group',
                                                  inventory=self.inventory)
        self.group.hosts.add(self.host)
        self.launch_job = LaunchJob.objects.create(name='test-launch-job',
                                                   inventory=self.inventory,
                                                   project=self.project)
        # Pass test database name in environment for use by the inventory script.
        os.environ['ACOM_TEST_DATABASE_NAME'] = settings.DATABASES['default']['NAME']

    def tearDown(self):
        super(RunLaunchJobTest, self).tearDown()
        os.environ.pop('ACOM_TEST_DATABASE_NAME', None)
        os.remove(self.test_playbook)

    def test_run_launch_job(self):
        launch_job_status = self.launch_job.start()
        self.assertEqual(launch_job_status.status, 'pending')
        launch_job_status = LaunchJobStatus.objects.get(pk=launch_job_status.pk)
        #print 'stdout:', launch_job_status.result_stdout
        #print 'stderr:', launch_job_status.result_stderr
        self.assertEqual(launch_job_status.status, 'successful')
        self.assertTrue(launch_job_status.result_stdout)
        launch_job_status_events = launch_job_status.launch_job_status_events.all()
        #for ev in launch_job_status_events:
        #    print ev.event, ev.event_data
        self.assertEqual(launch_job_status_events.filter(event='playbook_on_start').count(), 1)
        self.assertEqual(launch_job_status_events.filter(event='playbook_on_play_start').count(), 1)
        self.assertEqual(launch_job_status_events.filter(event='playbook_on_task_start').count(), 1)
        self.assertEqual(launch_job_status_events.filter(event='playbook_on_stats').count(), 1)

    # FIXME: Test with a task that fails.