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
|
#!/usr/bin/python
# unit tests for group functionality in func
import os
import func.overlord.client as fc
import ConfigParser
GROUP_TEST="/tmp/func.test.groups"
class GroupFileBuilder(object):
def __init__(self, filename=GROUP_TEST):
self.filename = filename
if os.access(self.filename, os.R_OK):
os.unlink(self.filename)
self.cp = ConfigParser.SafeConfigParser()
def create(self, dict):
# dict is a dict of section names, whose values
# are a list of tuples of key=value
# aka foo = {'section1':[(key1, value1), (key2, value2)],
# "section2":[(key3, value3)]}
for section in dict.keys():
self.cp.add_section(section)
for (option, value) in dict[section]:
self.cp.set(section, option, value)
fo = open(self.filename, "a+")
self.cp.write(fo)
class GroupsBase(object):
def __init__(self):
self.minions = fc.Minions("*", groups_file=GROUP_TEST)
def test_expand_servers(self):
result = self.minions.get_urls()
print result
def test_get_groups(self):
result = self.minions.group_class.get_groups()
print result
class Groups(GroupsBase):
def get_hosts_by_group_goo(self, group_goo):
group_dict = fc.get_groups()
hosts = fc.get_hosts_by_groupgoo(group_dict, group_goo)
print hosts
class TestGroups(Groups):
def __init__(self):
self.minions = fc.Minions("@blippy", groups_file=GROUP_TEST)
def setUp(self):
self.gfb = GroupFileBuilder()
self.gfb.create({'blippy':[('host', 'localhost')]})
def test_get_host_by_group_goo(self):
results = self.minions.get_urls()
print results
# FIXME: comment this out till I setup a way to test with a speciic
# test config -akl
# def test_get_hosts_by_groupgoo(self):
# group_dict = fc.get_groups()
# hosts = fc.get_hosts_by_groupgoo(group_dict, "@blippy")
# print hosts
|