fix: fix binary mode

Author: dantetemplar

backend/src/modules/random_generator/repository.py

@@ -37,7 +37,7 @@ def generate_binary_sequence(self, length: int) -> str:
         return "".join(self._rng.choice("01") for _ in range(length))
 
     def generate_bits(self, length: int) -> bytes:
-        return self._rng.getrandbits(length).to_bytes(length, "big")
+        return self._rng.getrandbits(length).to_bytes((length + 7) // 8, "big")
 
 
 class RandomGeneratorRepository:

backend/src/modules/random_generator/routes.py

@@ -81,12 +81,15 @@ async def file_binary(
     random_generator = random_repository.generator(seed_info.seed_bytes)
     if mode == FileOutputMode.ZERO_ONES:
         filename = f"zero_ones_{length}_{seed_info.seed_hex[:16]}.txt"
+        media_type = "text/plain"
         content = random_generator.generate_binary_sequence(length)
     elif mode == FileOutputMode.BINARY:
         filename = f"binary_{length}_{seed_info.seed_hex[:16]}.bin"
+        media_type = "application/octet-stream"
         content = random_generator.generate_bits(length)
     elif mode == FileOutputMode.NUMBERS_SEPARATED_BY_COMMAS:
         filename = f"numbers_{length}_{seed_info.seed_hex[:16]}.txt"
+        media_type = "text/plain"
         content = ",".join(str(random_generator.generate_random_number(0, 100)) for _ in range(length))
     else:
         raise HTTPException(status_code=400, detail="Invalid mode")
@@ -98,5 +101,5 @@ async def file_binary(
     return Response(
         content=content,
         headers=headers,
-        media_type="text/plain",
+        media_type=media_type,
     )