mirror of
https://github.com/codex-storage/bittorrent-benchmarks.git
synced 2025-02-10 18:16:30 +00:00
27 lines
649 B
Python
27 lines
649 B
Python
import os
|
|
import tempfile
|
|
from pathlib import Path
|
|
from typing import Generator
|
|
|
|
import pytest
|
|
from urllib3.util import Url, parse_url
|
|
|
|
from benchmarks.core.utils import megabytes
|
|
|
|
|
|
@pytest.fixture
|
|
def temp_random_file() -> Generator[Path, None, None]:
|
|
with tempfile.TemporaryDirectory() as temp_dir_str:
|
|
temp_dir = Path(temp_dir_str)
|
|
random_file = temp_dir / 'data.bin'
|
|
random_bytes = os.urandom(megabytes(1))
|
|
with random_file.open('wb') as outfile:
|
|
outfile.write(random_bytes)
|
|
|
|
yield random_file
|
|
|
|
|
|
@pytest.fixture
|
|
def tracker() -> Url:
|
|
return parse_url('http://127.0.0.1:8000/announce')
|