import pytest
from project.remotectl.models import RemoteHost, CommandLog

@pytest.mark.django_db
def test_create_host():
    h = RemoteHost.objects.create(name='Local', hostname='127.0.0.1', username='root')
    assert str(h).startswith('Local')

@pytest.mark.django_db
def test_command_log_mark_finished(user_factory, django_user_model):
    user = django_user_model.objects.create(username='tester')
    h = RemoteHost.objects.create(name='Local', hostname='127.0.0.1', username='root')
    log = CommandLog.objects.create(host=h, command='echo hi', created_by=user)
    log.mark_finished('ok', 0, 'hi')
    assert log.status == 'ok'
    assert log.exit_code == 0