# -*- coding: utf-8 -*-
#
# Copyright (C) 2005-2007 Christopher Lenz <cmlenz@gmx.de>
# Copyright (C) 2007-2010 Edgewall Software
# All rights reserved.
#
# This software is licensed as described in the file COPYING, which
# you should have received as part of this distribution. The terms
# are also available at http://bitten.edgewall.org/wiki/License.

from trac.core import *
from trac.web.chrome import add_script
from bitten.api import IReportChartGenerator, IReportSummarizer

__docformat__ = 'restructuredtext en'


class TestResultsChartGenerator(Component):
    implements(IReportChartGenerator)

    # IReportChartGenerator methods

    def get_supported_categories(self):
        return ['test']

    def generate_chart_data(self, req, config, category):
        assert category == 'test'

        db = self.env.get_db_cnx()
        cursor = db.cursor()
        cursor.execute("""
SELECT build.rev, build.platform, item_status.value AS status, COUNT(*) AS num
FROM bitten_build AS build
 LEFT OUTER JOIN bitten_report AS report ON (report.build=build.id)
 LEFT OUTER JOIN bitten_report_item AS item_status
  ON (item_status.report=report.id AND item_status.name='status')
WHERE build.config=%s AND report.category='test'
  AND build.rev_time >= %s AND build.rev_time <= %s
GROUP BY build.rev_time, build.rev, build.platform, item_status.value
ORDER BY build.rev_time, build.platform""", (config.name,
                                             config.min_rev_time(self.env),
                                             config.max_rev_time(self.env)))

        prev_rev = None
        prev_platform, platform_total = None, 0
        tests = []
        for rev, platform, status, num in cursor:
            if rev != prev_rev:
                tests.append([rev, 0, 0, 0])
                prev_rev = rev
                platform_total = 0
            if platform != prev_platform:
                prev_platform = platform
                platform_total = 0

            platform_total += num
            tests[-1][1] = max(platform_total, tests[-1][1])
            if status == 'success':
                pass
            elif status == 'ignore':
                tests[-1][3] = max(num, tests[-1][3])
            else:
                tests[-1][2] = max(num, tests[-1][2])

        data = {'title': 'Unit Tests',
                'data': [
                    {'label': 'Total', 'data': [[item[0], item[1]] for item in tests], 'lines': {'fill': True}},
                    {'label': 'Failures', 'data': [[item[0], item[2]] for item in tests]},
                    {'label': 'Ignored', 'data': [[item[0], item[3]] for item in tests]},
                ],
                'options': {
                    'legend': {'position': 'sw', 'backgroundOpacity': 0.7},
                    'xaxis': {'tickDecimals': 0},
                    'yaxis': {'tickDecimals': 0},
                },
               }

        return 'json.txt', {"json": data}


class TestResultsSummarizer(Component):
    implements(IReportSummarizer)

    # IReportSummarizer methods

    def get_supported_categories(self):
        return ['test']

    def render_summary(self, req, config, build, step, category):
        assert category == 'test'

        db = self.env.get_db_cnx()
        cursor = db.cursor()
        cursor.execute("""
SELECT item_fixture.value AS fixture, item_file.value AS file,
       COUNT(item_success.value) AS num_success,
       COUNT(item_ignore.value) AS num_ignore,
       COUNT(item_failure.value) AS num_failure,
       COUNT(item_error.value) AS num_error
FROM bitten_report AS report
 LEFT OUTER JOIN bitten_report_item AS item_fixture
  ON (item_fixture.report=report.id AND item_fixture.name='fixture')
 LEFT OUTER JOIN bitten_report_item AS item_file
  ON (item_file.report=report.id AND item_file.item=item_fixture.item AND
      item_file.name='file')
 LEFT OUTER JOIN bitten_report_item AS item_success
  ON (item_success.report=report.id AND item_success.item=item_fixture.item AND
      item_success.name='status' AND item_success.value='success')
 LEFT OUTER JOIN bitten_report_item AS item_ignore
  ON (item_ignore.report=report.id AND item_ignore.item=item_fixture.item AND
      item_ignore.name='status' AND item_ignore.value='ignore')
 LEFT OUTER JOIN bitten_report_item AS item_failure
  ON (item_failure.report=report.id AND item_failure.item=item_fixture.item AND
      item_failure.name='status' AND item_failure.value='failure')
 LEFT OUTER JOIN bitten_report_item AS item_error
  ON (item_error.report=report.id AND item_error.item=item_fixture.item AND
      item_error.name='status' AND item_error.value='error')
WHERE category='test' AND build=%s AND step=%s
GROUP BY file, fixture
ORDER BY fixture""", (build.id, step.name))

        fixtures = []
        total_success, total_ignore, total_failure, total_error = 0, 0, 0, 0
        for fixture, file, num_success, num_ignore, num_failure, num_error in cursor:
            fixtures.append({'name': fixture, 
                             'num_success': num_success,
                             'num_ignore': num_ignore,
                             'num_error': num_error,
                             'num_failure': num_failure})
            total_success += num_success
            total_ignore += num_ignore
            total_failure += num_failure
            total_error += num_error
            if file:
                fixtures[-1]['href'] = req.href.browser(config.path, file)

        # For each fixture, get a list of tests that don't succeed
        for fixture in fixtures:
            cursor.execute("""
SELECT item_status.value AS status, item_name.value AS name,
       item_traceback.value AS traceback
FROM bitten_report
 LEFT OUTER JOIN bitten_report_item AS item_fixture
  ON (item_fixture.report=bitten_report.id AND
      item_fixture.name='fixture')
 LEFT OUTER JOIN bitten_report_item AS item_status
  ON (item_status.report=bitten_report.id AND
      item_status.item=item_fixture.item AND
      item_status.name='status')
 LEFT OUTER JOIN bitten_report_item AS item_name
  ON (item_name.report=bitten_report.id AND
      item_name.item=item_fixture.item AND
      item_name.name='name')
 LEFT OUTER JOIN bitten_report_item AS item_traceback
  ON (item_traceback.report=bitten_report.id AND
      item_traceback.item=item_fixture.item AND
      item_traceback.name='traceback')
WHERE category='test' AND build=%s AND step=%s AND item_status.value<>'success' AND
      item_fixture.value=%s""", (build.id, step.name, fixture['name']))

            failures = []
            for status, name, traceback in cursor:
                # use the fixture name if a name isn't supplied for the
                # individual test
                if not name:
                    name = fixture['name']
                failures.append({'status': status,
                                 'name': name,
                                 'traceback': traceback})
            if failures:
                fixture['failures'] = failures

        data = {'fixtures': fixtures,
                'totals': {'success': total_success, 
                           'ignore': total_ignore,
                           'failure': total_failure,
                           'error': total_error}
               }
        return 'bitten_summary_tests.html', data