from gpu_rent.bootstrap import bootstrap_script


def test_bootstrap_script_is_native_swarmui():
    script = bootstrap_script()
    assert "docker.io" not in script
    assert "docker run" not in script
    assert "nvidia-container" not in script
    assert "/opt/swarmui" in script
    assert "/mnt/swarm_data" in script
    assert "launch-linux.sh --launch_mode none --host 127.0.0.1 --port 7801" in script
    assert "systemctl enable swarmui" in script
    assert "mkfs.ext4" in script
    assert "apt-get upgrade" not in script
    assert ".gpu-rent-ready" in script
    assert "src/BuiltinExtensions/ComfyUIBackend/DLNodes" in script