Merge lp:~gnuoy/charms/trusty/nova-compute/1453940 into lp:~openstack-charmers-archive/charms/trusty/nova-compute/next
- Trusty Tahr (14.04)
- 1453940
- Merge into next
Status: | Merged |
---|---|
Merged at revision: | 157 |
Proposed branch: | lp:~gnuoy/charms/trusty/nova-compute/1453940 |
Merge into: | lp:~openstack-charmers-archive/charms/trusty/nova-compute/next |
Diff against target: |
1252 lines (+866/-91) 7 files modified
hooks/charmhelpers/contrib/openstack/context.py (+8/-9) hooks/charmhelpers/contrib/storage/linux/ceph.py (+224/-2) hooks/nova_compute_hooks.py (+14/-23) tests/charmhelpers/contrib/amulet/utils.py (+234/-52) tests/charmhelpers/contrib/openstack/amulet/deployment.py (+20/-5) tests/charmhelpers/contrib/openstack/amulet/utils.py (+359/-0) unit_tests/test_nova_compute_hooks.py (+7/-0) |
To merge this branch: | bzr merge lp:~gnuoy/charms/trusty/nova-compute/1453940 |
Related bugs: |
Reviewer | Review Type | Date Requested | Status |
---|---|---|---|
Edward Hope-Morley | Approve | ||
Review via email: mp+269377@code.launchpad.net |
Commit message
Description of the change
uosci-testing-bot (uosci-testing-bot) wrote : | # |
uosci-testing-bot (uosci-testing-bot) wrote : | # |
charm_unit_test #8830 nova-compute-next for gnuoy mp269377
UNIT OK: passed
uosci-testing-bot (uosci-testing-bot) wrote : | # |
charm_amulet_test #6315 nova-compute-next for gnuoy mp269377
AMULET OK: passed
Build: http://
- 159. By Liam Young
-
Charm helper sync
uosci-testing-bot (uosci-testing-bot) wrote : | # |
charm_lint_check #9709 nova-compute-next for gnuoy mp269377
LINT FAIL: lint-test failed
LINT FAIL: charm-proof failed
LINT Results (max last 2 lines):
make: *** [lint] Error 100
ERROR:root:Make target returned non-zero.
Full lint test output: http://
Build: http://
uosci-testing-bot (uosci-testing-bot) wrote : | # |
charm_unit_test #8941 nova-compute-next for gnuoy mp269377
UNIT OK: passed
uosci-testing-bot (uosci-testing-bot) wrote : | # |
charm_amulet_test #6337 nova-compute-next for gnuoy mp269377
AMULET FAIL: amulet-test failed
AMULET Results (max last 2 lines):
make: *** [test] Error 1
ERROR:root:Make target returned non-zero.
Full amulet test output: http://
Build: http://
Edward Hope-Morley (hopem) wrote : | # |
LGTM +1 (think amulet fail is unrelated)
Preview Diff
1 | === modified file 'hooks/charmhelpers/contrib/openstack/context.py' | |||
2 | --- hooks/charmhelpers/contrib/openstack/context.py 2015-09-02 15:03:16 +0000 | |||
3 | +++ hooks/charmhelpers/contrib/openstack/context.py 2015-09-10 09:35:32 +0000 | |||
4 | @@ -485,13 +485,15 @@ | |||
5 | 485 | 485 | ||
6 | 486 | log('Generating template context for ceph', level=DEBUG) | 486 | log('Generating template context for ceph', level=DEBUG) |
7 | 487 | mon_hosts = [] | 487 | mon_hosts = [] |
11 | 488 | auth = None | 488 | ctxt = { |
12 | 489 | key = None | 489 | 'use_syslog': str(config('use-syslog')).lower() |
13 | 490 | use_syslog = str(config('use-syslog')).lower() | 490 | } |
14 | 491 | for rid in relation_ids('ceph'): | 491 | for rid in relation_ids('ceph'): |
15 | 492 | for unit in related_units(rid): | 492 | for unit in related_units(rid): |
18 | 493 | auth = relation_get('auth', rid=rid, unit=unit) | 493 | if not ctxt.get('auth'): |
19 | 494 | key = relation_get('key', rid=rid, unit=unit) | 494 | ctxt['auth'] = relation_get('auth', rid=rid, unit=unit) |
20 | 495 | if not ctxt.get('key'): | ||
21 | 496 | ctxt['key'] = relation_get('key', rid=rid, unit=unit) | ||
22 | 495 | ceph_pub_addr = relation_get('ceph-public-address', rid=rid, | 497 | ceph_pub_addr = relation_get('ceph-public-address', rid=rid, |
23 | 496 | unit=unit) | 498 | unit=unit) |
24 | 497 | unit_priv_addr = relation_get('private-address', rid=rid, | 499 | unit_priv_addr = relation_get('private-address', rid=rid, |
25 | @@ -500,10 +502,7 @@ | |||
26 | 500 | ceph_addr = format_ipv6_addr(ceph_addr) or ceph_addr | 502 | ceph_addr = format_ipv6_addr(ceph_addr) or ceph_addr |
27 | 501 | mon_hosts.append(ceph_addr) | 503 | mon_hosts.append(ceph_addr) |
28 | 502 | 504 | ||
33 | 503 | ctxt = {'mon_hosts': ' '.join(sorted(mon_hosts)), | 505 | ctxt['mon_hosts'] = ' '.join(sorted(mon_hosts)) |
30 | 504 | 'auth': auth, | ||
31 | 505 | 'key': key, | ||
32 | 506 | 'use_syslog': use_syslog} | ||
34 | 507 | 506 | ||
35 | 508 | if not os.path.isdir('/etc/ceph'): | 507 | if not os.path.isdir('/etc/ceph'): |
36 | 509 | os.mkdir('/etc/ceph') | 508 | os.mkdir('/etc/ceph') |
37 | 510 | 509 | ||
38 | === modified file 'hooks/charmhelpers/contrib/storage/linux/ceph.py' | |||
39 | --- hooks/charmhelpers/contrib/storage/linux/ceph.py 2015-07-16 20:18:54 +0000 | |||
40 | +++ hooks/charmhelpers/contrib/storage/linux/ceph.py 2015-09-10 09:35:32 +0000 | |||
41 | @@ -28,6 +28,7 @@ | |||
42 | 28 | import shutil | 28 | import shutil |
43 | 29 | import json | 29 | import json |
44 | 30 | import time | 30 | import time |
45 | 31 | import uuid | ||
46 | 31 | 32 | ||
47 | 32 | from subprocess import ( | 33 | from subprocess import ( |
48 | 33 | check_call, | 34 | check_call, |
49 | @@ -35,8 +36,10 @@ | |||
50 | 35 | CalledProcessError, | 36 | CalledProcessError, |
51 | 36 | ) | 37 | ) |
52 | 37 | from charmhelpers.core.hookenv import ( | 38 | from charmhelpers.core.hookenv import ( |
53 | 39 | local_unit, | ||
54 | 38 | relation_get, | 40 | relation_get, |
55 | 39 | relation_ids, | 41 | relation_ids, |
56 | 42 | relation_set, | ||
57 | 40 | related_units, | 43 | related_units, |
58 | 41 | log, | 44 | log, |
59 | 42 | DEBUG, | 45 | DEBUG, |
60 | @@ -411,17 +414,52 @@ | |||
61 | 411 | 414 | ||
62 | 412 | The API is versioned and defaults to version 1. | 415 | The API is versioned and defaults to version 1. |
63 | 413 | """ | 416 | """ |
65 | 414 | def __init__(self, api_version=1): | 417 | def __init__(self, api_version=1, request_id=None): |
66 | 415 | self.api_version = api_version | 418 | self.api_version = api_version |
67 | 419 | if request_id: | ||
68 | 420 | self.request_id = request_id | ||
69 | 421 | else: | ||
70 | 422 | self.request_id = str(uuid.uuid1()) | ||
71 | 416 | self.ops = [] | 423 | self.ops = [] |
72 | 417 | 424 | ||
73 | 418 | def add_op_create_pool(self, name, replica_count=3): | 425 | def add_op_create_pool(self, name, replica_count=3): |
74 | 419 | self.ops.append({'op': 'create-pool', 'name': name, | 426 | self.ops.append({'op': 'create-pool', 'name': name, |
75 | 420 | 'replicas': replica_count}) | 427 | 'replicas': replica_count}) |
76 | 421 | 428 | ||
77 | 429 | def set_ops(self, ops): | ||
78 | 430 | """Set request ops to provided value. | ||
79 | 431 | |||
80 | 432 | Useful for injecting ops that come from a previous request | ||
81 | 433 | to allow comparisons to ensure validity. | ||
82 | 434 | """ | ||
83 | 435 | self.ops = ops | ||
84 | 436 | |||
85 | 422 | @property | 437 | @property |
86 | 423 | def request(self): | 438 | def request(self): |
88 | 424 | return json.dumps({'api-version': self.api_version, 'ops': self.ops}) | 439 | return json.dumps({'api-version': self.api_version, 'ops': self.ops, |
89 | 440 | 'request-id': self.request_id}) | ||
90 | 441 | |||
91 | 442 | def _ops_equal(self, other): | ||
92 | 443 | if len(self.ops) == len(other.ops): | ||
93 | 444 | for req_no in range(0, len(self.ops)): | ||
94 | 445 | for key in ['replicas', 'name', 'op']: | ||
95 | 446 | if self.ops[req_no][key] != other.ops[req_no][key]: | ||
96 | 447 | return False | ||
97 | 448 | else: | ||
98 | 449 | return False | ||
99 | 450 | return True | ||
100 | 451 | |||
101 | 452 | def __eq__(self, other): | ||
102 | 453 | if not isinstance(other, self.__class__): | ||
103 | 454 | return False | ||
104 | 455 | if self.api_version == other.api_version and \ | ||
105 | 456 | self._ops_equal(other): | ||
106 | 457 | return True | ||
107 | 458 | else: | ||
108 | 459 | return False | ||
109 | 460 | |||
110 | 461 | def __ne__(self, other): | ||
111 | 462 | return not self.__eq__(other) | ||
112 | 425 | 463 | ||
113 | 426 | 464 | ||
114 | 427 | class CephBrokerRsp(object): | 465 | class CephBrokerRsp(object): |
115 | @@ -431,14 +469,198 @@ | |||
116 | 431 | 469 | ||
117 | 432 | The API is versioned and defaults to version 1. | 470 | The API is versioned and defaults to version 1. |
118 | 433 | """ | 471 | """ |
119 | 472 | |||
120 | 434 | def __init__(self, encoded_rsp): | 473 | def __init__(self, encoded_rsp): |
121 | 435 | self.api_version = None | 474 | self.api_version = None |
122 | 436 | self.rsp = json.loads(encoded_rsp) | 475 | self.rsp = json.loads(encoded_rsp) |
123 | 437 | 476 | ||
124 | 438 | @property | 477 | @property |
125 | 478 | def request_id(self): | ||
126 | 479 | return self.rsp.get('request-id') | ||
127 | 480 | |||
128 | 481 | @property | ||
129 | 439 | def exit_code(self): | 482 | def exit_code(self): |
130 | 440 | return self.rsp.get('exit-code') | 483 | return self.rsp.get('exit-code') |
131 | 441 | 484 | ||
132 | 442 | @property | 485 | @property |
133 | 443 | def exit_msg(self): | 486 | def exit_msg(self): |
134 | 444 | return self.rsp.get('stderr') | 487 | return self.rsp.get('stderr') |
135 | 488 | |||
136 | 489 | |||
137 | 490 | # Ceph Broker Conversation: | ||
138 | 491 | # If a charm needs an action to be taken by ceph it can create a CephBrokerRq | ||
139 | 492 | # and send that request to ceph via the ceph relation. The CephBrokerRq has a | ||
140 | 493 | # unique id so that the client can identity which CephBrokerRsp is associated | ||
141 | 494 | # with the request. Ceph will also respond to each client unit individually | ||
142 | 495 | # creating a response key per client unit eg glance/0 will get a CephBrokerRsp | ||
143 | 496 | # via key broker-rsp-glance-0 | ||
144 | 497 | # | ||
145 | 498 | # To use this the charm can just do something like: | ||
146 | 499 | # | ||
147 | 500 | # from charmhelpers.contrib.storage.linux.ceph import ( | ||
148 | 501 | # send_request_if_needed, | ||
149 | 502 | # is_request_complete, | ||
150 | 503 | # CephBrokerRq, | ||
151 | 504 | # ) | ||
152 | 505 | # | ||
153 | 506 | # @hooks.hook('ceph-relation-changed') | ||
154 | 507 | # def ceph_changed(): | ||
155 | 508 | # rq = CephBrokerRq() | ||
156 | 509 | # rq.add_op_create_pool(name='poolname', replica_count=3) | ||
157 | 510 | # | ||
158 | 511 | # if is_request_complete(rq): | ||
159 | 512 | # <Request complete actions> | ||
160 | 513 | # else: | ||
161 | 514 | # send_request_if_needed(get_ceph_request()) | ||
162 | 515 | # | ||
163 | 516 | # CephBrokerRq and CephBrokerRsp are serialized into JSON. Below is an example | ||
164 | 517 | # of glance having sent a request to ceph which ceph has successfully processed | ||
165 | 518 | # 'ceph:8': { | ||
166 | 519 | # 'ceph/0': { | ||
167 | 520 | # 'auth': 'cephx', | ||
168 | 521 | # 'broker-rsp-glance-0': '{"request-id": "0bc7dc54", "exit-code": 0}', | ||
169 | 522 | # 'broker_rsp': '{"request-id": "0da543b8", "exit-code": 0}', | ||
170 | 523 | # 'ceph-public-address': '10.5.44.103', | ||
171 | 524 | # 'key': 'AQCLDttVuHXINhAAvI144CB09dYchhHyTUY9BQ==', | ||
172 | 525 | # 'private-address': '10.5.44.103', | ||
173 | 526 | # }, | ||
174 | 527 | # 'glance/0': { | ||
175 | 528 | # 'broker_req': ('{"api-version": 1, "request-id": "0bc7dc54", ' | ||
176 | 529 | # '"ops": [{"replicas": 3, "name": "glance", ' | ||
177 | 530 | # '"op": "create-pool"}]}'), | ||
178 | 531 | # 'private-address': '10.5.44.109', | ||
179 | 532 | # }, | ||
180 | 533 | # } | ||
181 | 534 | |||
182 | 535 | def get_previous_request(rid): | ||
183 | 536 | """Return the last ceph broker request sent on a given relation | ||
184 | 537 | |||
185 | 538 | @param rid: Relation id to query for request | ||
186 | 539 | """ | ||
187 | 540 | request = None | ||
188 | 541 | broker_req = relation_get(attribute='broker_req', rid=rid, | ||
189 | 542 | unit=local_unit()) | ||
190 | 543 | if broker_req: | ||
191 | 544 | request_data = json.loads(broker_req) | ||
192 | 545 | request = CephBrokerRq(api_version=request_data['api-version'], | ||
193 | 546 | request_id=request_data['request-id']) | ||
194 | 547 | request.set_ops(request_data['ops']) | ||
195 | 548 | |||
196 | 549 | return request | ||
197 | 550 | |||
198 | 551 | |||
199 | 552 | def get_request_states(request): | ||
200 | 553 | """Return a dict of requests per relation id with their corresponding | ||
201 | 554 | completion state. | ||
202 | 555 | |||
203 | 556 | This allows a charm, which has a request for ceph, to see whether there is | ||
204 | 557 | an equivalent request already being processed and if so what state that | ||
205 | 558 | request is in. | ||
206 | 559 | |||
207 | 560 | @param request: A CephBrokerRq object | ||
208 | 561 | """ | ||
209 | 562 | complete = [] | ||
210 | 563 | requests = {} | ||
211 | 564 | for rid in relation_ids('ceph'): | ||
212 | 565 | complete = False | ||
213 | 566 | previous_request = get_previous_request(rid) | ||
214 | 567 | if request == previous_request: | ||
215 | 568 | sent = True | ||
216 | 569 | complete = is_request_complete_for_rid(previous_request, rid) | ||
217 | 570 | else: | ||
218 | 571 | sent = False | ||
219 | 572 | complete = False | ||
220 | 573 | |||
221 | 574 | requests[rid] = { | ||
222 | 575 | 'sent': sent, | ||
223 | 576 | 'complete': complete, | ||
224 | 577 | } | ||
225 | 578 | |||
226 | 579 | return requests | ||
227 | 580 | |||
228 | 581 | |||
229 | 582 | def is_request_sent(request): | ||
230 | 583 | """Check to see if a functionally equivalent request has already been sent | ||
231 | 584 | |||
232 | 585 | Returns True if a similair request has been sent | ||
233 | 586 | |||
234 | 587 | @param request: A CephBrokerRq object | ||
235 | 588 | """ | ||
236 | 589 | states = get_request_states(request) | ||
237 | 590 | for rid in states.keys(): | ||
238 | 591 | if not states[rid]['sent']: | ||
239 | 592 | return False | ||
240 | 593 | |||
241 | 594 | return True | ||
242 | 595 | |||
243 | 596 | |||
244 | 597 | def is_request_complete(request): | ||
245 | 598 | """Check to see if a functionally equivalent request has already been | ||
246 | 599 | completed | ||
247 | 600 | |||
248 | 601 | Returns True if a similair request has been completed | ||
249 | 602 | |||
250 | 603 | @param request: A CephBrokerRq object | ||
251 | 604 | """ | ||
252 | 605 | states = get_request_states(request) | ||
253 | 606 | for rid in states.keys(): | ||
254 | 607 | if not states[rid]['complete']: | ||
255 | 608 | return False | ||
256 | 609 | |||
257 | 610 | return True | ||
258 | 611 | |||
259 | 612 | |||
260 | 613 | def is_request_complete_for_rid(request, rid): | ||
261 | 614 | """Check if a given request has been completed on the given relation | ||
262 | 615 | |||
263 | 616 | @param request: A CephBrokerRq object | ||
264 | 617 | @param rid: Relation ID | ||
265 | 618 | """ | ||
266 | 619 | broker_key = get_broker_rsp_key() | ||
267 | 620 | for unit in related_units(rid): | ||
268 | 621 | rdata = relation_get(rid=rid, unit=unit) | ||
269 | 622 | if rdata.get(broker_key): | ||
270 | 623 | rsp = CephBrokerRsp(rdata.get(broker_key)) | ||
271 | 624 | if rsp.request_id == request.request_id: | ||
272 | 625 | if not rsp.exit_code: | ||
273 | 626 | return True | ||
274 | 627 | else: | ||
275 | 628 | # The remote unit sent no reply targeted at this unit so either the | ||
276 | 629 | # remote ceph cluster does not support unit targeted replies or it | ||
277 | 630 | # has not processed our request yet. | ||
278 | 631 | if rdata.get('broker_rsp'): | ||
279 | 632 | request_data = json.loads(rdata['broker_rsp']) | ||
280 | 633 | if request_data.get('request-id'): | ||
281 | 634 | log('Ignoring legacy broker_rsp without unit key as remote ' | ||
282 | 635 | 'service supports unit specific replies', level=DEBUG) | ||
283 | 636 | else: | ||
284 | 637 | log('Using legacy broker_rsp as remote service does not ' | ||
285 | 638 | 'supports unit specific replies', level=DEBUG) | ||
286 | 639 | rsp = CephBrokerRsp(rdata['broker_rsp']) | ||
287 | 640 | if not rsp.exit_code: | ||
288 | 641 | return True | ||
289 | 642 | |||
290 | 643 | return False | ||
291 | 644 | |||
292 | 645 | |||
293 | 646 | def get_broker_rsp_key(): | ||
294 | 647 | """Return broker response key for this unit | ||
295 | 648 | |||
296 | 649 | This is the key that ceph is going to use to pass request status | ||
297 | 650 | information back to this unit | ||
298 | 651 | """ | ||
299 | 652 | return 'broker-rsp-' + local_unit().replace('/', '-') | ||
300 | 653 | |||
301 | 654 | |||
302 | 655 | def send_request_if_needed(request): | ||
303 | 656 | """Send broker request if an equivalent request has not already been sent | ||
304 | 657 | |||
305 | 658 | @param request: A CephBrokerRq object | ||
306 | 659 | """ | ||
307 | 660 | if is_request_sent(request): | ||
308 | 661 | log('Request already sent but not complete, not sending new request', | ||
309 | 662 | level=DEBUG) | ||
310 | 663 | else: | ||
311 | 664 | for rid in relation_ids('ceph'): | ||
312 | 665 | log('Sending request {}'.format(request.request_id), level=DEBUG) | ||
313 | 666 | relation_set(relation_id=rid, broker_req=request.request) | ||
314 | 445 | 667 | ||
315 | === modified file 'hooks/nova_compute_hooks.py' | |||
316 | --- hooks/nova_compute_hooks.py 2015-08-19 15:07:10 +0000 | |||
317 | +++ hooks/nova_compute_hooks.py 2015-09-10 09:35:32 +0000 | |||
318 | @@ -6,7 +6,6 @@ | |||
319 | 6 | config, | 6 | config, |
320 | 7 | is_relation_made, | 7 | is_relation_made, |
321 | 8 | log, | 8 | log, |
322 | 9 | INFO, | ||
323 | 10 | ERROR, | 9 | ERROR, |
324 | 11 | relation_ids, | 10 | relation_ids, |
325 | 12 | relation_get, | 11 | relation_get, |
326 | @@ -38,8 +37,9 @@ | |||
327 | 38 | from charmhelpers.contrib.storage.linux.ceph import ( | 37 | from charmhelpers.contrib.storage.linux.ceph import ( |
328 | 39 | ensure_ceph_keyring, | 38 | ensure_ceph_keyring, |
329 | 40 | CephBrokerRq, | 39 | CephBrokerRq, |
330 | 41 | CephBrokerRsp, | ||
331 | 42 | delete_keyring, | 40 | delete_keyring, |
332 | 41 | send_request_if_needed, | ||
333 | 42 | is_request_complete, | ||
334 | 43 | ) | 43 | ) |
335 | 44 | from charmhelpers.payload.execd import execd_preinstall | 44 | from charmhelpers.payload.execd import execd_preinstall |
336 | 45 | from nova_compute_utils import ( | 45 | from nova_compute_utils import ( |
337 | @@ -264,6 +264,13 @@ | |||
338 | 264 | service_restart('libvirt-bin') | 264 | service_restart('libvirt-bin') |
339 | 265 | 265 | ||
340 | 266 | 266 | ||
341 | 267 | def get_ceph_request(): | ||
342 | 268 | rq = CephBrokerRq() | ||
343 | 269 | replicas = config('ceph-osd-replication-count') | ||
344 | 270 | rq.add_op_create_pool(name=config('rbd-pool'), replica_count=replicas) | ||
345 | 271 | return rq | ||
346 | 272 | |||
347 | 273 | |||
348 | 267 | @hooks.hook('ceph-relation-changed') | 274 | @hooks.hook('ceph-relation-changed') |
349 | 268 | @restart_on_change(restart_map()) | 275 | @restart_on_change(restart_map()) |
350 | 269 | def ceph_changed(): | 276 | def ceph_changed(): |
351 | @@ -282,36 +289,20 @@ | |||
352 | 282 | 289 | ||
353 | 283 | # With some refactoring, this can move into NovaComputeCephContext | 290 | # With some refactoring, this can move into NovaComputeCephContext |
354 | 284 | # and allow easily extended to support other compute flavors. | 291 | # and allow easily extended to support other compute flavors. |
356 | 285 | if config('virt-type') in ['kvm', 'qemu', 'lxc']: | 292 | if config('virt-type') in ['kvm', 'qemu', 'lxc'] and relation_get('key'): |
357 | 286 | create_libvirt_secret(secret_file=CEPH_SECRET, | 293 | create_libvirt_secret(secret_file=CEPH_SECRET, |
358 | 287 | secret_uuid=CEPH_SECRET_UUID, | 294 | secret_uuid=CEPH_SECRET_UUID, |
359 | 288 | key=relation_get('key')) | 295 | key=relation_get('key')) |
360 | 289 | 296 | ||
361 | 290 | if (config('libvirt-image-backend') == 'rbd' and | 297 | if (config('libvirt-image-backend') == 'rbd' and |
362 | 291 | assert_libvirt_imagebackend_allowed()): | 298 | assert_libvirt_imagebackend_allowed()): |
374 | 292 | settings = relation_get() | 299 | if is_request_complete(get_ceph_request()): |
375 | 293 | if settings and 'broker_rsp' in settings: | 300 | log('Request complete') |
365 | 294 | rsp = CephBrokerRsp(settings['broker_rsp']) | ||
366 | 295 | # Non-zero return code implies failure | ||
367 | 296 | if rsp.exit_code: | ||
368 | 297 | log("Ceph broker request failed (rc=%s, msg=%s)" % | ||
369 | 298 | (rsp.exit_code, rsp.exit_msg), level=ERROR) | ||
370 | 299 | return | ||
371 | 300 | |||
372 | 301 | log("Ceph broker request succeeded (rc=%s, msg=%s)" % | ||
373 | 302 | (rsp.exit_code, rsp.exit_msg), level=INFO) | ||
376 | 303 | # Ensure that nova-compute is restarted since only now can we | 301 | # Ensure that nova-compute is restarted since only now can we |
377 | 304 | # guarantee that ceph resources are ready. | 302 | # guarantee that ceph resources are ready. |
380 | 305 | if config('libvirt-image-backend') == 'rbd': | 303 | service_restart('nova-compute') |
379 | 306 | service_restart('nova-compute') | ||
381 | 307 | else: | 304 | else: |
389 | 308 | rq = CephBrokerRq() | 305 | send_request_if_needed(get_ceph_request()) |
383 | 309 | replicas = config('ceph-osd-replication-count') | ||
384 | 310 | rq.add_op_create_pool(name=config('rbd-pool'), | ||
385 | 311 | replica_count=replicas) | ||
386 | 312 | for rid in relation_ids('ceph'): | ||
387 | 313 | relation_set(broker_req=rq.request) | ||
388 | 314 | log("Request(s) sent to Ceph broker (rid=%s)" % (rid)) | ||
390 | 315 | 306 | ||
391 | 316 | 307 | ||
392 | 317 | @hooks.hook('ceph-relation-broken') | 308 | @hooks.hook('ceph-relation-broken') |
393 | 318 | 309 | ||
394 | === modified file 'tests/charmhelpers/contrib/amulet/utils.py' | |||
395 | --- tests/charmhelpers/contrib/amulet/utils.py 2015-08-19 14:19:48 +0000 | |||
396 | +++ tests/charmhelpers/contrib/amulet/utils.py 2015-09-10 09:35:32 +0000 | |||
397 | @@ -19,9 +19,11 @@ | |||
398 | 19 | import logging | 19 | import logging |
399 | 20 | import os | 20 | import os |
400 | 21 | import re | 21 | import re |
401 | 22 | import socket | ||
402 | 22 | import subprocess | 23 | import subprocess |
403 | 23 | import sys | 24 | import sys |
404 | 24 | import time | 25 | import time |
405 | 26 | import uuid | ||
406 | 25 | 27 | ||
407 | 26 | import amulet | 28 | import amulet |
408 | 27 | import distro_info | 29 | import distro_info |
409 | @@ -114,7 +116,7 @@ | |||
410 | 114 | # /!\ DEPRECATION WARNING (beisner): | 116 | # /!\ DEPRECATION WARNING (beisner): |
411 | 115 | # New and existing tests should be rewritten to use | 117 | # New and existing tests should be rewritten to use |
412 | 116 | # validate_services_by_name() as it is aware of init systems. | 118 | # validate_services_by_name() as it is aware of init systems. |
414 | 117 | self.log.warn('/!\\ DEPRECATION WARNING: use ' | 119 | self.log.warn('DEPRECATION WARNING: use ' |
415 | 118 | 'validate_services_by_name instead of validate_services ' | 120 | 'validate_services_by_name instead of validate_services ' |
416 | 119 | 'due to init system differences.') | 121 | 'due to init system differences.') |
417 | 120 | 122 | ||
418 | @@ -269,33 +271,52 @@ | |||
419 | 269 | """Get last modification time of directory.""" | 271 | """Get last modification time of directory.""" |
420 | 270 | return sentry_unit.directory_stat(directory)['mtime'] | 272 | return sentry_unit.directory_stat(directory)['mtime'] |
421 | 271 | 273 | ||
440 | 272 | def _get_proc_start_time(self, sentry_unit, service, pgrep_full=False): | 274 | def _get_proc_start_time(self, sentry_unit, service, pgrep_full=None): |
441 | 273 | """Get process' start time. | 275 | """Get start time of a process based on the last modification time |
442 | 274 | 276 | of the /proc/pid directory. | |
443 | 275 | Determine start time of the process based on the last modification | 277 | |
444 | 276 | time of the /proc/pid directory. If pgrep_full is True, the process | 278 | :sentry_unit: The sentry unit to check for the service on |
445 | 277 | name is matched against the full command line. | 279 | :service: service name to look for in process table |
446 | 278 | """ | 280 | :pgrep_full: [Deprecated] Use full command line search mode with pgrep |
447 | 279 | if pgrep_full: | 281 | :returns: epoch time of service process start |
448 | 280 | cmd = 'pgrep -o -f {}'.format(service) | 282 | :param commands: list of bash commands |
449 | 281 | else: | 283 | :param sentry_units: list of sentry unit pointers |
450 | 282 | cmd = 'pgrep -o {}'.format(service) | 284 | :returns: None if successful; Failure message otherwise |
451 | 283 | cmd = cmd + ' | grep -v pgrep || exit 0' | 285 | """ |
452 | 284 | cmd_out = sentry_unit.run(cmd) | 286 | if pgrep_full is not None: |
453 | 285 | self.log.debug('CMDout: ' + str(cmd_out)) | 287 | # /!\ DEPRECATION WARNING (beisner): |
454 | 286 | if cmd_out[0]: | 288 | # No longer implemented, as pidof is now used instead of pgrep. |
455 | 287 | self.log.debug('Pid for %s %s' % (service, str(cmd_out[0]))) | 289 | # https://bugs.launchpad.net/charm-helpers/+bug/1474030 |
456 | 288 | proc_dir = '/proc/{}'.format(cmd_out[0].strip()) | 290 | self.log.warn('DEPRECATION WARNING: pgrep_full bool is no ' |
457 | 289 | return self._get_dir_mtime(sentry_unit, proc_dir) | 291 | 'longer implemented re: lp 1474030.') |
458 | 292 | |||
459 | 293 | pid_list = self.get_process_id_list(sentry_unit, service) | ||
460 | 294 | pid = pid_list[0] | ||
461 | 295 | proc_dir = '/proc/{}'.format(pid) | ||
462 | 296 | self.log.debug('Pid for {} on {}: {}'.format( | ||
463 | 297 | service, sentry_unit.info['unit_name'], pid)) | ||
464 | 298 | |||
465 | 299 | return self._get_dir_mtime(sentry_unit, proc_dir) | ||
466 | 290 | 300 | ||
467 | 291 | def service_restarted(self, sentry_unit, service, filename, | 301 | def service_restarted(self, sentry_unit, service, filename, |
469 | 292 | pgrep_full=False, sleep_time=20): | 302 | pgrep_full=None, sleep_time=20): |
470 | 293 | """Check if service was restarted. | 303 | """Check if service was restarted. |
471 | 294 | 304 | ||
472 | 295 | Compare a service's start time vs a file's last modification time | 305 | Compare a service's start time vs a file's last modification time |
473 | 296 | (such as a config file for that service) to determine if the service | 306 | (such as a config file for that service) to determine if the service |
474 | 297 | has been restarted. | 307 | has been restarted. |
475 | 298 | """ | 308 | """ |
476 | 309 | # /!\ DEPRECATION WARNING (beisner): | ||
477 | 310 | # This method is prone to races in that no before-time is known. | ||
478 | 311 | # Use validate_service_config_changed instead. | ||
479 | 312 | |||
480 | 313 | # NOTE(beisner) pgrep_full is no longer implemented, as pidof is now | ||
481 | 314 | # used instead of pgrep. pgrep_full is still passed through to ensure | ||
482 | 315 | # deprecation WARNS. lp1474030 | ||
483 | 316 | self.log.warn('DEPRECATION WARNING: use ' | ||
484 | 317 | 'validate_service_config_changed instead of ' | ||
485 | 318 | 'service_restarted due to known races.') | ||
486 | 319 | |||
487 | 299 | time.sleep(sleep_time) | 320 | time.sleep(sleep_time) |
488 | 300 | if (self._get_proc_start_time(sentry_unit, service, pgrep_full) >= | 321 | if (self._get_proc_start_time(sentry_unit, service, pgrep_full) >= |
489 | 301 | self._get_file_mtime(sentry_unit, filename)): | 322 | self._get_file_mtime(sentry_unit, filename)): |
490 | @@ -304,15 +325,15 @@ | |||
491 | 304 | return False | 325 | return False |
492 | 305 | 326 | ||
493 | 306 | def service_restarted_since(self, sentry_unit, mtime, service, | 327 | def service_restarted_since(self, sentry_unit, mtime, service, |
496 | 307 | pgrep_full=False, sleep_time=20, | 328 | pgrep_full=None, sleep_time=20, |
497 | 308 | retry_count=2): | 329 | retry_count=2, retry_sleep_time=30): |
498 | 309 | """Check if service was been started after a given time. | 330 | """Check if service was been started after a given time. |
499 | 310 | 331 | ||
500 | 311 | Args: | 332 | Args: |
501 | 312 | sentry_unit (sentry): The sentry unit to check for the service on | 333 | sentry_unit (sentry): The sentry unit to check for the service on |
502 | 313 | mtime (float): The epoch time to check against | 334 | mtime (float): The epoch time to check against |
503 | 314 | service (string): service name to look for in process table | 335 | service (string): service name to look for in process table |
505 | 315 | pgrep_full (boolean): Use full command line search mode with pgrep | 336 | pgrep_full: [Deprecated] Use full command line search mode with pgrep |
506 | 316 | sleep_time (int): Seconds to sleep before looking for process | 337 | sleep_time (int): Seconds to sleep before looking for process |
507 | 317 | retry_count (int): If service is not found, how many times to retry | 338 | retry_count (int): If service is not found, how many times to retry |
508 | 318 | 339 | ||
509 | @@ -321,30 +342,44 @@ | |||
510 | 321 | False if service is older than mtime or if service was | 342 | False if service is older than mtime or if service was |
511 | 322 | not found. | 343 | not found. |
512 | 323 | """ | 344 | """ |
514 | 324 | self.log.debug('Checking %s restarted since %s' % (service, mtime)) | 345 | # NOTE(beisner) pgrep_full is no longer implemented, as pidof is now |
515 | 346 | # used instead of pgrep. pgrep_full is still passed through to ensure | ||
516 | 347 | # deprecation WARNS. lp1474030 | ||
517 | 348 | |||
518 | 349 | unit_name = sentry_unit.info['unit_name'] | ||
519 | 350 | self.log.debug('Checking that %s service restarted since %s on ' | ||
520 | 351 | '%s' % (service, mtime, unit_name)) | ||
521 | 325 | time.sleep(sleep_time) | 352 | time.sleep(sleep_time) |
531 | 326 | proc_start_time = self._get_proc_start_time(sentry_unit, service, | 353 | proc_start_time = None |
532 | 327 | pgrep_full) | 354 | tries = 0 |
533 | 328 | while retry_count > 0 and not proc_start_time: | 355 | while tries <= retry_count and not proc_start_time: |
534 | 329 | self.log.debug('No pid file found for service %s, will retry %i ' | 356 | try: |
535 | 330 | 'more times' % (service, retry_count)) | 357 | proc_start_time = self._get_proc_start_time(sentry_unit, |
536 | 331 | time.sleep(30) | 358 | service, |
537 | 332 | proc_start_time = self._get_proc_start_time(sentry_unit, service, | 359 | pgrep_full) |
538 | 333 | pgrep_full) | 360 | self.log.debug('Attempt {} to get {} proc start time on {} ' |
539 | 334 | retry_count = retry_count - 1 | 361 | 'OK'.format(tries, service, unit_name)) |
540 | 362 | except IOError: | ||
541 | 363 | # NOTE(beisner) - race avoidance, proc may not exist yet. | ||
542 | 364 | # https://bugs.launchpad.net/charm-helpers/+bug/1474030 | ||
543 | 365 | self.log.debug('Attempt {} to get {} proc start time on {} ' | ||
544 | 366 | 'failed'.format(tries, service, unit_name)) | ||
545 | 367 | time.sleep(retry_sleep_time) | ||
546 | 368 | tries += 1 | ||
547 | 335 | 369 | ||
548 | 336 | if not proc_start_time: | 370 | if not proc_start_time: |
549 | 337 | self.log.warn('No proc start time found, assuming service did ' | 371 | self.log.warn('No proc start time found, assuming service did ' |
550 | 338 | 'not start') | 372 | 'not start') |
551 | 339 | return False | 373 | return False |
552 | 340 | if proc_start_time >= mtime: | 374 | if proc_start_time >= mtime: |
555 | 341 | self.log.debug('proc start time is newer than provided mtime' | 375 | self.log.debug('Proc start time is newer than provided mtime' |
556 | 342 | '(%s >= %s)' % (proc_start_time, mtime)) | 376 | '(%s >= %s) on %s (OK)' % (proc_start_time, |
557 | 377 | mtime, unit_name)) | ||
558 | 343 | return True | 378 | return True |
559 | 344 | else: | 379 | else: |
563 | 345 | self.log.warn('proc start time (%s) is older than provided mtime ' | 380 | self.log.warn('Proc start time (%s) is older than provided mtime ' |
564 | 346 | '(%s), service did not restart' % (proc_start_time, | 381 | '(%s) on %s, service did not ' |
565 | 347 | mtime)) | 382 | 'restart' % (proc_start_time, mtime, unit_name)) |
566 | 348 | return False | 383 | return False |
567 | 349 | 384 | ||
568 | 350 | def config_updated_since(self, sentry_unit, filename, mtime, | 385 | def config_updated_since(self, sentry_unit, filename, mtime, |
569 | @@ -374,8 +409,9 @@ | |||
570 | 374 | return False | 409 | return False |
571 | 375 | 410 | ||
572 | 376 | def validate_service_config_changed(self, sentry_unit, mtime, service, | 411 | def validate_service_config_changed(self, sentry_unit, mtime, service, |
575 | 377 | filename, pgrep_full=False, | 412 | filename, pgrep_full=None, |
576 | 378 | sleep_time=20, retry_count=2): | 413 | sleep_time=20, retry_count=2, |
577 | 414 | retry_sleep_time=30): | ||
578 | 379 | """Check service and file were updated after mtime | 415 | """Check service and file were updated after mtime |
579 | 380 | 416 | ||
580 | 381 | Args: | 417 | Args: |
581 | @@ -383,9 +419,10 @@ | |||
582 | 383 | mtime (float): The epoch time to check against | 419 | mtime (float): The epoch time to check against |
583 | 384 | service (string): service name to look for in process table | 420 | service (string): service name to look for in process table |
584 | 385 | filename (string): The file to check mtime of | 421 | filename (string): The file to check mtime of |
587 | 386 | pgrep_full (boolean): Use full command line search mode with pgrep | 422 | pgrep_full: [Deprecated] Use full command line search mode with pgrep |
588 | 387 | sleep_time (int): Seconds to sleep before looking for process | 423 | sleep_time (int): Initial sleep in seconds to pass to test helpers |
589 | 388 | retry_count (int): If service is not found, how many times to retry | 424 | retry_count (int): If service is not found, how many times to retry |
590 | 425 | retry_sleep_time (int): Time in seconds to wait between retries | ||
591 | 389 | 426 | ||
592 | 390 | Typical Usage: | 427 | Typical Usage: |
593 | 391 | u = OpenStackAmuletUtils(ERROR) | 428 | u = OpenStackAmuletUtils(ERROR) |
594 | @@ -402,15 +439,25 @@ | |||
595 | 402 | mtime, False if service is older than mtime or if service was | 439 | mtime, False if service is older than mtime or if service was |
596 | 403 | not found or if filename was modified before mtime. | 440 | not found or if filename was modified before mtime. |
597 | 404 | """ | 441 | """ |
607 | 405 | self.log.debug('Checking %s restarted since %s' % (service, mtime)) | 442 | |
608 | 406 | time.sleep(sleep_time) | 443 | # NOTE(beisner) pgrep_full is no longer implemented, as pidof is now |
609 | 407 | service_restart = self.service_restarted_since(sentry_unit, mtime, | 444 | # used instead of pgrep. pgrep_full is still passed through to ensure |
610 | 408 | service, | 445 | # deprecation WARNS. lp1474030 |
611 | 409 | pgrep_full=pgrep_full, | 446 | |
612 | 410 | sleep_time=0, | 447 | service_restart = self.service_restarted_since( |
613 | 411 | retry_count=retry_count) | 448 | sentry_unit, mtime, |
614 | 412 | config_update = self.config_updated_since(sentry_unit, filename, mtime, | 449 | service, |
615 | 413 | sleep_time=0) | 450 | pgrep_full=pgrep_full, |
616 | 451 | sleep_time=sleep_time, | ||
617 | 452 | retry_count=retry_count, | ||
618 | 453 | retry_sleep_time=retry_sleep_time) | ||
619 | 454 | |||
620 | 455 | config_update = self.config_updated_since( | ||
621 | 456 | sentry_unit, | ||
622 | 457 | filename, | ||
623 | 458 | mtime, | ||
624 | 459 | sleep_time=0) | ||
625 | 460 | |||
626 | 414 | return service_restart and config_update | 461 | return service_restart and config_update |
627 | 415 | 462 | ||
628 | 416 | def get_sentry_time(self, sentry_unit): | 463 | def get_sentry_time(self, sentry_unit): |
629 | @@ -428,7 +475,6 @@ | |||
630 | 428 | """Return a list of all Ubuntu releases in order of release.""" | 475 | """Return a list of all Ubuntu releases in order of release.""" |
631 | 429 | _d = distro_info.UbuntuDistroInfo() | 476 | _d = distro_info.UbuntuDistroInfo() |
632 | 430 | _release_list = _d.all | 477 | _release_list = _d.all |
633 | 431 | self.log.debug('Ubuntu release list: {}'.format(_release_list)) | ||
634 | 432 | return _release_list | 478 | return _release_list |
635 | 433 | 479 | ||
636 | 434 | def file_to_url(self, file_rel_path): | 480 | def file_to_url(self, file_rel_path): |
637 | @@ -568,6 +614,142 @@ | |||
638 | 568 | 614 | ||
639 | 569 | return None | 615 | return None |
640 | 570 | 616 | ||
641 | 617 | def validate_sectionless_conf(self, file_contents, expected): | ||
642 | 618 | """A crude conf parser. Useful to inspect configuration files which | ||
643 | 619 | do not have section headers (as would be necessary in order to use | ||
644 | 620 | the configparser). Such as openstack-dashboard or rabbitmq confs.""" | ||
645 | 621 | for line in file_contents.split('\n'): | ||
646 | 622 | if '=' in line: | ||
647 | 623 | args = line.split('=') | ||
648 | 624 | if len(args) <= 1: | ||
649 | 625 | continue | ||
650 | 626 | key = args[0].strip() | ||
651 | 627 | value = args[1].strip() | ||
652 | 628 | if key in expected.keys(): | ||
653 | 629 | if expected[key] != value: | ||
654 | 630 | msg = ('Config mismatch. Expected, actual: {}, ' | ||
655 | 631 | '{}'.format(expected[key], value)) | ||
656 | 632 | amulet.raise_status(amulet.FAIL, msg=msg) | ||
657 | 633 | |||
658 | 634 | def get_unit_hostnames(self, units): | ||
659 | 635 | """Return a dict of juju unit names to hostnames.""" | ||
660 | 636 | host_names = {} | ||
661 | 637 | for unit in units: | ||
662 | 638 | host_names[unit.info['unit_name']] = \ | ||
663 | 639 | str(unit.file_contents('/etc/hostname').strip()) | ||
664 | 640 | self.log.debug('Unit host names: {}'.format(host_names)) | ||
665 | 641 | return host_names | ||
666 | 642 | |||
667 | 643 | def run_cmd_unit(self, sentry_unit, cmd): | ||
668 | 644 | """Run a command on a unit, return the output and exit code.""" | ||
669 | 645 | output, code = sentry_unit.run(cmd) | ||
670 | 646 | if code == 0: | ||
671 | 647 | self.log.debug('{} `{}` command returned {} ' | ||
672 | 648 | '(OK)'.format(sentry_unit.info['unit_name'], | ||
673 | 649 | cmd, code)) | ||
674 | 650 | else: | ||
675 | 651 | msg = ('{} `{}` command returned {} ' | ||
676 | 652 | '{}'.format(sentry_unit.info['unit_name'], | ||
677 | 653 | cmd, code, output)) | ||
678 | 654 | amulet.raise_status(amulet.FAIL, msg=msg) | ||
679 | 655 | return str(output), code | ||
680 | 656 | |||
681 | 657 | def file_exists_on_unit(self, sentry_unit, file_name): | ||
682 | 658 | """Check if a file exists on a unit.""" | ||
683 | 659 | try: | ||
684 | 660 | sentry_unit.file_stat(file_name) | ||
685 | 661 | return True | ||
686 | 662 | except IOError: | ||
687 | 663 | return False | ||
688 | 664 | except Exception as e: | ||
689 | 665 | msg = 'Error checking file {}: {}'.format(file_name, e) | ||
690 | 666 | amulet.raise_status(amulet.FAIL, msg=msg) | ||
691 | 667 | |||
692 | 668 | def file_contents_safe(self, sentry_unit, file_name, | ||
693 | 669 | max_wait=60, fatal=False): | ||
694 | 670 | """Get file contents from a sentry unit. Wrap amulet file_contents | ||
695 | 671 | with retry logic to address races where a file checks as existing, | ||
696 | 672 | but no longer exists by the time file_contents is called. | ||
697 | 673 | Return None if file not found. Optionally raise if fatal is True.""" | ||
698 | 674 | unit_name = sentry_unit.info['unit_name'] | ||
699 | 675 | file_contents = False | ||
700 | 676 | tries = 0 | ||
701 | 677 | while not file_contents and tries < (max_wait / 4): | ||
702 | 678 | try: | ||
703 | 679 | file_contents = sentry_unit.file_contents(file_name) | ||
704 | 680 | except IOError: | ||
705 | 681 | self.log.debug('Attempt {} to open file {} from {} ' | ||
706 | 682 | 'failed'.format(tries, file_name, | ||
707 | 683 | unit_name)) | ||
708 | 684 | time.sleep(4) | ||
709 | 685 | tries += 1 | ||
710 | 686 | |||
711 | 687 | if file_contents: | ||
712 | 688 | return file_contents | ||
713 | 689 | elif not fatal: | ||
714 | 690 | return None | ||
715 | 691 | elif fatal: | ||
716 | 692 | msg = 'Failed to get file contents from unit.' | ||
717 | 693 | amulet.raise_status(amulet.FAIL, msg) | ||
718 | 694 | |||
719 | 695 | def port_knock_tcp(self, host="localhost", port=22, timeout=15): | ||
720 | 696 | """Open a TCP socket to check for a listening sevice on a host. | ||
721 | 697 | |||
722 | 698 | :param host: host name or IP address, default to localhost | ||
723 | 699 | :param port: TCP port number, default to 22 | ||
724 | 700 | :param timeout: Connect timeout, default to 15 seconds | ||
725 | 701 | :returns: True if successful, False if connect failed | ||
726 | 702 | """ | ||
727 | 703 | |||
728 | 704 | # Resolve host name if possible | ||
729 | 705 | try: | ||
730 | 706 | connect_host = socket.gethostbyname(host) | ||
731 | 707 | host_human = "{} ({})".format(connect_host, host) | ||
732 | 708 | except socket.error as e: | ||
733 | 709 | self.log.warn('Unable to resolve address: ' | ||
734 | 710 | '{} ({}) Trying anyway!'.format(host, e)) | ||
735 | 711 | connect_host = host | ||
736 | 712 | host_human = connect_host | ||
737 | 713 | |||
738 | 714 | # Attempt socket connection | ||
739 | 715 | try: | ||
740 | 716 | knock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) | ||
741 | 717 | knock.settimeout(timeout) | ||
742 | 718 | knock.connect((connect_host, port)) | ||
743 | 719 | knock.close() | ||
744 | 720 | self.log.debug('Socket connect OK for host ' | ||
745 | 721 | '{} on port {}.'.format(host_human, port)) | ||
746 | 722 | return True | ||
747 | 723 | except socket.error as e: | ||
748 | 724 | self.log.debug('Socket connect FAIL for' | ||
749 | 725 | ' {} port {} ({})'.format(host_human, port, e)) | ||
750 | 726 | return False | ||
751 | 727 | |||
752 | 728 | def port_knock_units(self, sentry_units, port=22, | ||
753 | 729 | timeout=15, expect_success=True): | ||
754 | 730 | """Open a TCP socket to check for a listening sevice on each | ||
755 | 731 | listed juju unit. | ||
756 | 732 | |||
757 | 733 | :param sentry_units: list of sentry unit pointers | ||
758 | 734 | :param port: TCP port number, default to 22 | ||
759 | 735 | :param timeout: Connect timeout, default to 15 seconds | ||
760 | 736 | :expect_success: True by default, set False to invert logic | ||
761 | 737 | :returns: None if successful, Failure message otherwise | ||
762 | 738 | """ | ||
763 | 739 | for unit in sentry_units: | ||
764 | 740 | host = unit.info['public-address'] | ||
765 | 741 | connected = self.port_knock_tcp(host, port, timeout) | ||
766 | 742 | if not connected and expect_success: | ||
767 | 743 | return 'Socket connect failed.' | ||
768 | 744 | elif connected and not expect_success: | ||
769 | 745 | return 'Socket connected unexpectedly.' | ||
770 | 746 | |||
771 | 747 | def get_uuid_epoch_stamp(self): | ||
772 | 748 | """Returns a stamp string based on uuid4 and epoch time. Useful in | ||
773 | 749 | generating test messages which need to be unique-ish.""" | ||
774 | 750 | return '[{}-{}]'.format(uuid.uuid4(), time.time()) | ||
775 | 751 | |||
776 | 752 | # amulet juju action helpers: | ||
777 | 571 | def run_action(self, unit_sentry, action, | 753 | def run_action(self, unit_sentry, action, |
778 | 572 | _check_output=subprocess.check_output): | 754 | _check_output=subprocess.check_output): |
779 | 573 | """Run the named action on a given unit sentry. | 755 | """Run the named action on a given unit sentry. |
780 | 574 | 756 | ||
781 | === modified file 'tests/charmhelpers/contrib/openstack/amulet/deployment.py' | |||
782 | --- tests/charmhelpers/contrib/openstack/amulet/deployment.py 2015-08-19 14:19:48 +0000 | |||
783 | +++ tests/charmhelpers/contrib/openstack/amulet/deployment.py 2015-09-10 09:35:32 +0000 | |||
784 | @@ -44,8 +44,15 @@ | |||
785 | 44 | Determine if the local branch being tested is derived from its | 44 | Determine if the local branch being tested is derived from its |
786 | 45 | stable or next (dev) branch, and based on this, use the corresonding | 45 | stable or next (dev) branch, and based on this, use the corresonding |
787 | 46 | stable or next branches for the other_services.""" | 46 | stable or next branches for the other_services.""" |
788 | 47 | |||
789 | 48 | # Charms outside the lp:~openstack-charmers namespace | ||
790 | 47 | base_charms = ['mysql', 'mongodb', 'nrpe'] | 49 | base_charms = ['mysql', 'mongodb', 'nrpe'] |
791 | 48 | 50 | ||
792 | 51 | # Force these charms to current series even when using an older series. | ||
793 | 52 | # ie. Use trusty/nrpe even when series is precise, as the P charm | ||
794 | 53 | # does not possess the necessary external master config and hooks. | ||
795 | 54 | force_series_current = ['nrpe'] | ||
796 | 55 | |||
797 | 49 | if self.series in ['precise', 'trusty']: | 56 | if self.series in ['precise', 'trusty']: |
798 | 50 | base_series = self.series | 57 | base_series = self.series |
799 | 51 | else: | 58 | else: |
800 | @@ -53,11 +60,17 @@ | |||
801 | 53 | 60 | ||
802 | 54 | if self.stable: | 61 | if self.stable: |
803 | 55 | for svc in other_services: | 62 | for svc in other_services: |
804 | 63 | if svc['name'] in force_series_current: | ||
805 | 64 | base_series = self.current_next | ||
806 | 65 | |||
807 | 56 | temp = 'lp:charms/{}/{}' | 66 | temp = 'lp:charms/{}/{}' |
808 | 57 | svc['location'] = temp.format(base_series, | 67 | svc['location'] = temp.format(base_series, |
809 | 58 | svc['name']) | 68 | svc['name']) |
810 | 59 | else: | 69 | else: |
811 | 60 | for svc in other_services: | 70 | for svc in other_services: |
812 | 71 | if svc['name'] in force_series_current: | ||
813 | 72 | base_series = self.current_next | ||
814 | 73 | |||
815 | 61 | if svc['name'] in base_charms: | 74 | if svc['name'] in base_charms: |
816 | 62 | temp = 'lp:charms/{}/{}' | 75 | temp = 'lp:charms/{}/{}' |
817 | 63 | svc['location'] = temp.format(base_series, | 76 | svc['location'] = temp.format(base_series, |
818 | @@ -77,21 +90,23 @@ | |||
819 | 77 | 90 | ||
820 | 78 | services = other_services | 91 | services = other_services |
821 | 79 | services.append(this_service) | 92 | services.append(this_service) |
822 | 93 | |||
823 | 94 | # Charms which should use the source config option | ||
824 | 80 | use_source = ['mysql', 'mongodb', 'rabbitmq-server', 'ceph', | 95 | use_source = ['mysql', 'mongodb', 'rabbitmq-server', 'ceph', |
825 | 81 | 'ceph-osd', 'ceph-radosgw'] | 96 | 'ceph-osd', 'ceph-radosgw'] |
829 | 82 | # Most OpenStack subordinate charms do not expose an origin option | 97 | |
830 | 83 | # as that is controlled by the principle. | 98 | # Charms which can not use openstack-origin, ie. many subordinates |
831 | 84 | ignore = ['cinder-ceph', 'hacluster', 'neutron-openvswitch', 'nrpe'] | 99 | no_origin = ['cinder-ceph', 'hacluster', 'neutron-openvswitch', 'nrpe'] |
832 | 85 | 100 | ||
833 | 86 | if self.openstack: | 101 | if self.openstack: |
834 | 87 | for svc in services: | 102 | for svc in services: |
836 | 88 | if svc['name'] not in use_source + ignore: | 103 | if svc['name'] not in use_source + no_origin: |
837 | 89 | config = {'openstack-origin': self.openstack} | 104 | config = {'openstack-origin': self.openstack} |
838 | 90 | self.d.configure(svc['name'], config) | 105 | self.d.configure(svc['name'], config) |
839 | 91 | 106 | ||
840 | 92 | if self.source: | 107 | if self.source: |
841 | 93 | for svc in services: | 108 | for svc in services: |
843 | 94 | if svc['name'] in use_source and svc['name'] not in ignore: | 109 | if svc['name'] in use_source and svc['name'] not in no_origin: |
844 | 95 | config = {'source': self.source} | 110 | config = {'source': self.source} |
845 | 96 | self.d.configure(svc['name'], config) | 111 | self.d.configure(svc['name'], config) |
846 | 97 | 112 | ||
847 | 98 | 113 | ||
848 | === modified file 'tests/charmhelpers/contrib/openstack/amulet/utils.py' | |||
849 | --- tests/charmhelpers/contrib/openstack/amulet/utils.py 2015-07-16 20:18:54 +0000 | |||
850 | +++ tests/charmhelpers/contrib/openstack/amulet/utils.py 2015-09-10 09:35:32 +0000 | |||
851 | @@ -27,6 +27,7 @@ | |||
852 | 27 | import heatclient.v1.client as heat_client | 27 | import heatclient.v1.client as heat_client |
853 | 28 | import keystoneclient.v2_0 as keystone_client | 28 | import keystoneclient.v2_0 as keystone_client |
854 | 29 | import novaclient.v1_1.client as nova_client | 29 | import novaclient.v1_1.client as nova_client |
855 | 30 | import pika | ||
856 | 30 | import swiftclient | 31 | import swiftclient |
857 | 31 | 32 | ||
858 | 32 | from charmhelpers.contrib.amulet.utils import ( | 33 | from charmhelpers.contrib.amulet.utils import ( |
859 | @@ -602,3 +603,361 @@ | |||
860 | 602 | self.log.debug('Ceph {} samples (OK): ' | 603 | self.log.debug('Ceph {} samples (OK): ' |
861 | 603 | '{}'.format(sample_type, samples)) | 604 | '{}'.format(sample_type, samples)) |
862 | 604 | return None | 605 | return None |
863 | 606 | |||
864 | 607 | # rabbitmq/amqp specific helpers: | ||
865 | 608 | def add_rmq_test_user(self, sentry_units, | ||
866 | 609 | username="testuser1", password="changeme"): | ||
867 | 610 | """Add a test user via the first rmq juju unit, check connection as | ||
868 | 611 | the new user against all sentry units. | ||
869 | 612 | |||
870 | 613 | :param sentry_units: list of sentry unit pointers | ||
871 | 614 | :param username: amqp user name, default to testuser1 | ||
872 | 615 | :param password: amqp user password | ||
873 | 616 | :returns: None if successful. Raise on error. | ||
874 | 617 | """ | ||
875 | 618 | self.log.debug('Adding rmq user ({})...'.format(username)) | ||
876 | 619 | |||
877 | 620 | # Check that user does not already exist | ||
878 | 621 | cmd_user_list = 'rabbitmqctl list_users' | ||
879 | 622 | output, _ = self.run_cmd_unit(sentry_units[0], cmd_user_list) | ||
880 | 623 | if username in output: | ||
881 | 624 | self.log.warning('User ({}) already exists, returning ' | ||
882 | 625 | 'gracefully.'.format(username)) | ||
883 | 626 | return | ||
884 | 627 | |||
885 | 628 | perms = '".*" ".*" ".*"' | ||
886 | 629 | cmds = ['rabbitmqctl add_user {} {}'.format(username, password), | ||
887 | 630 | 'rabbitmqctl set_permissions {} {}'.format(username, perms)] | ||
888 | 631 | |||
889 | 632 | # Add user via first unit | ||
890 | 633 | for cmd in cmds: | ||
891 | 634 | output, _ = self.run_cmd_unit(sentry_units[0], cmd) | ||
892 | 635 | |||
893 | 636 | # Check connection against the other sentry_units | ||
894 | 637 | self.log.debug('Checking user connect against units...') | ||
895 | 638 | for sentry_unit in sentry_units: | ||
896 | 639 | connection = self.connect_amqp_by_unit(sentry_unit, ssl=False, | ||
897 | 640 | username=username, | ||
898 | 641 | password=password) | ||
899 | 642 | connection.close() | ||
900 | 643 | |||
901 | 644 | def delete_rmq_test_user(self, sentry_units, username="testuser1"): | ||
902 | 645 | """Delete a rabbitmq user via the first rmq juju unit. | ||
903 | 646 | |||
904 | 647 | :param sentry_units: list of sentry unit pointers | ||
905 | 648 | :param username: amqp user name, default to testuser1 | ||
906 | 649 | :param password: amqp user password | ||
907 | 650 | :returns: None if successful or no such user. | ||
908 | 651 | """ | ||
909 | 652 | self.log.debug('Deleting rmq user ({})...'.format(username)) | ||
910 | 653 | |||
911 | 654 | # Check that the user exists | ||
912 | 655 | cmd_user_list = 'rabbitmqctl list_users' | ||
913 | 656 | output, _ = self.run_cmd_unit(sentry_units[0], cmd_user_list) | ||
914 | 657 | |||
915 | 658 | if username not in output: | ||
916 | 659 | self.log.warning('User ({}) does not exist, returning ' | ||
917 | 660 | 'gracefully.'.format(username)) | ||
918 | 661 | return | ||
919 | 662 | |||
920 | 663 | # Delete the user | ||
921 | 664 | cmd_user_del = 'rabbitmqctl delete_user {}'.format(username) | ||
922 | 665 | output, _ = self.run_cmd_unit(sentry_units[0], cmd_user_del) | ||
923 | 666 | |||
924 | 667 | def get_rmq_cluster_status(self, sentry_unit): | ||
925 | 668 | """Execute rabbitmq cluster status command on a unit and return | ||
926 | 669 | the full output. | ||
927 | 670 | |||
928 | 671 | :param unit: sentry unit | ||
929 | 672 | :returns: String containing console output of cluster status command | ||
930 | 673 | """ | ||
931 | 674 | cmd = 'rabbitmqctl cluster_status' | ||
932 | 675 | output, _ = self.run_cmd_unit(sentry_unit, cmd) | ||
933 | 676 | self.log.debug('{} cluster_status:\n{}'.format( | ||
934 | 677 | sentry_unit.info['unit_name'], output)) | ||
935 | 678 | return str(output) | ||
936 | 679 | |||
937 | 680 | def get_rmq_cluster_running_nodes(self, sentry_unit): | ||
938 | 681 | """Parse rabbitmqctl cluster_status output string, return list of | ||
939 | 682 | running rabbitmq cluster nodes. | ||
940 | 683 | |||
941 | 684 | :param unit: sentry unit | ||
942 | 685 | :returns: List containing node names of running nodes | ||
943 | 686 | """ | ||
944 | 687 | # NOTE(beisner): rabbitmqctl cluster_status output is not | ||
945 | 688 | # json-parsable, do string chop foo, then json.loads that. | ||
946 | 689 | str_stat = self.get_rmq_cluster_status(sentry_unit) | ||
947 | 690 | if 'running_nodes' in str_stat: | ||
948 | 691 | pos_start = str_stat.find("{running_nodes,") + 15 | ||
949 | 692 | pos_end = str_stat.find("]},", pos_start) + 1 | ||
950 | 693 | str_run_nodes = str_stat[pos_start:pos_end].replace("'", '"') | ||
951 | 694 | run_nodes = json.loads(str_run_nodes) | ||
952 | 695 | return run_nodes | ||
953 | 696 | else: | ||
954 | 697 | return [] | ||
955 | 698 | |||
956 | 699 | def validate_rmq_cluster_running_nodes(self, sentry_units): | ||
957 | 700 | """Check that all rmq unit hostnames are represented in the | ||
958 | 701 | cluster_status output of all units. | ||
959 | 702 | |||
960 | 703 | :param host_names: dict of juju unit names to host names | ||
961 | 704 | :param units: list of sentry unit pointers (all rmq units) | ||
962 | 705 | :returns: None if successful, otherwise return error message | ||
963 | 706 | """ | ||
964 | 707 | host_names = self.get_unit_hostnames(sentry_units) | ||
965 | 708 | errors = [] | ||
966 | 709 | |||
967 | 710 | # Query every unit for cluster_status running nodes | ||
968 | 711 | for query_unit in sentry_units: | ||
969 | 712 | query_unit_name = query_unit.info['unit_name'] | ||
970 | 713 | running_nodes = self.get_rmq_cluster_running_nodes(query_unit) | ||
971 | 714 | |||
972 | 715 | # Confirm that every unit is represented in the queried unit's | ||
973 | 716 | # cluster_status running nodes output. | ||
974 | 717 | for validate_unit in sentry_units: | ||
975 | 718 | val_host_name = host_names[validate_unit.info['unit_name']] | ||
976 | 719 | val_node_name = 'rabbit@{}'.format(val_host_name) | ||
977 | 720 | |||
978 | 721 | if val_node_name not in running_nodes: | ||
979 | 722 | errors.append('Cluster member check failed on {}: {} not ' | ||
980 | 723 | 'in {}\n'.format(query_unit_name, | ||
981 | 724 | val_node_name, | ||
982 | 725 | running_nodes)) | ||
983 | 726 | if errors: | ||
984 | 727 | return ''.join(errors) | ||
985 | 728 | |||
986 | 729 | def rmq_ssl_is_enabled_on_unit(self, sentry_unit, port=None): | ||
987 | 730 | """Check a single juju rmq unit for ssl and port in the config file.""" | ||
988 | 731 | host = sentry_unit.info['public-address'] | ||
989 | 732 | unit_name = sentry_unit.info['unit_name'] | ||
990 | 733 | |||
991 | 734 | conf_file = '/etc/rabbitmq/rabbitmq.config' | ||
992 | 735 | conf_contents = str(self.file_contents_safe(sentry_unit, | ||
993 | 736 | conf_file, max_wait=16)) | ||
994 | 737 | # Checks | ||
995 | 738 | conf_ssl = 'ssl' in conf_contents | ||
996 | 739 | conf_port = str(port) in conf_contents | ||
997 | 740 | |||
998 | 741 | # Port explicitly checked in config | ||
999 | 742 | if port and conf_port and conf_ssl: | ||
1000 | 743 | self.log.debug('SSL is enabled @{}:{} ' | ||
1001 | 744 | '({})'.format(host, port, unit_name)) | ||
1002 | 745 | return True | ||
1003 | 746 | elif port and not conf_port and conf_ssl: | ||
1004 | 747 | self.log.debug('SSL is enabled @{} but not on port {} ' | ||
1005 | 748 | '({})'.format(host, port, unit_name)) | ||
1006 | 749 | return False | ||
1007 | 750 | # Port not checked (useful when checking that ssl is disabled) | ||
1008 | 751 | elif not port and conf_ssl: | ||
1009 | 752 | self.log.debug('SSL is enabled @{}:{} ' | ||
1010 | 753 | '({})'.format(host, port, unit_name)) | ||
1011 | 754 | return True | ||
1012 | 755 | elif not port and not conf_ssl: | ||
1013 | 756 | self.log.debug('SSL not enabled @{}:{} ' | ||
1014 | 757 | '({})'.format(host, port, unit_name)) | ||
1015 | 758 | return False | ||
1016 | 759 | else: | ||
1017 | 760 | msg = ('Unknown condition when checking SSL status @{}:{} ' | ||
1018 | 761 | '({})'.format(host, port, unit_name)) | ||
1019 | 762 | amulet.raise_status(amulet.FAIL, msg) | ||
1020 | 763 | |||
1021 | 764 | def validate_rmq_ssl_enabled_units(self, sentry_units, port=None): | ||
1022 | 765 | """Check that ssl is enabled on rmq juju sentry units. | ||
1023 | 766 | |||
1024 | 767 | :param sentry_units: list of all rmq sentry units | ||
1025 | 768 | :param port: optional ssl port override to validate | ||
1026 | 769 | :returns: None if successful, otherwise return error message | ||
1027 | 770 | """ | ||
1028 | 771 | for sentry_unit in sentry_units: | ||
1029 | 772 | if not self.rmq_ssl_is_enabled_on_unit(sentry_unit, port=port): | ||
1030 | 773 | return ('Unexpected condition: ssl is disabled on unit ' | ||
1031 | 774 | '({})'.format(sentry_unit.info['unit_name'])) | ||
1032 | 775 | return None | ||
1033 | 776 | |||
1034 | 777 | def validate_rmq_ssl_disabled_units(self, sentry_units): | ||
1035 | 778 | """Check that ssl is enabled on listed rmq juju sentry units. | ||
1036 | 779 | |||
1037 | 780 | :param sentry_units: list of all rmq sentry units | ||
1038 | 781 | :returns: True if successful. Raise on error. | ||
1039 | 782 | """ | ||
1040 | 783 | for sentry_unit in sentry_units: | ||
1041 | 784 | if self.rmq_ssl_is_enabled_on_unit(sentry_unit): | ||
1042 | 785 | return ('Unexpected condition: ssl is enabled on unit ' | ||
1043 | 786 | '({})'.format(sentry_unit.info['unit_name'])) | ||
1044 | 787 | return None | ||
1045 | 788 | |||
1046 | 789 | def configure_rmq_ssl_on(self, sentry_units, deployment, | ||
1047 | 790 | port=None, max_wait=60): | ||
1048 | 791 | """Turn ssl charm config option on, with optional non-default | ||
1049 | 792 | ssl port specification. Confirm that it is enabled on every | ||
1050 | 793 | unit. | ||
1051 | 794 | |||
1052 | 795 | :param sentry_units: list of sentry units | ||
1053 | 796 | :param deployment: amulet deployment object pointer | ||
1054 | 797 | :param port: amqp port, use defaults if None | ||
1055 | 798 | :param max_wait: maximum time to wait in seconds to confirm | ||
1056 | 799 | :returns: None if successful. Raise on error. | ||
1057 | 800 | """ | ||
1058 | 801 | self.log.debug('Setting ssl charm config option: on') | ||
1059 | 802 | |||
1060 | 803 | # Enable RMQ SSL | ||
1061 | 804 | config = {'ssl': 'on'} | ||
1062 | 805 | if port: | ||
1063 | 806 | config['ssl_port'] = port | ||
1064 | 807 | |||
1065 | 808 | deployment.configure('rabbitmq-server', config) | ||
1066 | 809 | |||
1067 | 810 | # Confirm | ||
1068 | 811 | tries = 0 | ||
1069 | 812 | ret = self.validate_rmq_ssl_enabled_units(sentry_units, port=port) | ||
1070 | 813 | while ret and tries < (max_wait / 4): | ||
1071 | 814 | time.sleep(4) | ||
1072 | 815 | self.log.debug('Attempt {}: {}'.format(tries, ret)) | ||
1073 | 816 | ret = self.validate_rmq_ssl_enabled_units(sentry_units, port=port) | ||
1074 | 817 | tries += 1 | ||
1075 | 818 | |||
1076 | 819 | if ret: | ||
1077 | 820 | amulet.raise_status(amulet.FAIL, ret) | ||
1078 | 821 | |||
1079 | 822 | def configure_rmq_ssl_off(self, sentry_units, deployment, max_wait=60): | ||
1080 | 823 | """Turn ssl charm config option off, confirm that it is disabled | ||
1081 | 824 | on every unit. | ||
1082 | 825 | |||
1083 | 826 | :param sentry_units: list of sentry units | ||
1084 | 827 | :param deployment: amulet deployment object pointer | ||
1085 | 828 | :param max_wait: maximum time to wait in seconds to confirm | ||
1086 | 829 | :returns: None if successful. Raise on error. | ||
1087 | 830 | """ | ||
1088 | 831 | self.log.debug('Setting ssl charm config option: off') | ||
1089 | 832 | |||
1090 | 833 | # Disable RMQ SSL | ||
1091 | 834 | config = {'ssl': 'off'} | ||
1092 | 835 | deployment.configure('rabbitmq-server', config) | ||
1093 | 836 | |||
1094 | 837 | # Confirm | ||
1095 | 838 | tries = 0 | ||
1096 | 839 | ret = self.validate_rmq_ssl_disabled_units(sentry_units) | ||
1097 | 840 | while ret and tries < (max_wait / 4): | ||
1098 | 841 | time.sleep(4) | ||
1099 | 842 | self.log.debug('Attempt {}: {}'.format(tries, ret)) | ||
1100 | 843 | ret = self.validate_rmq_ssl_disabled_units(sentry_units) | ||
1101 | 844 | tries += 1 | ||
1102 | 845 | |||
1103 | 846 | if ret: | ||
1104 | 847 | amulet.raise_status(amulet.FAIL, ret) | ||
1105 | 848 | |||
1106 | 849 | def connect_amqp_by_unit(self, sentry_unit, ssl=False, | ||
1107 | 850 | port=None, fatal=True, | ||
1108 | 851 | username="testuser1", password="changeme"): | ||
1109 | 852 | """Establish and return a pika amqp connection to the rabbitmq service | ||
1110 | 853 | running on a rmq juju unit. | ||
1111 | 854 | |||
1112 | 855 | :param sentry_unit: sentry unit pointer | ||
1113 | 856 | :param ssl: boolean, default to False | ||
1114 | 857 | :param port: amqp port, use defaults if None | ||
1115 | 858 | :param fatal: boolean, default to True (raises on connect error) | ||
1116 | 859 | :param username: amqp user name, default to testuser1 | ||
1117 | 860 | :param password: amqp user password | ||
1118 | 861 | :returns: pika amqp connection pointer or None if failed and non-fatal | ||
1119 | 862 | """ | ||
1120 | 863 | host = sentry_unit.info['public-address'] | ||
1121 | 864 | unit_name = sentry_unit.info['unit_name'] | ||
1122 | 865 | |||
1123 | 866 | # Default port logic if port is not specified | ||
1124 | 867 | if ssl and not port: | ||
1125 | 868 | port = 5671 | ||
1126 | 869 | elif not ssl and not port: | ||
1127 | 870 | port = 5672 | ||
1128 | 871 | |||
1129 | 872 | self.log.debug('Connecting to amqp on {}:{} ({}) as ' | ||
1130 | 873 | '{}...'.format(host, port, unit_name, username)) | ||
1131 | 874 | |||
1132 | 875 | try: | ||
1133 | 876 | credentials = pika.PlainCredentials(username, password) | ||
1134 | 877 | parameters = pika.ConnectionParameters(host=host, port=port, | ||
1135 | 878 | credentials=credentials, | ||
1136 | 879 | ssl=ssl, | ||
1137 | 880 | connection_attempts=3, | ||
1138 | 881 | retry_delay=5, | ||
1139 | 882 | socket_timeout=1) | ||
1140 | 883 | connection = pika.BlockingConnection(parameters) | ||
1141 | 884 | assert connection.server_properties['product'] == 'RabbitMQ' | ||
1142 | 885 | self.log.debug('Connect OK') | ||
1143 | 886 | return connection | ||
1144 | 887 | except Exception as e: | ||
1145 | 888 | msg = ('amqp connection failed to {}:{} as ' | ||
1146 | 889 | '{} ({})'.format(host, port, username, str(e))) | ||
1147 | 890 | if fatal: | ||
1148 | 891 | amulet.raise_status(amulet.FAIL, msg) | ||
1149 | 892 | else: | ||
1150 | 893 | self.log.warn(msg) | ||
1151 | 894 | return None | ||
1152 | 895 | |||
1153 | 896 | def publish_amqp_message_by_unit(self, sentry_unit, message, | ||
1154 | 897 | queue="test", ssl=False, | ||
1155 | 898 | username="testuser1", | ||
1156 | 899 | password="changeme", | ||
1157 | 900 | port=None): | ||
1158 | 901 | """Publish an amqp message to a rmq juju unit. | ||
1159 | 902 | |||
1160 | 903 | :param sentry_unit: sentry unit pointer | ||
1161 | 904 | :param message: amqp message string | ||
1162 | 905 | :param queue: message queue, default to test | ||
1163 | 906 | :param username: amqp user name, default to testuser1 | ||
1164 | 907 | :param password: amqp user password | ||
1165 | 908 | :param ssl: boolean, default to False | ||
1166 | 909 | :param port: amqp port, use defaults if None | ||
1167 | 910 | :returns: None. Raises exception if publish failed. | ||
1168 | 911 | """ | ||
1169 | 912 | self.log.debug('Publishing message to {} queue:\n{}'.format(queue, | ||
1170 | 913 | message)) | ||
1171 | 914 | connection = self.connect_amqp_by_unit(sentry_unit, ssl=ssl, | ||
1172 | 915 | port=port, | ||
1173 | 916 | username=username, | ||
1174 | 917 | password=password) | ||
1175 | 918 | |||
1176 | 919 | # NOTE(beisner): extra debug here re: pika hang potential: | ||
1177 | 920 | # https://github.com/pika/pika/issues/297 | ||
1178 | 921 | # https://groups.google.com/forum/#!topic/rabbitmq-users/Ja0iyfF0Szw | ||
1179 | 922 | self.log.debug('Defining channel...') | ||
1180 | 923 | channel = connection.channel() | ||
1181 | 924 | self.log.debug('Declaring queue...') | ||
1182 | 925 | channel.queue_declare(queue=queue, auto_delete=False, durable=True) | ||
1183 | 926 | self.log.debug('Publishing message...') | ||
1184 | 927 | channel.basic_publish(exchange='', routing_key=queue, body=message) | ||
1185 | 928 | self.log.debug('Closing channel...') | ||
1186 | 929 | channel.close() | ||
1187 | 930 | self.log.debug('Closing connection...') | ||
1188 | 931 | connection.close() | ||
1189 | 932 | |||
1190 | 933 | def get_amqp_message_by_unit(self, sentry_unit, queue="test", | ||
1191 | 934 | username="testuser1", | ||
1192 | 935 | password="changeme", | ||
1193 | 936 | ssl=False, port=None): | ||
1194 | 937 | """Get an amqp message from a rmq juju unit. | ||
1195 | 938 | |||
1196 | 939 | :param sentry_unit: sentry unit pointer | ||
1197 | 940 | :param queue: message queue, default to test | ||
1198 | 941 | :param username: amqp user name, default to testuser1 | ||
1199 | 942 | :param password: amqp user password | ||
1200 | 943 | :param ssl: boolean, default to False | ||
1201 | 944 | :param port: amqp port, use defaults if None | ||
1202 | 945 | :returns: amqp message body as string. Raise if get fails. | ||
1203 | 946 | """ | ||
1204 | 947 | connection = self.connect_amqp_by_unit(sentry_unit, ssl=ssl, | ||
1205 | 948 | port=port, | ||
1206 | 949 | username=username, | ||
1207 | 950 | password=password) | ||
1208 | 951 | channel = connection.channel() | ||
1209 | 952 | method_frame, _, body = channel.basic_get(queue) | ||
1210 | 953 | |||
1211 | 954 | if method_frame: | ||
1212 | 955 | self.log.debug('Retreived message from {} queue:\n{}'.format(queue, | ||
1213 | 956 | body)) | ||
1214 | 957 | channel.basic_ack(method_frame.delivery_tag) | ||
1215 | 958 | channel.close() | ||
1216 | 959 | connection.close() | ||
1217 | 960 | return body | ||
1218 | 961 | else: | ||
1219 | 962 | msg = 'No message retrieved.' | ||
1220 | 963 | amulet.raise_status(amulet.FAIL, msg) | ||
1221 | 605 | 964 | ||
1222 | === modified file 'unit_tests/test_nova_compute_hooks.py' | |||
1223 | --- unit_tests/test_nova_compute_hooks.py 2015-08-13 10:54:25 +0000 | |||
1224 | +++ unit_tests/test_nova_compute_hooks.py 2015-09-10 09:35:32 +0000 | |||
1225 | @@ -54,6 +54,9 @@ | |||
1226 | 54 | # misc_utils | 54 | # misc_utils |
1227 | 55 | 'ensure_ceph_keyring', | 55 | 'ensure_ceph_keyring', |
1228 | 56 | 'execd_preinstall', | 56 | 'execd_preinstall', |
1229 | 57 | 'assert_libvirt_imagebackend_allowed', | ||
1230 | 58 | 'is_request_complete', | ||
1231 | 59 | 'send_request_if_needed', | ||
1232 | 57 | # socket | 60 | # socket |
1233 | 58 | 'gethostname', | 61 | 'gethostname', |
1234 | 59 | 'create_sysctl', | 62 | 'create_sysctl', |
1235 | @@ -450,6 +453,9 @@ | |||
1236 | 450 | @patch.object(hooks, 'CONFIGS') | 453 | @patch.object(hooks, 'CONFIGS') |
1237 | 451 | def test_ceph_changed_with_key_and_relation_data(self, configs, | 454 | def test_ceph_changed_with_key_and_relation_data(self, configs, |
1238 | 452 | service_name): | 455 | service_name): |
1239 | 456 | self.test_config.set('libvirt-image-backend', 'rbd') | ||
1240 | 457 | self.is_request_complete.return_value = True | ||
1241 | 458 | self.assert_libvirt_imagebackend_allowed.return_value = True | ||
1242 | 453 | configs.complete_contexts = MagicMock() | 459 | configs.complete_contexts = MagicMock() |
1243 | 454 | configs.complete_contexts.return_value = ['ceph'] | 460 | configs.complete_contexts.return_value = ['ceph'] |
1244 | 455 | configs.write = MagicMock() | 461 | configs.write = MagicMock() |
1245 | @@ -462,6 +468,7 @@ | |||
1246 | 462 | call('/etc/nova/nova.conf'), | 468 | call('/etc/nova/nova.conf'), |
1247 | 463 | ] | 469 | ] |
1248 | 464 | self.assertEquals(ex, configs.write.call_args_list) | 470 | self.assertEquals(ex, configs.write.call_args_list) |
1249 | 471 | self.service_restart.assert_called_with('nova-compute') | ||
1250 | 465 | 472 | ||
1251 | 466 | @patch.object(hooks, 'CONFIGS') | 473 | @patch.object(hooks, 'CONFIGS') |
1252 | 467 | def test_neutron_plugin_changed(self, configs): | 474 | def test_neutron_plugin_changed(self, configs): |
charm_lint_check #9592 nova-compute-next for gnuoy mp269377
LINT OK: passed
Build: http:// 10.245. 162.77: 8080/job/ charm_lint_ check/9592/