add btc teleport tests

Author: yungwine

modules/btc_teleport.py

@@ -1,9 +1,9 @@
-import json
 import os
 import subprocess
+from typing import Optional
 
 from modules.module import MtcModule
-from mypylib.mypylib import run_as_root, color_print, bcolors, print_table
+from mypylib.mypylib import run_as_root, color_print
 from mytoncore.utils import get_package_resource_path
 from mytonctrl.utils import get_current_user
 
@@ -55,14 +55,14 @@ def create_env_file(self, reinit=False):
         with open(env_path, 'w') as f:
             f.write(text)
 
-    def add_daemon(self, user: str = None):
+    def add_daemon(self, user: Optional[str] = None):
         start = f'{self.bin_dir}/oracle'
         if user is None:
             user = get_current_user()
         with get_package_resource_path('mytoninstaller', 'scripts/add2systemd.sh') as script_path:
             run_as_root(['bash', script_path, '-n', 'btc_teleport', '-u', user, '-g', user, '-s', start, '-w', self.bin_dir])
 
-    def install(self, branch: str, user: str = None):
+    def install(self, branch: str, user: Optional[str] = None):
         if user is None:
             user = get_current_user()
         with get_package_resource_path('mytonctrl', 'scripts/btc_teleport1.sh') as script_path:
@@ -72,7 +72,7 @@ def install(self, branch: str, user: str = None):
         with get_package_resource_path('mytonctrl', 'scripts/btc_teleport2.sh') as script_path:
             subprocess.run(["bash", script_path, "-s", self.src_dir])
 
-    def init(self, reinstall=False, branch: str = 'master', user: str = None):
+    def init(self, reinstall=False, branch: str = 'master', user: Optional[str] = None):
         if os.path.exists(self.src_dir) and not reinstall:
             return
         if self.ton.local.db.get('btcTeleportDisabled'):

tests/integration/test_btc_teleport_commands.py

@@ -0,0 +1,52 @@
+from modules import btc_teleport as btc_teleport_module
+from mytoncore import get_package_resource_path
+from mytoncore.mytoncore import MyTonCore
+
+
+def test_remove_btc_teleport(cli, monkeypatch, ton):
+    with get_package_resource_path('mytonctrl', 'scripts/remove_btc_teleport.sh') as script_path:
+        assert script_path.is_file()
+
+    # non-masterchain validator, no --force needed
+    return_code = 0
+    run_args = []
+    def fake_run_as_root(args: list):
+        nonlocal return_code, run_args
+        run_args = args
+        return return_code
+
+    monkeypatch.setattr(btc_teleport_module, "run_as_root", fake_run_as_root)
+
+    monkeypatch.setattr(MyTonCore, "GetValidatorIndex", lambda self: -1)
+    monkeypatch.setattr(MyTonCore, "GetConfig34", lambda self: {"mainValidators": 100})
+
+    output = cli.execute("remove_btc_teleport", no_color=True)
+    assert "Removed btc_teleport" in output
+    assert run_args == ['bash', script_path, '-s', '/usr/src/ton-teleport-btc-periphery', '-k', ton.local.buffer.my_work_dir + '/btc_oracle_keystore']
+
+    # bad args
+    run_args = []
+    output = cli.execute("remove_btc_teleport arg1 arg2")
+    assert "Bad args" in output
+    assert not run_args
+
+    # masterchain validator without --force
+    monkeypatch.setattr(MyTonCore, "GetValidatorIndex", lambda self: 10)
+    monkeypatch.setattr(MyTonCore, "GetConfig34", lambda self: {"mainValidators": 100})
+    output = cli.execute("remove_btc_teleport", no_color=True)
+    assert run_args == []
+    assert 'You can not remove btc_teleport on working masterchain validator' in output
+
+    # masterchain validator with --force
+    monkeypatch.setattr(MyTonCore, "GetValidatorIndex", lambda self: 0)
+    monkeypatch.setattr(MyTonCore, "GetConfig34", lambda self: {"mainValidators": 10})
+    output = cli.execute("remove_btc_teleport --force", no_color=True)
+    assert "Removed btc_teleport" in output
+    assert run_args == ['bash', script_path, '-s', '/usr/src/ton-teleport-btc-periphery', '-k', ton.local.buffer.my_work_dir + '/btc_oracle_keystore']
+
+    # non-masterchain validator
+    monkeypatch.setattr(MyTonCore, "GetValidatorIndex", lambda self: 10)
+    monkeypatch.setattr(MyTonCore, "GetConfig34", lambda self: {"mainValidators": 10})
+    output = cli.execute("remove_btc_teleport", no_color=True)
+    assert "Removed btc_teleport" in output
+    assert run_args == ['bash', script_path, '-s', '/usr/src/ton-teleport-btc-periphery', '-k', ton.local.buffer.my_work_dir + '/btc_oracle_keystore']