# Copyright 2013 Rackspace Australia
#
# 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 json
import os
import testtools

from turbo_hipster.task_plugins.real_db_upgrade import handle_results

TESTS_DIR = os.path.join(os.path.dirname(__file__))


class TestHandleResults(testtools.TestCase):
    def test_line_to_time(self):
        test_line = '2013-11-22 21:42:45,908 [output] 141 -> 142...  '
        logfile = os.path.join(TESTS_DIR, 'assets/logcontent')
        lp = handle_results.LogParser(logfile, None)
        result = lp.line_to_time(test_line)
        self.assertEqual(result, 1385156565)

    def test_check_migration(self):
        with open(os.path.join(TESTS_DIR,
                               'datasets/some_dataset_example/config.json'),
                  'r') as config_stream:
            dataset_config = json.load(config_stream)
        duration = 120

        result = handle_results.check_migration({'to': '151',
                                                 'from': '150'},
                                                'maximum_migration_times',
                                                duration, dataset_config)
        self.assertFalse(result)

        result = handle_results.check_migration({'to': '152',
                                                 'from': '151'},
                                                'maximum_migration_times',
                                                duration, dataset_config)
        self.assertTrue(result)

    def test_check_log_for_errors(self):
        logfile = os.path.join(TESTS_DIR,
                               'assets/20131007_devstack_export.log')

        def fake_find_schemas_230():
            return [230]

        lp = handle_results.LogParser(logfile, '/tmp/foo')
        lp.find_schemas = fake_find_schemas_230
        lp.process_log()
        self.assertEqual(['FAILURE - Final schema version does not match '
                         'expectation'], lp.errors)
        self.assertEqual([], lp.warnings)

        def fake_find_schemas_228():
            return [228]

        lp = handle_results.LogParser(logfile, '/tmp/foo')
        lp.find_schemas = fake_find_schemas_228
        lp.process_log()
        self.assertEqual([], lp.errors)
        self.assertEqual([], lp.warnings)

    def test_parse_log(self):
        # This is a regression test for a log which didn't used to parse.
        logfile = os.path.join(TESTS_DIR, 'assets/logcontent')
        lp = handle_results.LogParser(logfile, None)
        lp.process_log()

        self.assertEqual([], lp.errors)
        self.assertEqual([], lp.warnings)

        migrations = []
        for migration in lp.migrations:
            migrations.append(migration['to'])

        for migration in range(134, 229):
            self.assertTrue(migration in migrations,
                            'Migration %d missing from %s'
                            % (migration, migrations))

    def test_innodb_stats(self):
        logfile = os.path.join(TESTS_DIR, 'assets/user_001.log')

        def fake_find_schemas_228():
            return [228]

        lp = handle_results.LogParser(logfile, '/tmp/foo')
        lp.find_schemas = fake_find_schemas_228
        lp.process_log()

        migration = lp.migrations[0]
        self.assertTrue('stats' in migration)
        self.assertTrue('Innodb_rows_read' in migration['stats'])
        self.assertEqual(5, migration['stats']['Innodb_rows_read'])