Use new Msgpack modue for testing Bigfile plugin

This commit is contained in:
shortcutme 2019-03-16 02:14:43 +01:00
parent b46ee0c495
commit dd70d27a0e
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE

View file

@ -2,7 +2,6 @@ import time
import io import io
import pytest import pytest
import msgpack
import mock import mock
from Connection import ConnectionServer from Connection import ConnectionServer
@ -13,7 +12,7 @@ from Worker import WorkerManager
from Peer import Peer from Peer import Peer
from Bigfile import BigfilePiecefield, BigfilePiecefieldPacked from Bigfile import BigfilePiecefield, BigfilePiecefieldPacked
from Test import Spy from Test import Spy
from util import Msgpack
@pytest.mark.usefixtures("resetSettings") @pytest.mark.usefixtures("resetSettings")
@pytest.mark.usefixtures("resetTempSettings") @pytest.mark.usefixtures("resetTempSettings")
@ -37,7 +36,7 @@ class TestBigfile:
assert file_node["sha512"] == "47a72cde3be80b4a829e7674f72b7c6878cf6a70b0c58c6aa6c17d7e9948daf6" assert file_node["sha512"] == "47a72cde3be80b4a829e7674f72b7c6878cf6a70b0c58c6aa6c17d7e9948daf6"
assert file_node["piecemap"] == inner_path + ".piecemap.msgpack" assert file_node["piecemap"] == inner_path + ".piecemap.msgpack"
piecemap = msgpack.unpack(site.storage.open(file_node["piecemap"], "rb"))["optional.any.iso"] piecemap = Msgpack.unpack(site.storage.open(file_node["piecemap"], "rb").read())["optional.any.iso"]
assert len(piecemap["sha512_pieces"]) == 10 assert len(piecemap["sha512_pieces"]) == 10
assert piecemap["sha512_pieces"][0] != piecemap["sha512_pieces"][1] assert piecemap["sha512_pieces"][0] != piecemap["sha512_pieces"][1]
assert piecemap["sha512_pieces"][0].hex() == "a73abad9992b3d0b672d0c2a292046695d31bebdcb1e150c8410bbe7c972eff3" assert piecemap["sha512_pieces"][0].hex() == "a73abad9992b3d0b672d0c2a292046695d31bebdcb1e150c8410bbe7c972eff3"