17 lines
667 B
Python
17 lines
667 B
Python
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
|