mirror of git://holbrook.no/eth-monitor.git
41 lines
990 B
Python
41 lines
990 B
Python
# standard imports
|
|
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_address_include(self):
|
|
outs = [self.alice]
|
|
ins = []
|
|
execs = []
|
|
rule = RuleSimple(outs, ins, execs)
|
|
c = AddressRules()
|
|
c.include(rule)
|
|
data = b''
|
|
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)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|