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))