def test_compare_file_sizes_cmd_fail(self, monkeypatch: MonkeyPatch):
class MockSSHShell2(MockSSHShell):
def run(self, cmd, timeout=30):
return False, ""
scput = SplitCopyPut()
scput.sshshell = MockSSHShell2()
scput.scs = MockSplitCopyShared()
with raises(SystemExit):
scput.compare_file_sizes(100000)
def test_compare_file_sizes_mismatch(self, monkeypatch: MonkeyPatch):
class MockSSHShell2(MockSSHShell):
def run(self, cmd, timeout=30):
stdout = (
b"foo@bar ~ % ls -l /var/tmp/foo\r\r\n"
b"-rw-r--r-- 1 foo bar 400000 19 Dec 2019 /var/tmp/foo\r\n"
b"foo@bar ~ %").decode()
return True, stdout
scput = SplitCopyPut()
scput.sshshell = MockSSHShell2()
scput.scs = MockSplitCopyShared()
with raises(SystemExit):
scput.compare_file_sizes(100000)
def test_compare_file_sizes(self, monkeypatch: MonkeyPatch):
class MockSSHShell2(MockSSHShell):
def run(self, cmd, timeout=30):
stdout = (
b"foo@bar ~ % ls -l /var/tmp/foo\r\r\n"
b"-rw-r--r-- 1 foo bar 100000 19 Dec 2019 /var/tmp/foo\r\n"
b"foo@bar ~ % ").decode()
return True, stdout
scput = SplitCopyPut()
scput.sshshell = MockSSHShell2()
result = scput.compare_file_sizes(100000)
assert result == None
