def test_get_chunk_info_cmdfail(self):
class MockSSHShell2(MockSSHShell):
def run(self, cmd):
result = False
stdout = ""
return result, stdout
scget = SplitCopyGet()
scget.sshshell = MockSSHShell2()
scget.scs = MockSplitCopyShared()
scget.remote_file = "somefile.0.gz"
with raises(SystemExit):
scget.get_chunk_info(
"/var/tmp/splitcopy_somefile.0.gz.220622105712")
def test_get_chunk_info_matchfail(self):
class MockSSHShell2(MockSSHShell):
def run(self, cmd):
result = True
stdout = (
"ls -l /var/tmp/splitcopy_somefile.0.gz.220622105712/\n"
"total 2K\n"
"-rw------- 1 foo bar 269 Jun 22 00:57 split.sh\n"
"foo@bar:~$")
return result, stdout
scget = SplitCopyGet()
scget.sshshell = MockSSHShell2()
scget.scs = MockSplitCopyShared()
scget.remote_file = "somefile.0.gz"
with raises(SystemExit):
scget.get_chunk_info(
"/var/tmp/splitcopy_somefile.0.gz.220622105712")
def test_get_chunk_info(self):
class MockSSHShell2(MockSSHShell):
def run(self, cmd):
result = True
stdout = (
"ls -l /var/tmp/splitcopy_somefile.0.gz.220622105712/\n"
"total 6640K\n"
"-rw------- 1 foo bar 677888 Jun 22 00:57 somefile.0.gz_00\n"
"-rw------- 1 foo bar 673790 Jun 22 00:57 somefile.0.gz_01\n"
"-rw------- 1 foo bar 269 Jun 22 00:57 split.sh\n"
"foo@bar:~$")
return result, stdout
scget = SplitCopyGet()
scget.sshshell = MockSSHShell2()
scget.remote_file = "somefile.0.gz"
result = scget.get_chunk_info(
"/var/tmp/splitcopy_somefile.0.gz.220622105712")
expected = [["somefile.0.gz_00", 677888], ["somefile.0.gz_01", 673790]]
assert expected == result