32 lines
1014 B
Python
32 lines
1014 B
Python
import pytest
|
|
|
|
from drydock_provisioner import objects
|
|
|
|
|
|
class TestPostgres(object):
|
|
def test_result_message_insert(self, populateddb, drydock_state):
|
|
"""Test that a result message for a task can be added."""
|
|
msg1 = objects.TaskStatusMessage('Error 1', True, 'node', 'node1')
|
|
msg2 = objects.TaskStatusMessage('Status 1', False, 'node', 'node1')
|
|
|
|
result = drydock_state.post_result_message(populateddb.task_id, msg1)
|
|
assert result
|
|
result = drydock_state.post_result_message(populateddb.task_id, msg2)
|
|
assert result
|
|
|
|
task = drydock_state.get_task(populateddb.task_id)
|
|
|
|
assert task.result.error_count == 1
|
|
|
|
assert len(task.result.message_list) == 2
|
|
|
|
@pytest.fixture(scope='function')
|
|
def populateddb(self, blank_state):
|
|
"""Add dummy task to test against."""
|
|
task = objects.Task(
|
|
action='prepare_site', design_ref='http://test.com/design')
|
|
|
|
blank_state.post_task(task)
|
|
|
|
return task
|