def test_add_wallet(self):
sys.argv = [exe_cli] + [
"wallet", "add", "test",
utils.generate_random_private_key()
]
main()
self.assertTrue(Vault().exist_wallet("test"))
def test_deploy(self):
config_file = os.path.join(self.my_co_path, "config.yaml")
sys.argv = [exe_cli
] + ["deploy", config_file, "--wallet", self.wallet.name]
main()
self.assertTrue(os.path.isfile(self.c_org_manager.build_file))
contract = self.c_org_manager.contract
self.assertTrue(len(contract.find_functions_by_name('buy')))
def test_deploy_default_wallet(self):
Vault().default_wallet().add_ether(init_ether)
config_file = os.path.join(self.my_co_path, "config.yaml")
sys.argv = [exe_cli, "deploy", config_file]
main()
self.assertTrue(os.path.isfile(self.c_org_manager.build_file))
contract = self.c_org_manager.contract
self.assertTrue(len(contract.find_functions_by_name('buy')))
def test_rm_wallet(self):
Vault().create_wallet(name="name")
self.assertTrue(Vault().exist_wallet("name"))
sys.argv = [exe_cli, "wallet", "remove", "name"]
main()
self.assertFalse(Vault().exist_wallet("name"))
def test_stats(self):
self.c_org_manager.buy(0.1, self.wallet)
sys.argv = [exe_cli] + ["stats", "my-co", "--wallet", self.wallet.name]
main()
def test_revenue(self):
sys.argv = [exe_cli] + ["revenue", "my-co", "--revenue", "0.1"]
main()
def test_sell(self):
self.c_org_manager.buy(0.1, self.wallet)
sys.argv = [exe_cli] + [
"sell", "my-co", "--wallet", self.wallet.name, "--amount", "1"
]
main()
def test_buy(self):
sys.argv = [exe_cli] + [
"buy", "my-co", "--wallet", self.wallet.name, "--amount", "0.1"
]
main()
def test_add_ether_wallet(self):
wallet = Vault().create_wallet(name="test-add-ether")
sys.argv = [exe_cli, "wallet", "add_ether", "test-add-ether"]
main()
self.assertEqual(init_ether, wallet.balance)
def test_list_wallet(self):
Vault().create_wallet(name="test-list")
with self.assertLogs() as cm:
sys.argv = [exe_cli, "wallet", "list"]
main()
self.assertIn(True, ['test-list' in i for i in cm.output])
def test_create_wallet(self):
sys.argv = [exe_cli, "wallet", "create", "my-wallet"]
main()
self.assertTrue(Vault().exist_wallet("my-wallet"))
