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
