import base64

from gpu_rent.cloud import _ssh_user_data, _user_data_b64


def test_user_data_b64_is_selectel_cloud_config():
    key = "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIFakeKeyMaterialHere gpu-rent"
    plain = _ssh_user_data(key)
    assert plain.startswith("#cloud-config")
    assert key in plain
    raw = base64.b64decode(_user_data_b64(key))
    assert raw.decode("utf-8") == plain