mirror of
git://holbrook.no/eth-monitor.git
synced 2024-11-27 13:46:47 +01:00
91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
|
import logging
|
||
|
|
import unittest
|
||
|
|
import os
|
||
|
|
|
||
|
|
# local imports
|
||
|
|
from eth_monitor.rules import *
|
||
|
|
|
||
|
|
logging.basicConfig(level=logging.DEBUG)
|
||
|
|
logg = logging.getLogger()
|
||
|
|
|
||
|
|
|
||
|
|
class TestRule(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.alice = os.urandom(20).hex()
|
||
|
|
self.bob = os.urandom(20).hex()
|
||
|
|
self.carol = os.urandom(20).hex()
|
||
|
|
self.dave = os.urandom(20).hex()
|
||
|
|
self.x = os.urandom(20).hex()
|
||
|
|
self.y = os.urandom(20).hex()
|
||
|
|
self.hsh = os.urandom(32).hex()
|
||
|
|
|
||
|
|
|
||
|
|
def test_greedy_includes(self):
|
||
|
|
data = b''
|
||
|
|
outs = [self.alice]
|
||
|
|
ins = [self.carol]
|
||
|
|
execs = []
|
||
|
|
rule = RuleSimple(outs, ins, execs, match_all=True)
|
||
|
|
c = AddressRules()
|
||
|
|
c.include(rule)
|
||
|
|
r = c.apply_rules_addresses(self.alice, self.bob, data, self.hsh)
|
||
|
|
self.assertFalse(r)
|
||
|
|
r = c.apply_rules_addresses(self.bob, self.alice, data, self.hsh)
|
||
|
|
self.assertFalse(r)
|
||
|
|
r = c.apply_rules_addresses(self.bob, self.carol, data, self.hsh)
|
||
|
|
self.assertFalse(r)
|
||
|
|
r = c.apply_rules_addresses(self.alice, self.carol, data, self.hsh)
|
||
|
|
self.assertTrue(r)
|
||
|
|
|
||
|
|
rule = RuleSimple(outs, ins, execs)
|
||
|
|
c = AddressRules(match_all=True)
|
||
|
|
c.include(rule)
|
||
|
|
r = c.apply_rules_addresses(self.alice, self.bob, data, self.hsh)
|
||
|
|
self.assertTrue(r)
|
||
|
|
r = c.apply_rules_addresses(self.bob, self.alice, data, self.hsh)
|
||
|
|
self.assertFalse(r)
|
||
|
|
r = c.apply_rules_addresses(self.bob, self.carol, data, self.hsh)
|
||
|
|
self.assertTrue(r)
|
||
|
|
r = c.apply_rules_addresses(self.alice, self.carol, data, self.hsh)
|
||
|
|
self.assertTrue(r)
|
||
|
|
|
||
|
|
|
||
|
|
def test_greedy_data(self):
|
||
|
|
data = os.urandom(128).hex()
|
||
|
|
data_match_one = data[4:8]
|
||
|
|
data_match_two = data[32:42]
|
||
|
|
data_match_fail = os.urandom(64).hex()
|
||
|
|
data_match = [data_match_one]
|
||
|
|
|
||
|
|
rule = RuleData(data_match, match_all=True)
|
||
|
|
c = AddressRules()
|
||
|
|
c.include(rule)
|
||
|
|
r = c.apply_rules_addresses(self.alice, self.bob, data, self.hsh)
|
||
|
|
self.assertTrue(r)
|
||
|
|
|
||
|
|
data_match = [data_match_two]
|
||
|
|
rule = RuleData(data_match, match_all=True)
|
||
|
|
c = AddressRules()
|
||
|
|
c.include(rule)
|
||
|
|
r = c.apply_rules_addresses(self.alice, self.bob, data, self.hsh)
|
||
|
|
self.assertTrue(r)
|
||
|
|
|
||
|
|
data_match = [data_match_two, data_match_one]
|
||
|
|
rule = RuleData(data_match, match_all=True)
|
||
|
|
c = AddressRules()
|
||
|
|
c.include(rule)
|
||
|
|
r = c.apply_rules_addresses(self.alice, self.bob, data, self.hsh)
|
||
|
|
self.assertTrue(r)
|
||
|
|
|
||
|
|
data_match = [data_match_two, data_match_fail, data_match_one]
|
||
|
|
rule = RuleData(data_match, match_all=True)
|
||
|
|
c = AddressRules()
|
||
|
|
c.include(rule)
|
||
|
|
r = c.apply_rules_addresses(self.alice, self.bob, data, self.hsh)
|
||
|
|
self.assertFalse(r)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|