def test_get_vendor_2(self):
"""Test #2: Test that the vendor is returned correctly for host devices (roman chars)."""
macos.PLIST = self.plist0
self.assertEqual(macos.get_vendor(disk="disk0"), "VBOX")
def test_get_vendor_3(self):
"""Test #3: Test that the vendor is returned correctly for host devices (non-roman chars)."""
macos.PLIST = self.plist0nonroman
self.assertEqual(macos.get_vendor(disk="disk0"), "ΉΜήυΟομἝἲϾᾍᾈᾁὮᾌ")
def test_get_vendor_4(self):
"""Test #4: Test that the vendor is returned correctly for partitions, using the host disk data, when missing."""
macos.PLIST = self.plist0s1
self.assertEqual(macos.get_vendor(disk="disk0s1"), "ThereIsNone")
def test_get_vendor_1(self):
"""Test #1: Test that u"Unknown" is returned when vendor info is missing."""
macos.PLIST = self.badplist0
self.assertEqual(macos.get_vendor(disk="disk0"), "Unknown")