disk map refactoring changes

Author: bitplane

src/blkcache/cache/through.py

@@ -29,7 +29,10 @@ def __init__(self, config):
         # WRONG: WE HAVE A FUNCTION TO DO THIS DON'T GUESS
         self.block_size = config.get("block_size") or DEFAULT_BLOCK_SIZE
         self.map_path = self.cache.with_suffix(f"{self.cache.suffix}.log")
-        self.diskmap = DiskMap(self.map_path)
+
+        # Get device size for DiskMap
+        device_size = get_device_size(self.device)
+        self.diskmap = DiskMap(self.map_path, device_size)
 
         # Initialize cache
         self._initialize()

src/blkcache/plugin.py

@@ -11,8 +11,8 @@
     DEFAULT_BLOCK_SIZE,
 )
 
-# Import Cache class
-from blkcache.cache.cache import Cache
+# Import BlockCache class
+from blkcache.cache.through import BlockCache
 
 # Global state
 DEV: Path | None = None
@@ -57,9 +57,8 @@ def config_complete() -> None:
         raise RuntimeError("device= and cache= are required")
 
     # Create a BlockCache instance
-    CACHE_INSTANCE = BlockCache(
-        device_path=DEV, cache_path=CACHE, block_size=BLOCK if BLOCK != DEFAULT_BLOCK_SIZE else None
-    )
+    config = {"device_path": DEV, "cache_path": CACHE, "block_size": BLOCK if BLOCK != DEFAULT_BLOCK_SIZE else None}
+    CACHE_INSTANCE = BlockCache(config)
 
     # The BlockCache initialization will handle:
     # - Loading the mapfile if it exists

tests/unit/test_blocksize.py

@@ -64,7 +64,6 @@ def test_auto_detection_failure(mock_get_sector_size):
     assert error_msg in updates["block_size_source"]
 
 
-
 def testdetermine_block_size_with_auto_detect():
     """When auto-detection succeeds, the detected size should be used."""
     # We'll test the higher-level determine_block_size function instead