19
19
msg_mempool ,
20
20
msg_version ,
21
21
)
22
- from test_framework .mininode import P2PInterface
22
+ from test_framework .mininode import P2PInterface , mininode_lock
23
23
from test_framework .script import MAX_SCRIPT_ELEMENT_SIZE
24
24
from test_framework .test_framework import BitcoinTestFramework
25
25
@@ -36,6 +36,11 @@ class FilterNode(P2PInterface):
36
36
nFlags = 1 ,
37
37
)
38
38
39
+ def __init__ (self ):
40
+ super ().__init__ ()
41
+ self ._tx_received = False
42
+ self ._merkleblock_received = False
43
+
39
44
def on_inv (self , message ):
40
45
want = msg_getdata ()
41
46
for i in message .inv :
@@ -48,10 +53,30 @@ def on_inv(self, message):
48
53
self .send_message (want )
49
54
50
55
def on_merkleblock (self , message ):
51
- self .merkleblock_received = True
56
+ self ._merkleblock_received = True
52
57
53
58
def on_tx (self , message ):
54
- self .tx_received = True
59
+ self ._tx_received = True
60
+
61
+ @property
62
+ def tx_received (self ):
63
+ with mininode_lock :
64
+ return self ._tx_received
65
+
66
+ @tx_received .setter
67
+ def tx_received (self , value ):
68
+ with mininode_lock :
69
+ self ._tx_received = value
70
+
71
+ @property
72
+ def merkleblock_received (self ):
73
+ with mininode_lock :
74
+ return self ._merkleblock_received
75
+
76
+ @merkleblock_received .setter
77
+ def merkleblock_received (self , value ):
78
+ with mininode_lock :
79
+ self ._merkleblock_received = value
55
80
56
81
57
82
class FilterTest (BitcoinTestFramework ):
0 commit comments