#!/usr/bin/env python
# Copyright (c) 2018 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import json
import os
import sys
import unittest

ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, ROOT_DIR)

import ninjalog_uploader


class NinjalogUploaderTest(unittest.TestCase):
    def test_parse_gn_args(self):
        self.assertEqual(ninjalog_uploader.ParseGNArgs(json.dumps([])), {})

        # Extract current configs from GN's output json.
        self.assertEqual(ninjalog_uploader.ParseGNArgs(json.dumps([
            {
                'current': {'value': 'true'},
                'default': {'value': 'false'},
                'name': 'is_component_build'
            },
            {
                'default': {'value': '"x64"'},
                'name': 'host_cpu'
            },
        ])), {
            'is_component_build': 'true',
            'host_cpu': '"x64"',
        })

        self.assertEqual(ninjalog_uploader.ParseGNArgs(json.dumps([
            {
                'current': {'value': 'true'},
                'default': {'value': 'false'},
                'name': 'is_component_build'
            },
            {
                'current': {'value': 'false'},
                'default': {'value': 'false'},
                'name': 'use_goma'
            },
        ])), {'is_component_build': 'true',
              'use_goma': 'false'})

    def test_get_ninjalog(self):
        # No args => default to cwd.
        self.assertEqual(ninjalog_uploader.GetNinjalog(['ninja']),
                         './.ninja_log')

        # Specified by -C case.
        self.assertEqual(
            ninjalog_uploader.GetNinjalog(['ninja', '-C', 'out/Release']),
            'out/Release/.ninja_log')
        self.assertEqual(
            ninjalog_uploader.GetNinjalog(['ninja', '-Cout/Release']),
            'out/Release/.ninja_log')

        # Invalid -C flag case.
        self.assertEqual(ninjalog_uploader.GetNinjalog(['ninja', '-C']),
                         './.ninja_log')

        # Multiple target directories => use the last directory.
        self.assertEqual(ninjalog_uploader.GetNinjalog(
            ['ninja', '-C', 'out/Release', '-C', 'out/Debug']),
            'out/Debug/.ninja_log')

    def test_get_build_target_from_command_line(self):
        self.assertEqual(ninjalog_uploader.GetBuildTargetFromCommandLine(
            ['ninja', 'chrome']), ['chrome'])

        self.assertEqual(ninjalog_uploader.GetBuildTargetFromCommandLine(
            ['ninja']), [])

        self.assertEqual(ninjalog_uploader.GetBuildTargetFromCommandLine(
            ['ninja', '-j', '1000', 'chrome']), ['chrome'])

        self.assertEqual(ninjalog_uploader.GetBuildTargetFromCommandLine(
            ['ninja', 'chrome', '-j', '1000']), ['chrome'])

        self.assertEqual(ninjalog_uploader.GetBuildTargetFromCommandLine(
            ['ninja', '-C', 'chrome']), [])

        self.assertEqual(ninjalog_uploader.GetBuildTargetFromCommandLine(
            ['ninja', '-Cout/Release', 'chrome']), ['chrome'])

        self.assertEqual(ninjalog_uploader.GetBuildTargetFromCommandLine(
            ['ninja', '-C', 'out/Release', 'chrome', 'all']), ['chrome', 'all'])

    def test_get_j_flag(self):
        self.assertEqual(ninjalog_uploader.GetJflag(
            ['ninja']), None)

        self.assertEqual(ninjalog_uploader.GetJflag(
            ['ninja','-j', '1000']), 1000)

        self.assertEqual(ninjalog_uploader.GetJflag(
            ['ninja','-j', '1000a']), None)

        self.assertEqual(ninjalog_uploader.GetJflag(
            ['ninja','-j', 'a']), None)

        self.assertEqual(ninjalog_uploader.GetJflag(
            ['ninja','-j1000']), 1000)

        self.assertEqual(ninjalog_uploader.GetJflag(
            ['ninja','-ja']), None)

        self.assertEqual(ninjalog_uploader.GetJflag(
            ['ninja','-j']), None)


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