mirror of
https://github.com/chipsec/chipsec
synced 2026-06-08 13:31:00 +00:00
Nathaniel Mitchell
42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
# CHIPSEC: Platform Security Assessment Framework
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; Version 2.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
#
|
|
|
|
import unittest
|
|
|
|
from unittest.mock import MagicMock
|
|
from chipsec.hal.common.pci import Pci
|
|
from chipsec.library.register import ObjList, Register
|
|
from chipsec.cfg.parsers.ip.pci_device import PCIConfig
|
|
from chipsec.cfg.parsers.registers.pci import PCIRegisters
|
|
|
|
class TestPCI(unittest.TestCase):
|
|
def test_(self):
|
|
mock_cs = MagicMock()
|
|
vid = 0x8086
|
|
did = 0x1234
|
|
rid = 0xa
|
|
mock_cs.helper.read_pci_reg.return_value = (did << 16) | vid
|
|
pci_data = {'bus': 0, 'dev': 0, 'fun': 0, 'vid': vid, 'did': did,'rid': rid}
|
|
pcilist = ObjList([PCIConfig(pci_data)])
|
|
test_acpi = Pci(mock_cs)
|
|
new_data = test_acpi.get_viddidrid_from_device_list(pcilist)
|
|
self.assertEqual(len(new_data), 1)
|
|
new_vid, new_did, new_rid, _ = new_data[0]
|
|
self.assertEqual(new_vid, vid)
|
|
self.assertEqual(new_did, did)
|
|
self.assertEqual(new_rid, rid)
|