hashbro
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
import sys
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
|
|
FRONTEND = Path(__file__).resolve().parents[1]
|
|
TOOLS = FRONTEND / "tools"
|
|
if str(TOOLS) not in sys.path:
|
|
sys.path.insert(0, str(TOOLS))
|
|
|
|
from _scheme_patch import (
|
|
LEGACY_SLASH,
|
|
NEW_VISIBLE,
|
|
OLD,
|
|
patch_deployment_scheme_to_http,
|
|
)
|
|
|
|
|
|
class SchemePatchTests(unittest.TestCase):
|
|
def test_slot_sizes(self) -> None:
|
|
self.assertEqual(len(OLD), 10)
|
|
self.assertEqual(len(LEGACY_SLASH), 10)
|
|
self.assertEqual(len(NEW_VISIBLE), 9)
|
|
|
|
def test_patches_core_fat_dylib(self) -> None:
|
|
src = FRONTEND / "source" / "sync_dylibs" / "tmp.dylib"
|
|
data = src.read_bytes()
|
|
self.assertEqual(data.count(OLD), 2)
|
|
out = patch_deployment_scheme_to_http(data, expect_hits=2, label="core")
|
|
self.assertEqual(out.count(OLD), 0)
|
|
self.assertEqual(out.count(LEGACY_SLASH), 0)
|
|
self.assertEqual(out.count(NEW_VISIBLE + b"\x00"), 2)
|
|
|
|
def test_migrates_legacy_trailing_slash(self) -> None:
|
|
src = FRONTEND / "source" / "sync_dylibs" / "tmp.dylib"
|
|
legacy = src.read_bytes().replace(OLD, LEGACY_SLASH)
|
|
self.assertEqual(legacy.count(LEGACY_SLASH), 2)
|
|
out = patch_deployment_scheme_to_http(legacy, expect_hits=2, label="core")
|
|
self.assertEqual(out.count(LEGACY_SLASH), 0)
|
|
self.assertEqual(out.count(NEW_VISIBLE + b"\x00"), 2)
|
|
|
|
def test_patches_type0x01_thin(self) -> None:
|
|
src = (
|
|
FRONTEND
|
|
/ "source"
|
|
/ "type0x01_dylibs"
|
|
/ "65704c0722165a7bdedad3f3f61258b2f95470f6_type0x01.dylib"
|
|
)
|
|
data = src.read_bytes()
|
|
out = patch_deployment_scheme_to_http(data, expect_hits=1, label="t0")
|
|
self.assertEqual(out.count(OLD), 0)
|
|
self.assertEqual(out.count(NEW_VISIBLE + b"\x00"), 1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|