From 29f553e5f9178ac9d88fef357d4c84ce105d7558 Mon Sep 17 00:00:00 2001
From: The MMGen Project <mmgen@tuta.io>
Date: Thu, 4 Jun 2020 11:45:22 +0000
Subject: [PATCH] test.py: bogus tracking wallet data fixes

---
 mmgen/rpc.py                 | 3 +++
 mmgen/tw.py                  | 4 ++--
 test/test.py                 | 1 +
 test/test_py_d/ts_ethdev.py  | 1 -
 test/test_py_d/ts_main.py    | 4 +++-
 test/test_py_d/ts_regtest.py | 1 -
 test/test_py_d/ts_tool.py    | 3 ---
 7 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/mmgen/rpc.py b/mmgen/rpc.py
index 9df529c3..43a16c84 100755
--- a/mmgen/rpc.py
+++ b/mmgen/rpc.py
@@ -604,4 +604,7 @@ async def rpc_init(proto,backend=None):
 				proto.chain_name.upper(),
 				rpc.chain.upper() ))
 
+	if g.bogus_wallet_data:
+		rpc.blockcount = 1000000
+
 	return rpc
diff --git a/mmgen/tw.py b/mmgen/tw.py
index f064829f..d9f9d6ca 100755
--- a/mmgen/tw.py
+++ b/mmgen/tw.py
@@ -169,8 +169,8 @@ Actions: [q]uit view, [p]rint to file, pager [v]iew, [w]ide view, add [l]abel:
 		return await self.rpc.call('listunspent',self.minconf,*add_args)
 
 	async def get_unspent_data(self,sort_key=None,reverse_sort=False):
-		if g.bogus_wallet_data: # for debugging purposes only
-			us_raw = eval(get_data_from_file(g.bogus_wallet_data)) # testing, so ok
+		if g.bogus_wallet_data: # for debugging and test suite
+			us_raw = json.loads(get_data_from_file(g.bogus_wallet_data),parse_float=Decimal)
 		else:
 			us_raw = await self.get_unspent_rpc()
 
diff --git a/test/test.py b/test/test.py
index 5bcb64aa..4d7a961a 100755
--- a/test/test.py
+++ b/test/test.py
@@ -746,6 +746,7 @@ class TestSuiteRunner(object):
 		if not quiet:
 			bmsg('Executing ' + m)
 
+		os.environ['MMGEN_BOGUS_WALLET_DATA'] = '' # zero this here, so test group doesn't have to
 		self.ts = self.gm.gm_init_group(self,gname,self.spawn_wrapper)
 
 		if opt.resume_after:
diff --git a/test/test_py_d/ts_ethdev.py b/test/test_py_d/ts_ethdev.py
index 499d00ba..f3c7941b 100755
--- a/test/test_py_d/ts_ethdev.py
+++ b/test/test_py_d/ts_ethdev.py
@@ -311,7 +311,6 @@ class TestSuiteEthdev(TestSuiteBase,TestSuiteShared):
 		self.proto = init_proto(g.coin,network='regtest')
 		from mmgen.daemon import CoinDaemon
 		self.rpc_port = CoinDaemon(proto=self.proto,test_suite=True).rpc_port
-		os.environ['MMGEN_BOGUS_WALLET_DATA'] = ''
 
 	@property
 	def eth_args(self):
diff --git a/test/test_py_d/ts_main.py b/test/test_py_d/ts_main.py
index 35e2845f..5c675c2d 100755
--- a/test/test_py_d/ts_main.py
+++ b/test/test_py_d/ts_main.py
@@ -418,7 +418,9 @@ class TestSuiteMain(TestSuiteBase,TestSuiteShared):
 		silence()
 		ad,tx_data = self._create_tx_data(sources,addrs_per_wallet)
 		dfake = self._create_fake_unspent_data(ad,tx_data,non_mmgen_input,non_mmgen_input_compressed)
-		self._write_fake_data_to_file(repr(dfake))
+		import json
+		from mmgen.rpc import json_encoder
+		self._write_fake_data_to_file(json.dumps(dfake,cls=json_encoder))
 		cmd_args = self._make_txcreate_cmdline(tx_data)
 
 		if cmdline_inputs:
diff --git a/test/test_py_d/ts_regtest.py b/test/test_py_d/ts_regtest.py
index 4eef53a5..3026b03a 100755
--- a/test/test_py_d/ts_regtest.py
+++ b/test/test_py_d/ts_regtest.py
@@ -269,7 +269,6 @@ class TestSuiteRegtest(TestSuiteBase,TestSuiteShared):
 		end_silence()
 
 	def setup(self):
-		os.environ['MMGEN_BOGUS_WALLET_DATA'] = ''
 		try: shutil.rmtree(joinpath(self.tr.data_dir,'regtest'))
 		except: pass
 		os.environ['MMGEN_TEST_SUITE'] = '' # mnemonic is piped to stdin, so stop being a terminal
diff --git a/test/test_py_d/ts_tool.py b/test/test_py_d/ts_tool.py
index f838bdb9..c0fcf410 100755
--- a/test/test_py_d/ts_tool.py
+++ b/test/test_py_d/ts_tool.py
@@ -75,11 +75,8 @@ class TestSuiteTool(TestSuiteMain,TestSuiteBase):
 		return t
 
 	def tool_twview_bad_comment(self): # test correct operation of get_tw_label()
-		bw_save = os.getenv('MMGEN_BOGUS_WALLET_DATA')
 		os.environ['MMGEN_BOGUS_WALLET_DATA'] = joinpath(ref_dir,'bad-comment-unspent.json')
 		t = self.spawn('mmgen-tool',['twview'])
-		if bw_save:
-			os.environ['MMGEN_BOGUS_WALLET_DATA'] = bw_save
 		t.read()
 		t.req_exit_val = 2
 		return t