#!/usr/bin/env python3
#
# Copyright 2022 The Fuchsia Authors
#
# 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 unittest

from antlion import error


class ActsErrorTest(unittest.TestCase):

    def test_assert_key_pulled_from_acts_error_code(self):
        e = error.ActsError()
        self.assertEqual(e.error_code, 100)

    def test_assert_description_pulled_from_docstring(self):
        e = error.ActsError()
        self.assertEqual(e.error_doc, 'Base Acts Error')

    def test_error_without_args(self):
        e = error.ActsError()
        self.assertEqual(e.details, '')

    def test_error_with_args(self):
        args = ('hello', )
        e = error.ActsError(*args)
        self.assertEqual(e.details, 'hello')

    def test_error_with_kwargs(self):
        e = error.ActsError(key='value')
        self.assertIn(('key', 'value'), e.extras.items())


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