示例#1
0
class TestSystemModel(unittest.TestCase):


    def setUp(self):
        self.model = SystemModel()


    def test_get_system_data(self):
        active_checks = ['cpu']
        # Delete everything in the cpu collection
        cpu = self.model.mongo.get_collection('cpu')
        cpu.remove()
        

        empty_result = self.model.get_system_data(active_checks, minute_ago, now)
        
        eq_(len(empty_result), 1) # Should return only the cpu field {cpu: pymongoobject}

        # Nothing in the cpu field
        for cpu in empty_result['cpu']:
            eq_(cpu, False)

        cpu.insert({"system" : 1, "idle" : 98, "user" : 0, "time" : minute_ago})
        cpu.insert({"system" : 1, "idle" : 98, "user" : 0, "time" : two_minutes_ago})
        cpu.insert({"system" : 1, "idle" : 98, "user" : 0, "time" : now})

        result = self.model.get_system_data(active_checks, two_minutes_ago, now)
        eq_(result['cpu'].count(), 3)

        result = self.model.get_system_data(active_checks, minute_ago, now)
        eq_(result['cpu'].count(), 2)

        # Cleanup the collection
        cpu.remove()
示例#2
0
class TestSystemModel(unittest.TestCase):
    def setUp(self):
        self.model = SystemModel()

    def test_get_system_data(self):
        active_checks = ['cpu']
        # Delete everything in the cpu collection
        cpu = self.model.mongo.get_collection('cpu')
        cpu.remove()

        empty_result = self.model.get_system_data(active_checks, minute_ago,
                                                  now)

        eq_(len(empty_result),
            1)  # Should return only the cpu field {cpu: pymongoobject}

        # Nothing in the cpu field
        for cpu in empty_result['cpu']:
            eq_(cpu, False)

        cpu.insert({"system": 1, "idle": 98, "user": 0, "time": minute_ago})
        cpu.insert({
            "system": 1,
            "idle": 98,
            "user": 0,
            "time": two_minutes_ago
        })
        cpu.insert({"system": 1, "idle": 98, "user": 0, "time": now})

        result = self.model.get_system_data(active_checks, two_minutes_ago,
                                            now)
        eq_(result['cpu'].count(), 3)

        result = self.model.get_system_data(active_checks, minute_ago, now)
        eq_(result['cpu'].count(), 2)

        # Cleanup the collection
        cpu.remove()
示例#3
0
 def setUp(self):
     self.model = SystemModel()
示例#4
0
 def setUp(self):
     self.model = SystemModel()