Revision 44fbd23b test/ganeti.runtime_unittest.py
| b/test/ganeti.runtime_unittest.py | ||
|---|---|---|
| 25 | 25 |
from ganeti import runtime |
| 26 | 26 |
|
| 27 | 27 |
import testutils |
| 28 |
import unittest |
|
| 28 | 29 |
|
| 29 | 30 |
|
| 30 | 31 |
class _EntStub: |
| ... | ... | |
| 50 | 51 |
constants.RAPI_GROUP: _EntStub(gid=2), |
| 51 | 52 |
constants.DAEMONS_GROUP: _EntStub(gid=3), |
| 52 | 53 |
constants.ADMIN_GROUP: _EntStub(gid=4), |
| 54 |
constants.NODED_GROUP: _EntStub(gid=5), |
|
| 53 | 55 |
} |
| 54 | 56 |
return groups[group] |
| 55 | 57 |
|
| ... | ... | |
| 67 | 69 |
raise errors.ConfigurationError("No entries")
|
| 68 | 70 |
|
| 69 | 71 |
|
| 70 |
class TestErrors(testutils.GanetiTestCase):
|
|
| 71 |
def testEverythingSuccessful(self):
|
|
| 72 |
resolver = runtime.GetentResolver(_getpwnam=_StubGetpwnam, |
|
| 73 |
_getgrnam=_StubGetgrnam) |
|
| 72 |
class TestErrors(unittest.TestCase):
|
|
| 73 |
def setUp(self):
|
|
| 74 |
self.resolver = runtime.GetentResolver(_getpwnam=_StubGetpwnam,
|
|
| 75 |
_getgrnam=_StubGetgrnam)
|
|
| 74 | 76 |
|
| 75 |
self.assertEqual(resolver.masterd_uid, |
|
| 77 |
def testEverythingSuccessful(self): |
|
| 78 |
self.assertEqual(self.resolver.masterd_uid, |
|
| 76 | 79 |
_StubGetpwnam(constants.MASTERD_USER).pw_uid) |
| 77 |
self.assertEqual(resolver.masterd_gid, |
|
| 80 |
self.assertEqual(self.resolver.masterd_gid,
|
|
| 78 | 81 |
_StubGetgrnam(constants.MASTERD_GROUP).gr_gid) |
| 79 |
self.assertEqual(resolver.confd_uid, |
|
| 82 |
self.assertEqual(self.resolver.confd_uid,
|
|
| 80 | 83 |
_StubGetpwnam(constants.CONFD_USER).pw_uid) |
| 81 |
self.assertEqual(resolver.confd_gid, |
|
| 84 |
self.assertEqual(self.resolver.confd_gid,
|
|
| 82 | 85 |
_StubGetgrnam(constants.CONFD_GROUP).gr_gid) |
| 83 |
self.assertEqual(resolver.rapi_uid, |
|
| 86 |
self.assertEqual(self.resolver.rapi_uid,
|
|
| 84 | 87 |
_StubGetpwnam(constants.RAPI_USER).pw_uid) |
| 85 |
self.assertEqual(resolver.rapi_gid, |
|
| 88 |
self.assertEqual(self.resolver.rapi_gid,
|
|
| 86 | 89 |
_StubGetgrnam(constants.RAPI_GROUP).gr_gid) |
| 87 |
self.assertEqual(resolver.noded_uid, |
|
| 90 |
self.assertEqual(self.resolver.noded_uid,
|
|
| 88 | 91 |
_StubGetpwnam(constants.NODED_USER).pw_uid) |
| 89 | 92 |
|
| 90 |
self.assertEqual(resolver.daemons_gid, |
|
| 93 |
self.assertEqual(self.resolver.daemons_gid,
|
|
| 91 | 94 |
_StubGetgrnam(constants.DAEMONS_GROUP).gr_gid) |
| 92 |
self.assertEqual(resolver.admin_gid, |
|
| 95 |
self.assertEqual(self.resolver.admin_gid,
|
|
| 93 | 96 |
_StubGetgrnam(constants.ADMIN_GROUP).gr_gid) |
| 94 | 97 |
|
| 95 | 98 |
def testUserNotFound(self): |
| ... | ... | |
| 104 | 107 |
self.assertRaises(errors.ConfigurationError, runtime.GetEnts, |
| 105 | 108 |
resolver=ResolverStubRaising) |
| 106 | 109 |
|
| 110 |
def testLookupForUser(self): |
|
| 111 |
master_stub = _StubGetpwnam(constants.MASTERD_USER) |
|
| 112 |
rapi_stub = _StubGetpwnam(constants.RAPI_USER) |
|
| 113 |
self.assertEqual(self.resolver.LookupUid(master_stub.pw_uid), |
|
| 114 |
constants.MASTERD_USER) |
|
| 115 |
self.assertEqual(self.resolver.LookupUid(rapi_stub.pw_uid), |
|
| 116 |
constants.RAPI_USER) |
|
| 117 |
self.assertEqual(self.resolver.LookupUser(constants.MASTERD_USER), |
|
| 118 |
master_stub.pw_uid) |
|
| 119 |
self.assertEqual(self.resolver.LookupUser(constants.RAPI_USER), |
|
| 120 |
rapi_stub.pw_uid) |
|
| 121 |
|
|
| 122 |
def testLookupForGroup(self): |
|
| 123 |
master_stub = _StubGetgrnam(constants.MASTERD_GROUP) |
|
| 124 |
rapi_stub = _StubGetgrnam(constants.RAPI_GROUP) |
|
| 125 |
self.assertEqual(self.resolver.LookupGid(master_stub.gr_gid), |
|
| 126 |
constants.MASTERD_GROUP) |
|
| 127 |
self.assertEqual(self.resolver.LookupGid(rapi_stub.gr_gid), |
|
| 128 |
constants.RAPI_GROUP) |
|
| 129 |
|
|
| 130 |
def testLookupForUserNotFound(self): |
|
| 131 |
self.assertRaises(errors.ConfigurationError, self.resolver.LookupUid, 9999) |
|
| 132 |
self.assertRaises(errors.ConfigurationError, |
|
| 133 |
self.resolver.LookupUser, "does-not-exist-foo") |
|
| 134 |
|
|
| 135 |
def testLookupForGroupNotFound(self): |
|
| 136 |
self.assertRaises(errors.ConfigurationError, self.resolver.LookupGid, 9999) |
|
| 137 |
self.assertRaises(errors.ConfigurationError, |
|
| 138 |
self.resolver.LookupGroup, "does-not-exist-foo") |
|
| 139 |
|
|
| 107 | 140 |
|
| 108 | 141 |
if __name__ == "__main__": |
| 109 | 142 |
testutils.GanetiTestProgram() |
Also available in: Unified diff