from encode_daemon import _one_enc


def test_h264_uses_igpu_va_not_x264_or_arc_av1():
    cmd = _one_enc(640, 368, 30, 2000, 3, 4, "h264", render="/dev/dri/renderD129")
    s = " ".join(cmd)
    assert "varenderD129h264enc" in cmd
    assert "x264enc" not in cmd
    assert "vaav1enc" not in cmd
    assert "format=nv12" in s
    assert "key-int-max=1" in cmd
    assert "b-frames=0" in cmd


def test_h264_factory_follows_render_node():
    cmd = _one_enc(640, 368, 30, 2000, 3, 4, "h264", render="/dev/dri/renderD128")
    assert "varenderD128h264enc" in cmd


def test_c920_stays_fast_gop1():
    cmd = _one_enc(640, 368, 30, 2000, 3, 4, "h264", render="/dev/dri/renderD129")
    assert "bitrate=2000" in cmd
    assert "key-int-max=1" in cmd
    assert "target-usage=7" in cmd


def test_rally_1080_uses_higher_bitrate_and_no_motion_trails():
    cmd = _one_enc(1920, 1088, 30, 2000, 3, 4, "h264", render="/dev/dri/renderD129")
    assert "bitrate=20000" in cmd
    assert "key-int-max=1" in cmd
    assert "target-usage=2" in cmd
    assert "b-frames=0" in cmd


def test_rally_bitrate_rollbacks_via_hi_kbps():
    cmd = _one_enc(1920, 1088, 30, 2000, 3, 4, "h264",
                   render="/dev/dri/renderD129", hi_kbps=12000)
    assert "bitrate=12000" in cmd


def test_av1_stays_on_arc_vaav1enc():
    cmd = _one_enc(640, 368, 30, 2000, 3, 4, "av1")
    assert "vaav1enc" in cmd
    assert "varenderD129h264enc" not in cmd