# Copyright 2019 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import argparse

from signing import config, model, standard_invoker


class TestConfig(config.CodeSignConfig):
    def __init__(self, **kwargs):
        config_args = {
            'invoker': TestInvoker.factory_with_args(),
            'identity': '[IDENTITY]',
            'installer_identity': '[INSTALLER-IDENTITY]',
        }
        config_args.update(kwargs)
        super(TestConfig, self).__init__(**config_args)

    @staticmethod
    def is_chrome_branded():
        return True

    @property
    def enable_updater(self):
        return True

    @property
    def use_static_angle(self):
        return False

    @property
    def app_product(self):
        return 'App Product'

    @property
    def product(self):
        return 'Product'

    @property
    def version(self):
        return '99.0.9999.99'

    @property
    def base_bundle_id(self):
        return 'test.signing.bundle_id'

    @property
    def provisioning_profile_basename(self):
        return 'provisiontest'

    @property
    def run_spctl_assess(self):
        return True

    @property
    def main_executable_pinned_geometry(self):
        return None


class TestConfigNonChromeBranded(TestConfig):
    @staticmethod
    def is_chrome_branded():
        return False

    @property
    def enable_updater(self):
        return False


class TestConfigInjectGetTaskAllow(TestConfig):
    @property
    def inject_get_task_allow_entitlement(self):
        return True


class TestInvoker(standard_invoker.Invoker):
    @staticmethod
    def factory_with_args(**kwargs):
        if 'notary_arg' not in kwargs:
            kwargs['notary_arg'] = []
        args = argparse.Namespace(**kwargs)
        return lambda config: TestInvoker(args, config)