/usr/lib/python2.7/dist-packages/nss_cache/sources/consulsource_test.py is in nsscache 0.33-3.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 | """An implementation of a mock consul data source for nsscache."""
__author__ = 'hexedpackets@gmail.com (William Huba)'
import StringIO
import unittest
from nss_cache.maps import group
from nss_cache.maps import passwd
from nss_cache.sources import consulsource
class TestConsulSource(unittest.TestCase):
def setUp(self):
"""Initialize a basic config dict."""
super(TestConsulSource, self).setUp()
self.config = {
'passwd_url': 'PASSWD_URL',
'group_url': 'GROUP_URL',
'datacenter': 'TEST_DATACENTER',
'token': 'TEST_TOKEN',
}
def testDefaultConfiguration(self):
source = consulsource.ConsulFilesSource({})
self.assertEquals(source.conf['datacenter'],
consulsource.ConsulFilesSource.DATACENTER)
self.assertEquals(source.conf['token'],
consulsource.ConsulFilesSource.TOKEN)
def testOverrideDefaultConfiguration(self):
source = consulsource.ConsulFilesSource(self.config)
self.assertEquals(source.conf['datacenter'], 'TEST_DATACENTER')
self.assertEquals(source.conf['token'], 'TEST_TOKEN')
self.assertEquals(source.conf['passwd_url'], 'PASSWD_URL?recurse&token=TEST_TOKEN&dc=TEST_DATACENTER')
self.assertEquals(source.conf['group_url'], 'GROUP_URL?recurse&token=TEST_TOKEN&dc=TEST_DATACENTER')
class TestPasswdMapParser(unittest.TestCase):
def setUp(self):
"""Set some default avalible data for testing."""
self.good_entry = passwd.PasswdMapEntry()
self.good_entry.name = 'foo'
self.good_entry.passwd = 'x'
self.good_entry.uid = 10
self.good_entry.gid = 10
self.good_entry.gecos = 'How Now Brown Cow'
self.good_entry.dir = '/home/foo'
self.good_entry.shell = '/bin/bash'
self.parser = consulsource.ConsulPasswdMapParser()
def testGetMap(self):
passwd_map = passwd.PasswdMap()
cache_info = StringIO.StringIO('''[
{"Key": "org/users/foo/uid", "Value": "MTA="},
{"Key": "org/users/foo/gid", "Value": "MTA="},
{"Key": "org/users/foo/home", "Value": "L2hvbWUvZm9v"},
{"Key": "org/users/foo/shell", "Value": "L2Jpbi9iYXNo"},
{"Key": "org/users/foo/comment", "Value": "SG93IE5vdyBCcm93biBDb3c="},
{"Key": "org/users/foo/subkey/irrelevant_key", "Value": "YmFjb24="}
]''')
self.parser.GetMap(cache_info, passwd_map)
self.assertEquals(self.good_entry, passwd_map.PopItem())
def testReadEntry(self):
data = {'uid': '10', 'gid': '10', 'comment': 'How Now Brown Cow', 'shell': '/bin/bash', 'home': '/home/foo', 'passwd': 'x'}
entry = self.parser._ReadEntry('foo', data)
self.assertEquals(self.good_entry, entry)
def testDefaultEntryValues(self):
data = {'uid': '10', 'gid': '10'}
entry = self.parser._ReadEntry('foo', data)
self.assertEquals(entry.shell, '/bin/bash')
self.assertEquals(entry.dir, '/home/foo')
self.assertEquals(entry.gecos, '')
self.assertEquals(entry.passwd, 'x')
def testInvalidEntry(self):
data = {'irrelevant_key': 'bacon'}
entry = self.parser._ReadEntry('foo', data)
self.assertEquals(entry, None)
class TestConsulGroupMapParser(unittest.TestCase):
def setUp(self):
self.good_entry = group.GroupMapEntry()
self.good_entry.name = 'foo'
self.good_entry.passwd = 'x'
self.good_entry.gid = 10
self.good_entry.members = ['foo', 'bar']
self.parser = consulsource.ConsulGroupMapParser()
def testGetMap(self):
group_map = group.GroupMap()
cache_info = StringIO.StringIO('''[
{"Key": "org/groups/foo/gid", "Value": "MTA="},
{"Key": "org/groups/foo/members", "Value": "Zm9vCmJhcg=="},
{"Key": "org/groups/foo/subkey/irrelevant_key", "Value": "YmFjb24="}
]''')
self.parser.GetMap(cache_info, group_map)
self.assertEquals(self.good_entry, group_map.PopItem())
def testReadEntry(self):
data = {'passwd': 'x', 'gid': '10', 'members': 'foo\nbar'}
entry = self.parser._ReadEntry('foo', data)
self.assertEquals(self.good_entry, entry)
def testDefaultPasswd(self):
data = {'gid': '10', 'members': 'foo\nbar'}
entry = self.parser._ReadEntry('foo', data)
self.assertEquals(self.good_entry, entry)
def testNoMembers(self):
data = {'gid': '10', 'members': ''}
entry = self.parser._ReadEntry('foo', data)
self.assertEquals(entry.members, [''])
def testInvalidEntry(self):
data = {'irrelevant_key': 'bacon'}
entry = self.parser._ReadEntry('foo', data)
self.assertEquals(entry, None)
if __name__ == '__main__':
unittest.main()
|