fix imports

Signed-off-by: Steven Armstrong <steven@icarus.ethz.ch>
This commit is contained in:
Steven Armstrong 2011-10-14 09:05:22 +02:00
parent f6adefddec
commit 8e224b43c3

View file

@ -24,7 +24,7 @@ import tempfile
import unittest import unittest
import shutil import shutil
import cdist.core from cdist import core
import os.path as op import os.path as op
my_dir = op.abspath(op.dirname(__file__)) my_dir = op.abspath(op.dirname(__file__))
@ -35,19 +35,19 @@ type_base_path = op.join(fixtures, 'type')
class ObjectClassTestCase(unittest.TestCase): class ObjectClassTestCase(unittest.TestCase):
def test_list_object_names(self): def test_list_object_names(self):
object_names = list(cdist.core.Object.list_object_names(object_base_path)) object_names = list(core.Object.list_object_names(object_base_path))
self.assertEqual(object_names, ['__first/man', '__second/on-the', '__third/moon']) self.assertEqual(object_names, ['__first/man', '__second/on-the', '__third/moon'])
def test_list_type_names(self): def test_list_type_names(self):
type_names = list(cdist.core.Object.list_type_names(object_base_path)) type_names = list(core.Object.list_type_names(object_base_path))
self.assertEqual(type_names, ['__first', '__second', '__third']) self.assertEqual(type_names, ['__first', '__second', '__third'])
def test_list_objects(self): def test_list_objects(self):
objects = list(cdist.core.Object.list_objects(object_base_path, type_base_path)) objects = list(core.Object.list_objects(object_base_path, type_base_path))
objects_expected = [ objects_expected = [
cdist.core.Object(cdist.core.Type(type_base_path, '__first'), object_base_path, 'man'), core.Object(core.Type(type_base_path, '__first'), object_base_path, 'man'),
cdist.core.Object(cdist.core.Type(type_base_path, '__second'), object_base_path, 'on-the'), core.Object(core.Type(type_base_path, '__second'), object_base_path, 'on-the'),
cdist.core.Object(cdist.core.Type(type_base_path, '__third'), object_base_path, 'moon'), core.Object(core.Type(type_base_path, '__third'), object_base_path, 'moon'),
] ]
self.assertEqual(objects, objects_expected) self.assertEqual(objects, objects_expected)
@ -55,8 +55,8 @@ class ObjectClassTestCase(unittest.TestCase):
class ObjectTestCase(unittest.TestCase): class ObjectTestCase(unittest.TestCase):
def setUp(self): def setUp(self):
self.cdist_type = cdist.core.Type(type_base_path, '__third') self.cdist_type = core.Type(type_base_path, '__third')
self.cdist_object = cdist.core.Object(self.cdist_type, object_base_path, 'moon') self.cdist_object = core.Object(self.cdist_type, object_base_path, 'moon')
def tearDown(self): def tearDown(self):
self.cdist_object.changed = False self.cdist_object.changed = False