def test_join_files_remote_exception(self, monkeypatch: MonkeyPatch): class MockSSHShell2(MockSSHShell): def run(self, cmd, timeout): raise SSHException chunks = [["somefile_aa", 10000], ["somefile_ab", 10000]] scput = SplitCopyPut() scput.sshshell = MockSSHShell2() scput.scs = MockSplitCopyShared() scput.remote_dir = "/var/tmp/foo" scput.remote_file = "somefile" monkeypatch.setattr("builtins.open", MockOpen) with raises(SystemExit): scput.join_files_remote(MockSCPClient, chunks, "/tmp/foo")
def test_join_files_remote(self, monkeypatch: MonkeyPatch): class MockSSHShell2(MockSSHShell): def run(self, cmd, timeout): return True, "" chunks = [["somefile_aa", 10000], ["somefile_ab", 10000]] scput = SplitCopyPut() scput.sshshell = MockSSHShell2() scput.scs = MockSplitCopyShared() scput.remote_dir = "/var/tmp/foo" scput.remote_file = "somefile" monkeypatch.setattr("builtins.open", MockOpen) result = scput.join_files_remote(MockSCPClient, chunks, "/tmp/foo") assert result == None