def test_counter_time_requirement(self):
registered_factories[Operator] = MockAmountOperator
user = PicklableMock()
counter = PicklableMock()
counter.update_time = int(time()) - 10
user.counters = {"test": counter}
requirement = CounterUpdateTimeRequirement({
"operator": {
"type": "more_or_equal",
"amount": 5
},
"key": "test"
})
self.assertTrue(requirement.check(None, user))
def test_counter_value_requirement(self):
registered_factories[Operator] = MockAmountOperator
user = PicklableMock()
counter = PicklableMock()
counter.__gt__ = Mock(return_value=True)
user.counters = {"test": counter}
requirement = CounterValueRequirement({
"operator": {
"type": "equal",
"amount": 2
},
"key": "test"
})
self.assertTrue(requirement.check(None, user))
