def setUp(self):
self.colormap = MockColorMap()
d = self.colormap.d
self.colorset0 = ColorSet(self.colormap, [''])
self.colorset1 = ColorSet(self.colormap, [d[1]])
self.colorset1alt = ColorSet(self.colormap, [d[1],d[6]])
self.colorset2 = ColorSet(self.colormap, [d[2]])
self.colorset3 = ColorSet(self.colormap, [d[3], d[4]])
self.def1 = {'monikers': ['test1'],
'color_set': self.colorset1.get_data(),
'unit':10}
self.asset1 = AssetDefinition(self.colormap, self.def1)
self.master_key = '265a1a0ad05e82fa321e3f6f6767679df0c68515797e0e4e24be1afc3272ee658ec53cecb683ab76a8377273347161e123fddf5320cbbce8849b0a00557bd12c'
self.privkey = '5Kb8kLf9zgWQnogidDA76MzPL6TsZZY36hWXMssSzNydYXYB9KF'
self.pubkey = '1CC3X2gu58d6wXUWMffpuzN9JAfTUWu4Kj'
c = {
'dw_master_key': self.master_key,
'dwam': {
'genesis_color_sets':[self.colorset1.get_data(),
self.colorset2.get_data()],
'color_set_states':[
{'color_set':self.colorset0.get_data(), "max_index":0},
{'color_set':self.colorset1.get_data(), "max_index":3},
{'color_set':self.colorset2.get_data(), "max_index":2},
],
},
'addresses':[{'address_data': self.privkey,
'color_set': self.colorset1.get_data(),
}],
'testnet': False,
}
self.config = copy.deepcopy(c)
self.maindwam = DWalletAddressManager(self.colormap, c)
def test_init_new_wallet(self):
c = copy.deepcopy(self.config)
del c['dw_master_key']
del c['dwam']
newdwam = DWalletAddressManager(self.colormap, c)
params = newdwam.init_new_wallet()
self.assertNotEqual(self.config['dw_master_key'],
newdwam.config['dw_master_key'])
c = copy.deepcopy(self.config)
c['addresses'][0]['address_data'] = 'notreal'
self.assertRaises(EncodingError, DWalletAddressManager,
self.colormap, c)
class TestDeterministic(unittest.TestCase):
def setUp(self):
self.colormap = MockColorMap()
d = self.colormap.d
self.colorset0 = ColorSet(self.colormap, [''])
self.colorset1 = ColorSet(self.colormap, [d[1]])
self.colorset1alt = ColorSet(self.colormap, [d[1],d[6]])
self.colorset2 = ColorSet(self.colormap, [d[2]])
self.colorset3 = ColorSet(self.colormap, [d[3], d[4]])
self.def1 = {'monikers': ['test1'],
'color_set': self.colorset1.get_data(),
'unit':10}
self.asset1 = AssetDefinition(self.colormap, self.def1)
self.master_key = '265a1a0ad05e82fa321e3f6f6767679df0c68515797e0e4e24be1afc3272ee658ec53cecb683ab76a8377273347161e123fddf5320cbbce8849b0a00557bd12c'
self.privkey = '5Kb8kLf9zgWQnogidDA76MzPL6TsZZY36hWXMssSzNydYXYB9KF'
self.pubkey = '1CC3X2gu58d6wXUWMffpuzN9JAfTUWu4Kj'
c = {
'dw_master_key': self.master_key,
'dwam': {
'genesis_color_sets':[self.colorset1.get_data(),
self.colorset2.get_data()],
'color_set_states':[
{'color_set':self.colorset0.get_data(), "max_index":0},
{'color_set':self.colorset1.get_data(), "max_index":3},
{'color_set':self.colorset2.get_data(), "max_index":2},
],
},
'addresses':[{'address_data': self.privkey,
'color_set': self.colorset1.get_data(),
}],
'testnet': False,
}
self.config = copy.deepcopy(c)
self.maindwam = DWalletAddressManager(self.colormap, c)
def test_init_new_wallet(self):
c = copy.deepcopy(self.config)
del c['dw_master_key']
del c['dwam']
newdwam = DWalletAddressManager(self.colormap, c)
params = newdwam.init_new_wallet()
self.assertNotEqual(self.config['dw_master_key'],
newdwam.config['dw_master_key'])
c = copy.deepcopy(self.config)
c['addresses'][0]['address_data'] = 'notreal'
self.assertRaises(EncodingError, DWalletAddressManager,
self.colormap, c)
def test_get_new_address(self):
self.assertEqual(self.maindwam.get_new_address(self.colorset2).index,
3)
self.assertEqual(self.maindwam.get_new_address(self.asset1).index,
4)
def test_get_new_genesis_address(self):
addr = self.maindwam.get_new_genesis_address()
self.assertEqual(addr.index, 2)
addr2 = self.maindwam.get_new_address(addr.get_color_set())
self.assertEqual(addr2.index, 0)
def test_get_update_genesis_address(self):
addr = self.maindwam.get_genesis_address(0)
self.maindwam.update_genesis_address(addr, self.colorset1alt)
self.assertEqual(addr.get_color_set(), self.colorset1alt)
def test_get_change_address(self):
addr = self.maindwam.get_change_address(self.colorset0)
self.assertEqual(addr.get_color_set().__repr__(),
self.colorset0.__repr__())
self.assertEqual(self.maindwam.get_change_address(self.colorset3).index,
0)
def test_get_all_addresses(self):
addrs = [a.get_address() for a in self.maindwam.get_all_addresses()]
self.assertTrue(self.pubkey in addrs)
