Added unit test for create_solo task

This commit is contained in:
Mike Scherbakov 2012-07-31 20:32:18 +04:00
parent 512c6ad3ad
commit 163b782f9d
1 changed files with 35 additions and 0 deletions

View File

@ -1,7 +1,11 @@
import os
import json
import mock
from mock import call
from django.test import TestCase
from django.db.models import Model
from django.conf import settings
from celery.task import task
from nailgun import tasks
@ -228,3 +232,34 @@ class TestTasks(TestCase):
call().apply_async()
]
self.assertEquals(tasks.TaskPool.mock_calls, expected)
def test_create_solo_task(self):
dummy_list = []
tasks.create_solo(dummy_list, 1, self.recipe.id)
with open(
os.path.join(
settings.CHEF_CONF_FOLDER,
"".join([self.node.id, ".json"])
),
"r"
) as entity:
solo = entity.read()
expected = {
"cluster_id": 1,
"components_dict": {
self.node.id: {
"node_ips": {}
}
},
"components_list": [
{
"node_id": self.node.id,
"node_ips": {}
}
],
"run_list": [
"recipe[%s]" % self.recipe.recipe
]
}
self.assertEquals(expected, json.loads(solo))