def test_local_sha_get_mismatch(self, monkeypatch: MonkeyPatch):
class Mock512(MockHash):
pass
class Mock384(MockHash):
pass
class Mock256(MockHash):
pass
class Mock224(MockHash):
pass
class Mock1(MockHash):
pass
scget = SplitCopyGet()
monkeypatch.setattr("builtins.open", MockOpen)
monkeypatch.setattr("os.path.sep", "/")
monkeypatch.setattr("hashlib.sha512", Mock512)
monkeypatch.setattr("hashlib.sha384", Mock384)
monkeypatch.setattr("hashlib.sha256", Mock256)
monkeypatch.setattr("hashlib.sha224", Mock224)
monkeypatch.setattr("hashlib.sha1", Mock1)
scget.scs = MockSplitCopyShared()
sha_hash = {
512: "0bcdef0123456789",
}
scget.local_file = "foo"
scget.local_dir = "/var/tmp"
with raises(SystemExit):
scget.local_sha_get(sha_hash)
def test_delete_target_local(self, monkeypatch: MonkeyPatch):
def exists(*args):
return True
def remove(*args):
return True
monkeypatch.setattr("os.path.exists", exists)
monkeypatch.setattr("os.remove", remove)
monkeypatch.setattr("os.path.sep", "/")
scget = SplitCopyGet()
scget.local_dir = "/var/tmp"
scget.local_file = "foo"
result = scget.delete_target_local()
expected = None
assert expected == result
def test_local_sha_get(self, monkeypatch: MonkeyPatch):
class MockHash:
def __init__(self):
pass
def update(self, data):
pass
def hexdigest(self):
return "abcdef0123456789"
class Mock512(MockHash):
pass
class Mock384(MockHash):
pass
class Mock256(MockHash):
pass
class Mock224(MockHash):
pass
class Mock1(MockHash):
pass
scget = SplitCopyGet()
monkeypatch.setattr("builtins.open", MockOpen)
monkeypatch.setattr("os.path.sep", "/")
monkeypatch.setattr("hashlib.sha512", Mock512)
monkeypatch.setattr("hashlib.sha384", Mock384)
monkeypatch.setattr("hashlib.sha256", Mock256)
monkeypatch.setattr("hashlib.sha224", Mock224)
monkeypatch.setattr("hashlib.sha1", Mock1)
sha_hash = {
1: "abcdef0123456789",
224: "abcdef0123456789",
256: "abcdef0123456789",
384: "abcdef0123456789",
512: "abcdef0123456789",
}
scget.local_file = "foo"
scget.local_dir = "/var/tmp"
for i in [512, 384, 256, 224, 1]:
result = scget.local_sha_get(sha_hash)
assert result == None
del sha_hash[i]
def test_join_files_local_fail(self, monkeypatch: MonkeyPatch):
def isfile(path):
return False
def glob(path):
return ["file1", "file2"]
monkeypatch.setattr("glob.glob", glob)
monkeypatch.setattr("builtins.open", MockOpen)
monkeypatch.setattr("os.path.isfile", isfile)
monkeypatch.setattr("os.path.sep", "/")
scget = SplitCopyGet()
scget.remote_file = "foo"
scget.local_file = "foo"
scget.local_dir = "/var/tmp"
scget.scs = MockSplitCopyShared()
with raises(SystemExit):
scget.join_files_local()
def test_join_files_local(self, monkeypatch: MonkeyPatch):
def isfile(path):
return True
def glob(path):
return ["file1", "file2"]
monkeypatch.setattr("glob.glob", glob)
monkeypatch.setattr("builtins.open", MockOpen)
monkeypatch.setattr("os.path.isfile", isfile)
monkeypatch.setattr("os.path.sep", "/")
scget = SplitCopyGet()
scget.remote_file = "foo"
scget.local_file = "foo"
scget.local_dir = "/var/tmp"
scget.scs = MockSplitCopyShared()
result = scget.join_files_local()
assert result == None