import io
import json
import os

import numpy as np
import pytest

from model.model import Model


@pytest.fixture
def sample_img_url() -> str:
    current_dir = os.getcwd()
    return f"{current_dir}/tests/sample_images/computer.jpg"


def test_detect_objects(sample_img_url: str):
    with open(sample_img_url, "rb") as image_file:
        image_bytes = io.BytesIO(image_file.read()).read()
    processed_image = np.frombuffer(image_bytes, dtype=np.uint8)
    
    model = Model()
    image = model.detect_object(data=processed_image)
    list_encoded_image = np.frombuffer(image, dtype=np.uint8).tolist()
    assert json.dumps(list_encoded_image)[:20] == '[137, 80, 78, 71, 13'