drgn/tests/test_platform.py
Kevin Svetlitski d3c9e24115 tests: make all tests inherit from drgn's TestCase class
The majority of test cases already inherited from drgn's TestCase class.
The few outliers that inherited directly from unittest.TestCase have
been brought in line with the other tests.

Signed-off-by: Kevin Svetlitski <svetlitski@fb.com>
2022-01-11 17:28:17 -08:00

22 lines
678 B
Python

# Copyright (c) Meta Platforms, Inc. and affiliates.
# SPDX-License-Identifier: GPL-3.0-or-later
import itertools
from drgn import Architecture, Platform, PlatformFlags
from tests import TestCase
class TestPlatform(TestCase):
def test_default_flags(self):
Platform(Architecture.X86_64)
self.assertRaises(ValueError, Platform, Architecture.UNKNOWN)
def test_registers(self):
self.assertIn(
"rax",
itertools.chain.from_iterable(
reg.names for reg in Platform(Architecture.X86_64).registers
),
)
self.assertEqual(Platform(Architecture.UNKNOWN, PlatformFlags(0)).registers, ())