You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
3.8 KiB
95 lines
3.8 KiB
#! /usr/bin/env python |
|
from __future__ import absolute_import, print_function |
|
'''Creates wallets and yield generators in regtest. |
|
Provides seed for joinmarket-qt test. |
|
This should be run via pytest, even though |
|
it's NOT part of the test-suite, because that |
|
makes it much easier to handle start up and |
|
shut down of the environment. |
|
Run it like: |
|
PYTHONPATH=.:$PYTHONPATH py.test \ |
|
--btcroot=/path/to/bitcoin/bin/ \ |
|
--btcpwd=123456abcdef --btcconf=/blah/bitcoin.conf \ |
|
--nirc=2 -s test/ygrunner.py |
|
''' |
|
from common import make_wallets |
|
import os |
|
import pytest |
|
import sys |
|
import time |
|
import random |
|
from jmclient import (YieldGeneratorBasic, ygmain, load_program_config, |
|
jm_single, sync_wallet, JMClientProtocolFactory, |
|
start_reactor) |
|
|
|
class MaliciousYieldGenerator(YieldGeneratorBasic): |
|
"""Overrides, randomly, some maker functions |
|
to prevent taker continuing successfully (unless |
|
they can complete-with-subset). |
|
""" |
|
def set_maliciousness(self, frac): |
|
self.mfrac = frac |
|
def on_auth_received(self, nick, offer, commitment, cr, amount, kphex): |
|
if random.randint(1, 100) < self.mfrac: |
|
print("Counterparty commitment rejected maliciously") |
|
return (False,) |
|
return super(MaliciousYieldGenerator, self).on_auth_received(nick, |
|
offer, commitment, cr, amount, kphex) |
|
def on_tx_received(self, nick, txhex, offerinfo): |
|
if random.randint(1, 100) < self.mfrac: |
|
print("Counterparty tx rejected maliciously") |
|
return (False, "malicious tx rejection") |
|
return super(MaliciousYieldGenerator, self).on_tx_received(nick, txhex, |
|
offerinfo) |
|
|
|
|
|
@pytest.mark.parametrize( |
|
"num_ygs, wallet_structures, mean_amt, malicious", |
|
[ |
|
# 1sp 3yg, honest makers |
|
(3, [[1, 3, 0, 0, 0]] * 4, 2, 0), |
|
# 1sp 3yg, malicious makers reject on auth and on tx 30% of time |
|
#(4, [[1, 3, 0, 0, 0]] * 5, 2, 30), |
|
]) |
|
def test_start_ygs(setup_ygrunner, num_ygs, wallet_structures, mean_amt, |
|
malicious): |
|
"""Set up some wallets, for the ygs and 1 sp. |
|
Then start the ygs in background and publish |
|
the seed of the sp wallet for easy import into -qt |
|
""" |
|
wallets = make_wallets(num_ygs + 1, |
|
wallet_structures=wallet_structures, |
|
mean_amt=mean_amt) |
|
#the sendpayment bot uses the last wallet in the list |
|
wallet = wallets[num_ygs]['wallet'] |
|
print("Seed : " + wallets[num_ygs]['seed']) |
|
#useful to see the utxos on screen sometimes |
|
sync_wallet(wallet) |
|
print(wallet.unspent) |
|
txfee = 1000 |
|
cjfee_a = 4200 |
|
cjfee_r = '0.001' |
|
ordertype = 'swreloffer' |
|
minsize = 100000 |
|
ygclass = MaliciousYieldGenerator if malicious else YieldGeneratorBasic |
|
for i in range(num_ygs): |
|
|
|
cfg = [txfee, cjfee_a, cjfee_r, ordertype, minsize] |
|
sync_wallet(wallets[i]["wallet"]) |
|
yg = ygclass(wallets[i]["wallet"], cfg) |
|
if malicious: |
|
yg.set_maliciousness(malicious) |
|
clientfactory = JMClientProtocolFactory(yg, proto_type="MAKER") |
|
nodaemon = jm_single().config.getint("DAEMON", "no_daemon") |
|
daemon = True if nodaemon == 1 else False |
|
rs = True if i == num_ygs - 1 else False |
|
start_reactor(jm_single().config.get("DAEMON", "daemon_host"), |
|
jm_single().config.getint("DAEMON", "daemon_port"), |
|
clientfactory, daemon=daemon, rs=rs) |
|
time.sleep(2) #give it a chance |
|
|
|
@pytest.fixture(scope="module") |
|
def setup_ygrunner(): |
|
load_program_config() |
|
jm_single().bc_interface.tick_forward_chain_interval = 10 |
|
jm_single().bc_interface.simulate_blocks() |