class CpuProfilerTest(unittest.TestCase):
def setUp(self):
self.server = 'www.rmlink.cn'
self.profiler = CPUProfiler(self.server)
def test_getCpuCount(self):
returnedData = self.profiler.getProcessorCount()
self.assertIn('count', returnedData, "'count' is not a key in the returned object")
count = returnedData['count']
self.assertTrue(isinstance(count, int), "CPU processor count '%s' is not an integer" % count)
def getCpuCount(server):
profiler = CPUProfiler(server)
data = profiler.getProcessorCount()
return jsonify(data)
