reviewed list options parsing
[tests.git] / system / config_default.py
1 # Thierry Parmentelat <thierry.parmentelat@inria.fr>
2 # Copyright (C) 2010 INRIA 
3 #
4 # a configuration module is expected:
5 # (*) to define a config method
6 # (*) that takes two arguments
7 #     (**) the current set of plc_specs as output by the preceding config modules
8 #     (**) TestMain options field
9 # (*) and that returns the new set of plc_specs
10
11 # values like 'hostname', 'ip' and the like are rewritten later with a TestPool object
12
13 def nodes(options,index):
14     return [{'name':'node%d'%index,
15              'node_fields': {'hostname': 'deferred-nodename%d'%index,
16                              'model':'qemu/minhw', } ,
17              'host_box': 'deferred-node-hostbox-%d'%index,
18              'owner' : 'pi',
19              'nodegroups' : 'mynodegroup',
20              'interface_fields': { 'method':'static',
21                                    'type':'ipv4',
22                                    'ip':'xxx-deferred-xxx',
23                                    'gateway':'xxx-deferred-xxx',
24                                    'network':'xxx-deferred-xxx',
25                                    'broadcast':'xxx-deferred-xxx',
26                                    'netmask':'xxx-deferred-xxx',
27                                    'dns1': 'xxx-deferred-xxx',
28                                    'dns2': 'xxx-deferred-xxx',
29                                    },
30              }]
31
32 def all_nodenames (options,index):
33     return [ node['name'] for node in nodes(options,index)]
34
35 def users (options) :
36     domain="onelab.eu"
37     return [ {'name' : 'pi', 'keynames' : [ 'key1' ],
38               'user_fields' : {'first_name':'PI', 'last_name':'PI',
39                                'enabled':'True',
40                                'email':'fake-pi1@%s'%domain,
41                                'password':'testpi'},
42               'roles':['pi']},
43              {'name' : 'tech', 'keynames' : [ 'key1' ],
44               'user_fields' : {'first_name':'Tech', 'last_name':'Tech',
45                                'enabled':'true',
46                                'email':'fake-tech1@%s'%domain,
47                                'password':'testtech'},
48               'roles':['tech']},
49              {'name':'user', 'keynames' : [ 'key1' ],
50               'user_fields' : {'first_name':'User', 'last_name':'User',
51                                'enabled':'true',
52                                'email':'fake-user1@%s'%domain,
53                                'password':'testuser'},
54               'roles':['user']},
55              {'name':'techuser', 'keynames' : [ 'key1' ],
56               'user_fields' : {'first_name':'UserTech', 'last_name':'UserTech',
57                                'enabled':'true',
58                                'email':'fake-tech2@%s'%domain,
59                                'password':'testusertech'},
60               'roles':['tech','user']},
61              {'name':'pitech', 'keynames' : [ 'key1' ],
62               'user_fields' : {'first_name':'PiTech',
63                                'last_name':'PiTech',
64                                'enabled':'true',
65                                'email':'fake-pi2@%s'%domain,
66                                'password':'testusertech'},
67               'roles':['pi','tech']},
68              {'name':'admin', 'keynames' : [ 'key1' ],
69               'user_fields' : {'first_name':'Admin',
70                                'last_name':'Admin',
71                                'enabled':'true',
72                                'email':'admin@%s'%domain,
73                                'password':'testuseradmin'},
74               'roles':['admin']},
75              ]
76
77 def all_usernames (options):
78     return [ user['name'] for user in users(options)]
79
80 def sites (options,index):
81     return [ {'site_fields' : {'name':'mainsite',
82                                'login_base':'main',
83                                'abbreviated_name':'PLanettest',
84                                'max_slices':100,
85                                'url':'http://test.onelab.eu',
86                                },
87               'address_fields' : {'line1':'route des lucioles',
88                                   'city':'sophia',
89                                   'state':'fr',
90                                   'postalcode':'06600',
91                                   'country':'france',
92                                   },
93               'users' : users(options),
94               'nodes': nodes(options,index),
95             }]
96
97 ##########
98 public_key="""ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEA4jNj8yT9ieEc6nSJz/ESu4fui9WrJ2y/MCfqIZ5WcdVKhBFUYyIenmUaeTduMcSqvoYRQ4QnFR1BFdLG8XR9D6FWZ5zTKUgpkew22EVNeqai4IXeWYKyt1Qf3ehaz9E3o1PG/bmQNIM6aQay6TD1Y4lqXI+eTVXVQev4K2fixySjFQpp9RB4UHbeA8c28yoa/cgAYHqCqlvm9uvpGMjgm/Qa4M+ZeO7NdjowfaF/wF4BQIzVFN9YRhvQ/d8WDz84B5Pr0J7pWpaX7EyC4bvdskxl6kmdNIwIRcIe4OcuIiX5Z9oO+7h/chsEVJWF4vqNIYlL9Zvyhnr0hLLhhuk2bw== root@test.onelab.eu
99 """
100 private_key="""-----BEGIN RSA PRIVATE KEY-----
101 MIIEogIBAAKCAQEA4jNj8yT9ieEc6nSJz/ESu4fui9WrJ2y/MCfqIZ5WcdVKhBFU
102 YyIenmUaeTduMcSqvoYRQ4QnFR1BFdLG8XR9D6FWZ5zTKUgpkew22EVNeqai4IXe
103 WYKyt1Qf3ehaz9E3o1PG/bmQNIM6aQay6TD1Y4lqXI+eTVXVQev4K2fixySjFQpp
104 9RB4UHbeA8c28yoa/cgAYHqCqlvm9uvpGMjgm/Qa4M+ZeO7NdjowfaF/wF4BQIzV
105 FN9YRhvQ/d8WDz84B5Pr0J7pWpaX7EyC4bvdskxl6kmdNIwIRcIe4OcuIiX5Z9oO
106 +7h/chsEVJWF4vqNIYlL9Zvyhnr0hLLhhuk2bwIBIwKCAQATY32Yf4NyN93oNd/t
107 QIyTuzG0NuLI3W95J/4gI4PAnUDmv6glwRiRO92ynlnnAjqFW/LZ5sGFd4k8YoYU
108 sjaa8JJgpwrJBi9y6Fx47/9Tp+ITPqyoliyTXvtqysX0jkaY+I1mNHoTITDkbknZ
109 eTma0UOhiKcrMz4qOMwg+kajWsAhIplJXyf0Mio/XuyqjMT4wI/NyGZQ4bGuUjO7
110 gj3p+9psOvONsRBW4MV27W5ts9c7HEXg+VJ2PSCEMs+uyzXcdnMJcTb4zQ/+tVxR
111 5IMeEuR9ZzDNkDtNF6Nnw5kYcTBNoayzZbUfjcuSmsMklMXr0qJ4qcW9/ONKgBQ9
112 6qhDAoGBAPkvSYuF/bxwatEiUKyF97oGDe025h/58aqK1VBD5/BBVqqvbQOeNtR5
113 /LerGfFa5D9Jm+6U97gDdq3tH0j95Mo0F00LWq2+vp7U4DTQsiddepzNdbcvSrzT
114 NVZ2cnOAlKTHO4hGggShm04n/M5LOzkHtI5TLcIJjw4b5iiIw9EtAoGBAOhjLTds
115 Zz8UbXVTeGv8yBGhnjAeHQ5WISN6V5KenB4GIyaYCCcQUOUGqc5nCttlnPLv/GHp
116 4DOJ2/0KbwDEwk7HbAtXG2Tv1OkmfcOq9RH19V9lyqynA+zvI6taisCEaMvBlafd
117 k+RgXsR+NdLs96RakKt4BtgpuuADoSIryQ+LAoGBAKremNSzpq0Z4tiMjxc3Ssiz
118 scc7lnxTnmZQkkWq6C+3xmZpqdaYYByra3ahNlxblTK2IcgroozPLM8I/4KCNnwG
119 dmC3VB9eOZF8B3SsXOfLEj+i1GBa8WuJg6kAw4JmzFO70Qz9JfSMErk//c9Jh7IT
120 6YYqaIUN3nATIXrhcFTrAoGAVlC5BfUQZ+MEFaKpEiqwthd1RRJ/0h/9rhd/nNvT
121 lh+JZhs2OmUlXGGPhy2WUX2DcC1AfCOrC9Qego7YxcVsvizQW/vIWLDaDXSyXp6V
122 ilQKrmejDO2Tvmdzpguv4Rs83fdyGcdUMEENQas4kCwhd49aTlEnHRbQYdV2XSY0
123 vKECgYEAlhYzfSswIF2h5/hGDLETxgNJ2kD0HIZYh7aud6X6aEYNdJopbfbEhifU
124 vTbf8GtvERDoxWEsk9Qp7km8xXfKWdcZtqIwsSmn/ri5d7iyvpIk591YIHSY0dr2
125 BO+VyPNWF+kDNI8mSUwi7jLW6liMdhNOmDaSX0+0X8CHtK898xM=
126 -----END RSA PRIVATE KEY-----
127 """
128
129 def keys (options,index):
130     return [ {'name': 'key1',
131               'private' : private_key,
132               'key_fields' : {'key_type':'ssh',
133                               'key': public_key}}
134              ]
135
136 def initscripts(options,index): 
137     initscripts= [ { 'initscript_fields' : 
138                      { 'enabled' : True,
139                        'name':'script1',
140                        'script' : """#! /bin/sh
141 (echo Starting test initscript: script1; date) >> /tmp/script1.stamp
142 # expected to be 'start'
143 command=$1; shift
144 # get slice name
145 slicename=$1; shift
146 echo "This is the stdout of the sliver $slicename initscript $command (exp. start) pid=$$" 
147 echo "This is the stderr of the sliver $slicename initscript $command (exp. start) pid=$$" 1>&2
148 """,
149                        }},
150                    { 'initscript_fields' : 
151                      { 'enabled' : True,
152                        'name':'script2',
153                        'script' : """#! /bin/sh
154 (echo Starting loop-forever test initscript: script2; date) >> /tmp/script2.stamp
155 # expected to be 'start'
156 command=$1; shift
157 # get slice name
158 slicename=$1; shift
159 while true; do
160 echo "This is the stdout of the loop-for-ever sliver initscript $slicename $command (exp. start) $$" 
161 echo "This is the stderr of the loop-for-ever sliver initscript $slicename $command (exp. start) $$" 1>&2
162 sleep 10
163 done
164 """,
165                        }},
166                    ]
167     return initscripts
168
169 def slices (options,index):
170     return [ { 'slice_fields': {'name':'main_pslc%d'%i,
171                                 'instantiation':'plc-instantiated',
172                                 'url':'http://foo.com',
173                                 'description':'testslice number %d'%i,
174                                 'max_nodes':2,
175                                 },
176                'usernames' : [ 'pi','tech','techuser' ],
177                'nodenames' : all_nodenames(options,index),
178                'initscriptname' : 'script%d'%i,
179                'sitename' : 'main',
180                'owner' : 'pi',
181                } for i in range (2*index-1,2*index+1) ]
182
183 def all_slicenames (options,index):
184     return [ slice['slice_fields']['name'] for slice in slices(options,index)]
185
186 def tcp_tests (options,index):
187     if index == 1:
188         return [
189             # local test
190             { 'server_node': 'node1',
191               'server_slice' : 'main_pslc1',
192               'client_node' : 'node1',
193               'client_slice' : 'main_pslc1',
194               'port' : 2000,
195               }]
196     elif index == 2:
197         return [
198             # remote test
199             { 'server_node': 'node1',
200               'server_slice' : 'main_pslc1',
201               'client_node' : 'node2',
202               'client_slice' : 'main_pslc2',
203               'port' : 4000,
204               },
205             ]
206     else:
207         return []
208
209 # the semantic for 't_from' and 't_until' here is:
210 # if they are smaller than one year, they are relative to the current time, expressed in grains
211 # otherwise they are absolute
212 def leases (options, index):
213     leases=[]
214     counter=0
215     slices=all_slicenames(options,index)
216     slice_sequence = slices[:1] + slices + [None,]
217     for iterator in range(12):
218         for slice in slice_sequence:
219             leases.append ( {'slice' : slice, 't_from':counter,'t_until':counter+1} )
220             counter += 1
221     return leases
222
223 def plc (options,index) :
224     return { 
225         'name' : 'onetest%d'%index,
226         # as of yet, not sure we can handle foreign hosts, but this is required though
227         'hostname' : 'deferred-myplc-hostbox-%d'%index,
228         # set these two items to run within a vserver
229         'vservername': 'deferred-vservername',
230         'vserverip': 'deferred-vserverip',
231         'role' : 'root',
232         'PLC_ROOT_USER' : 'root@test.onelab.eu',
233         'PLC_ROOT_PASSWORD' : 'test++',
234         'PLC_NAME' : 'Regression TestLab',
235         'PLC_SHORTNAME' : 'Rlab',
236         'PLC_MAIL_ENABLED':'false',
237         'PLC_MAIL_SUPPORT_ADDRESS' : 'thierry.parmentelat@sophia.inria.fr',
238         'PLC_DB_HOST' : 'deferred-myplc-hostname',
239         'PLC_DB_PASSWORD' : 'mnbvcxzlkjhgfdsapoiuytrewq',
240         'PLC_API_HOST' : 'deferred-myplc-hostname',
241         'PLC_WWW_HOST' : 'deferred-myplc-hostname',
242         'PLC_BOOT_HOST' : 'deferred-myplc-hostname',
243         'PLC_NET_DNS1' : 'deferred-dns-1',
244         'PLC_NET_DNS2' : 'deferred-dns-2',
245         'PLC_RESERVATION_GRANULARITY':1800,
246         'PLC_OMF_ENABLED' : 'false',
247         'sites' : sites(options,index),
248         'keys' : keys(options,index),
249         'initscripts': initscripts(options,index),
250         'slices' : slices(options,index),
251         'tcp_test' : tcp_tests(options,index),
252         'sfa' : sfa(options,index),
253         'leases' : leases (options, index),
254     }
255
256 def sfa (options,index) :
257     if index==1:
258         root_auth='plc'
259     else:
260         root_auth='ple'
261     return { 
262         'SFA_REGISTRY_ROOT_AUTH' : root_auth,
263         'SFA_REGISTRY_LEVEL1_AUTH' : '',
264         'SFA_REGISTRY_HOST' : 'deferred-myplc-hostname',
265         'SFA_AGGREGATE_HOST': 'deferred-myplc-hostname',
266         'SFA_SM_HOST': 'deferred-myplc-hostname',
267         'SFA_PLC_USER' : 'root@test.onelab.eu',
268         'SFA_PLC_PASSWORD' : 'test++',
269         'SFA_PLC_DB_HOST':'deferred-myplc-hostname',
270         'SFA_PLC_DB_USER' : 'pgsqluser',
271         'SFA_PLC_DB_PASSWORD' : 'mnbvcxzlkjhgfdsapoiuytrewq',
272         'SFA_PLC_URL' : 'deferred-myplc-api-url',
273         'slices_sfa' : slices_sfa(options,index),
274         'sfa_slice_xml' : sfa_slice_xml(options,index),
275         'sfa_person_xml' : sfa_person_xml(options,index),
276         'sfa_slice_rspec' : sfa_slice_rspec(options,index)
277     }
278
279 def slices_sfa (options,index):
280     return [ { 'slice_fields': {'name':'main_fslc1',
281                                 'url':'http://foo%d@foo.com'%index,
282                                 'description':'SFA-testing',
283                                 'max_nodes':2,
284                                 },
285                'usernames' : [ ('sfafakeuser1','key1') ],
286                'nodenames' : all_nodenames(options,index),
287                'sitename' : 'main',
288                }]
289
290 def sfa_slice_xml(options,index):
291     if index==1:
292         hrn='plc.main.fslc1'
293         researcher='plc.main.fake-pi1'
294     else:
295         hrn='ple.main.fslc1'
296         researcher='ple.main.fake-pi1'
297
298     return  ['<record hrn="%s" type="slice" description="SFA-testing" url="http://test.onelab.eu/"><researcher>%s</researcher></record>'%(hrn, researcher)]
299
300 def sfa_person_xml(options,index):
301     if index==1:
302         hrn='plc.main.sfafakeuser1'
303     else:
304         hrn='ple.main.sfafakeuser1'
305
306     return ['<record email="sfafakeuser1@onelab.eu" enabled="True" first_name="Fake" hrn="%s" last_name="Sfa" name="%s" type="user"><keys>%s</keys><role_ids>20</role_ids><role_ids>10</role_ids><site_ids>1</site_ids><roles>pi</roles><roles>admin</roles><sites>plc.main</sites></record>'%(hrn,hrn,public_key)]
307
308 def sfa_slice_rspec(options,index):
309     node_name='deferred'
310     if index==1:
311        netspec_name='\"plc\"'
312     else:
313        netspec_name='\"ple\"'
314
315     return { 
316         'part1' : '<?xml version="1.0" ?><Rspec><networks><NetSpec name=',
317         'part2' : '%s'%netspec_name,
318         'part3' : '><nodes><NodeSpec cpu_min="" cpu_pct="" cpu_share="" disk_max="" init_params="" name=\"',
319         'part4' : '%s'%node_name,
320         'part5' : '\" start_time="" type=""><net_if><IfSpec init_params="" ip_spoof="" max_kbyte="" max_rate="" min_rate="" name="True" type="ipv4"/></net_if></NodeSpec></nodes></NetSpec></networks></Rspec>',
321            }
322              
323 def config (plc_specs,options):
324     result=plc_specs
325     for i in range (options.size):
326         result.append(plc(options,i+1))
327     return result