1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
import os
import unittest
__all__ = ['call', 'datastore', 'Bcfg2TestCase', 'DBModelTestCase', 'syncdb',
'XI', 'XI_NAMESPACE']
datastore = "/"
XI_NAMESPACE = "http://www.w3.org/2001/XInclude"
XI = "{%s}" % XI_NAMESPACE
try:
from django.core.management import setup_environ
has_django = True
os.environ['DJANGO_SETTINGS_MODULE'] = "Bcfg2.settings"
import Bcfg2.settings
Bcfg2.settings.DATABASE_NAME = \
os.path.join(os.path.dirname(os.path.abspath(__file__)), "test.sqlite")
Bcfg2.settings.DATABASES['default']['NAME'] = Bcfg2.settings.DATABASE_NAME
except ImportError:
has_django = False
try:
from mock import call
except ImportError:
def call(*args, **kwargs):
""" the Mock call object is a fairly recent addition, but it's
very very useful, so we create our own function to create Mock
calls """
return (args, kwargs)
class Bcfg2TestCase(unittest.TestCase):
def assertXMLEqual(self, el1, el2, msg=None):
self.assertEqual(el1.tag, el2.tag, msg=msg)
self.assertEqual(el1.text, el2.text, msg=msg)
self.assertItemsEqual(el1.attrib, el2.attrib, msg=msg)
self.assertEqual(len(el1.getchildren()),
len(el2.getchildren()))
for child1 in el1.getchildren():
cname = child1.get("name")
self.assertIsNotNone(cname,
msg="Element %s has no 'name' attribute" %
child1.tag)
children2 = el2.xpath("*[@name='%s']" % cname)
self.assertEqual(len(children2), 1,
msg="More than one element named %s" % cname)
self.assertXMLEqual(child1, children2[0], msg=msg)
class DBModelTestCase(Bcfg2TestCase):
models = []
@unittest.skipUnless(has_django, "Django not found, skipping")
def test_syncdb(self):
# create the test database
setup_environ(Bcfg2.settings)
from django.core.management.commands import syncdb
cmd = syncdb.Command()
cmd.handle_noargs(interactive=False)
self.assertTrue(os.path.exists(Bcfg2.settings.DATABASE_NAME))
@unittest.skipUnless(has_django, "Django not found, skipping")
def test_cleandb(self):
""" ensure that we a) can connect to the database; b) start with a
clean database """
for model in self.models:
model.objects.all().delete()
self.assertItemsEqual(list(model.objects.all()), [])
def syncdb(modeltest):
inst = modeltest(methodName='test_syncdb')
inst.test_syncdb()
inst.test_cleandb()
|
