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)