#!/usr/bin/env vpython3

# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""File for testing exception_recorder.py."""

import os
import sys
import tempfile
import unittest
import unittest.mock

_BUILD_UTIL_PATH = os.path.abspath(
    os.path.join(os.path.dirname(__file__), '..', '..')
)
if _BUILD_UTIL_PATH not in sys.path:
    sys.path.insert(0, _BUILD_UTIL_PATH)

from lib.proto import exception_recorder
from lib.proto.exception_occurrences_pb2 import ExceptionOccurrences


class MyClass:
    class MyException(Exception):
        pass


class ExceptionRecorderTest(unittest.TestCase):
    def myException(self):
        raise MyClass.MyException('Hello')

    def test_register(self) -> None:
        exception_recorder.clear()
        with self.assertRaises(MyClass.MyException) as cm:
            self.myException()
        record = exception_recorder.register(cm.exception)
        self.assertEqual(record.name, 'MyClass.MyException')
        self.assertTrue(len(record.stacktrace) > 0)
        self.assertTrue(record.occurred_time.ToSeconds() > 0)
        self.assertEqual(len(exception_recorder._records), 1)

    def test_to_dict(self) -> None:
        exception_recorder.clear()
        with self.assertRaises(MyClass.MyException) as cm:
            self.myException()
        record = exception_recorder.register(cm.exception)
        actual = exception_recorder.to_dict()
        self.assertIn('@type', actual)
        self.assertIn(
            ExceptionOccurrences.DESCRIPTOR.full_name, actual['@type']
        )
        actual_record = actual['datapoints'][0]
        self.assertEqual(actual_record['name'], record.name)
        self.assertEqual(actual_record['stacktrace'], record.stacktrace)

    def test_dump(self) -> None:
        exception_recorder.clear()
        with self.assertRaises(MyClass.MyException) as cm:
            self.myException()
        record = exception_recorder.register(cm.exception)
        with tempfile.TemporaryDirectory() as tmpdir:
            exception_recorder.dump(tmpdir)
            file_path = os.path.join(
                tmpdir, exception_recorder.EXCEPTION_OCCURRENCES_FILENAME
            )
            self.assertTrue(os.path.exists(file_path))


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