1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
#!/usr/bin/python
import os
import sys
sys.path.insert(0,"../")
import apt
import unittest
import shutil
import subprocess
from DistUpgrade.DistUpgradeQuirks import DistUpgradeQuirks
class MockController(object):
def __init__(self):
self._view = None
class MockConfig(object):
pass
class testQuirks(unittest.TestCase):
def testFglrx(self):
mock_lspci_good = set(['1002:7145'])
mock_lspci_bad = set(['8086:ac56'])
q = DistUpgradeQuirks(MockController(), MockConfig)
self.assert_(q._supportInModaliases("fglrx",
"../DistUpgrade/modaliases/",
mock_lspci_good) == True)
self.assert_(q._supportInModaliases("fglrx",
"../DistUpgrade/modaliases/",
mock_lspci_bad) == False)
def test_cpuHasSSESupport(self):
q = DistUpgradeQuirks(MockController(), MockConfig)
self.assert_(q._cpuHasSSESupport(cpuinfo="test-data/cpuinfo-with-sse") == True)
self.assert_(q._cpuHasSSESupport(cpuinfo="test-data/cpuinfo-without-sse") == False)
if __name__ == "__main__":
unittest.main()
|