1
0
mirror of https://github.com/kellyjonbrazil/jc.git synced 2025-06-17 00:07:37 +02:00
Files
jc/tests/test_os_prober.py
Kelly Brazil 8d03055b34 add tests
2022-11-07 16:41:13 -08:00

48 lines
1.3 KiB
Python

import os
import unittest
import json
from typing import Dict
from jc.parsers.os_prober import parse
THIS_DIR = os.path.dirname(os.path.abspath(__file__))
class MyTests(unittest.TestCase):
def test_os_prober_nodata(self):
"""
Test 'os_prober' with no data
"""
self.assertEqual(parse('', quiet=True), {})
def test_os_prober_1(self):
"""
Test 'os_prober' 1
"""
self.assertEqual(parse(
'/dev/sda1:Windows 7 (loader):Windows:chain', quiet=True),
{"partition":"/dev/sda1","name":"Windows 7 (loader)","short_name":"Windows","type":"chain"}
)
def test_os_prober_2(self):
"""
Test 'os_prober' 2
"""
self.assertEqual(parse(
'/dev/sda1:Windows 10:Windows:chain', quiet=True),
{"partition":"/dev/sda1","name":"Windows 10","short_name":"Windows","type":"chain"}
)
def test_os_prober_3(self):
"""
Test 'os_prober' 3
"""
self.assertEqual(parse(
'/dev/sda1@/efi/Microsoft/Boot/bootmgfw.efi:Windows Boot Manager:Windows:efi', quiet=True),
{"partition":"/dev/sda1@/efi/Microsoft/Boot/bootmgfw.efi","name":"Windows Boot Manager","short_name":"Windows","type":"efi"}
)
if __name__ == '__main__':
unittest.main()