# Copyright (c) 2010 The Chromium OS 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 logging
from autotest_lib.client.bin import test
from autotest_lib.client.common_lib import error

# A global variable to track how many times the test has been run.
global retry_count
retry_count = 0

class dummy_Fail(test.test):
    """The test fails by raising given exception, or succeeds at given retry."""
    version = 1

    def run_once(self, to_throw=None, retry_success_count=0):
        """Run test with argument to_throw, retry_count and retry_success_count.

        @param to_throw: Exception to throw in the test.
        @param retry_success_count: The number of times to fail before test is
                    completed successfully. 0 means the test will never complete
                    successfully with reties.

        """
        global retry_count
        retry_count += 1
        if retry_count == retry_success_count:
            return
        if to_throw:
            if to_throw == 'TestFail': logging.error('It is an error!')
            raise getattr(error, to_throw)('always fail')
        else:  # Generate a crash to test that behavior.
            self.write_perf_keyval({'perf_key': 102.7})
            self.job.record('INFO', self.tagged_testname,
                            'Received crash notification for sleep[273] sig 6')