Python ScannerQueryRule.change_state_to Examples

Example #1

File: test_models.py Project: trpik4/addons-server

def test_query_rule_change_state_to_invalid(current_state, target_state):
    rule = ScannerQueryRule(name='some_rule', scanner=YARA)
    rule.state = current_state
    with pytest.raises(ImproperScannerQueryRuleStateError):
        rule.change_state_to(target_state)
    # Manually changing state doesn't affect 'completed' property, and since
    # changing through change_state_to() failed before it should always be
    # None in this test.
    assert rule.completed is None

Example #2

File: test_models.py Project: trpik4/addons-server

def test_query_rule_change_state_to_valid(current_state, target_state):
    rule = ScannerQueryRule(name='some_rule', scanner=YARA)
    rule.state = current_state
    with freeze_time('2020-04-08 15:16:23.42') as frozen_time:
        rule.change_state_to(target_state)
    if target_state == COMPLETED:
        assert rule.completed == frozen_time()
    else:
        assert rule.completed is None

Example #3

File: test_models.py Project: CSCD01/addons-server-team02

def test_query_rule_change_state_to_invalid(current_state, target_state):
    rule = ScannerQueryRule(name='some_rule', scanner=YARA)
    rule.state = current_state
    with pytest.raises(ImproperScannerQueryRuleStateError):
        rule.change_state_to(target_state)

Example #4

File: test_models.py Project: CSCD01/addons-server-team02

def test_query_rule_change_state_to_valid(current_state, target_state):
    rule = ScannerQueryRule(name='some_rule', scanner=YARA)
    rule.state = current_state
    rule.change_state_to(target_state)