from datetime import datetime, timedelta
from candolint.models import database, Project, Change, Check
def setup_module():
    with database.transaction():
        Project.create(
            url='https://example.com/alice/test-models',
            domain='example.com',
            user='alice',
            name='test-models')
def test_get_title():
    project = Project()
    project.domain = 'example.com'
    project.user = 'alice'
    project.name = 'hello-world'
    assert project.get_title() == 'alice/hello-world'
    project.domain = 'hello-world.com'
    project.user = None
    project.name = 'central'
    assert project.get_title() == 'hello-world.com/central'
def test_duration():
    now = datetime.now()
    check = Check(started=now)
    check.finished = now + timedelta(seconds=10)
    assert check.get_duration() == '10 sec'
    check.finished = now + timedelta(minutes=5)
    assert check.get_duration() == '5 min'
    check.finished = now + timedelta(hours=2)
    assert check.get_duration() == '120 min'
def test_manual_upsert():
    data = {
        'rev': 42,
        'node': '92cfceb39d57d914ed8b14d0e37643de0797ae56',
        'branch': 'default',
        'date': datetime.now(),
        'author': 'alice',
        'message': 'project: add a thing',
        'project': Project.get(name='test-models')
    }
    with database.transaction():
        change, created = Change.manual_upsert(**data)
    assert change is not None
    assert created
    change = Change.get(id=change.id)
    assert change.rev == 42
    old_id = change.id
    data['rev'] = None
    with database.transaction():
        change, created = Change.manual_upsert(**data)
    assert change is not None
    assert not created
    change = Change.get(id=change.id)
    assert change.rev is None
    assert change.id == old_id