azure/aria/aria-extension-cloudify/src/aria/tests/orchestrator/workflows/helpers.py

   1 # Licensed to the Apache Software Foundation (ASF) under one or more
   2 # contributor license agreements.  See the NOTICE file distributed with
   3 # this work for additional information regarding copyright ownership.
   4 # The ASF licenses this file to You under the Apache License, Version 2.0
   5 # (the "License"); you may not use this file except in compliance with
   6 # the License.  You may obtain a copy of the License at
   7 #
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 from contextlib import contextmanager
  17
  18
  19 @contextmanager
  20 def events_collector(*signals):
  21     handlers = {}
  22     collected = {}
  23
  24     def handler_factory(key):
  25         def handler(*args, **kwargs):
  26             signal_events = collected.setdefault(key, [])
  27             signal_events.append({'args': args, 'kwargs': kwargs})
  28         handlers[signal] = handler
  29         return handler
  30
  31     for signal in signals:
  32         signal.connect(handler_factory(signal))
  33     try:
  34         yield collected
  35     finally:
  36         for signal in signals:
  37             signal.disconnect(handlers[signal])