# server_properties record['name'] = "'scalarizr.key'" record[ 'value'] = "'8mYTcBxiE70DtXCBRjn7AMuTQNzBJJcTa5uFok24X40ePafq1gUyyg=='" fill_server_properties(step) @step(u"^White Rabbit checks all messages has status (\d+)$") def check_messages_status(step, status): db = dbmanager.DB(lib.world.config['connections']['mysql']) query = "SELECT count(messageid) AS count FROM messages WHERE status != {0}".format( int(status)) result = db.execute(query)[0] assert result['count'] == 0 def before_scenario(scenario): pass def after_scenario(scenario): if hasattr(lib.world, 'wsgi_servers'): for wsgi_server in lib.world.wsgi_servers.values(): wsgi_server.stop() if hasattr(lib.world, 'script') and lib.world.script: lib.world.script.stop() before.each_scenario(before_scenario) after.each_scenario(after_scenario)
def stop_vpc_router(step): lib.world.vpc_router.stop() @step(u"White Rabbit checks server with server_id '(.*)' was updated$") def check_update1(step, server_id): assert server_id in lib.update_ok assert lib.update_ok[server_id] @step(u"White Rabbit checks server with server_id '(.*)' was updated (\d+) times$") def check_update2(step, server_id, count): if int(count): assert server_id in lib.update_count assert lib.update_count[server_id] == int(count) def before_scenario(scenario): lib.update_ok = {} lib.update_count = {} def after_scenario(scenario): if hasattr(lib.world, 'szr_upd_client'): for szr_upd_client in lib.world.szr_upd_client.values(): szr_upd_client.stop() before.each_scenario(before_scenario) after.each_scenario(after_scenario)
import json from django.core.management import call_command from leaves.models import * from lettuce import after, step from lettuce.django.steps.models import * from nose.tools import assert_equals after.each_scenario(clean_db) max_rego = 0 @creates_models(Harvester) def create_with_rego(step): data = hashes_data(step) for hash_ in data: hash_['rego'] = hash_['make'][:3].upper() + "001" create_models(Harvester, data) @checks_existence(Harvester) def check_with_rego(step): data = hashes_data(step) for hash_ in data: try: hash_['rego'] = hash_['rego'].upper() except KeyError: pass models_exist(Harvester, data)
import json from django.core.management import call_command from leaves.models import * from lettuce import after, step from lettuce.django.steps.models import * from nose.tools import assert_equals after.each_scenario(clean_db) max_rego = 0 @creates_models(Harvester) def create_with_rego(step): data = hashes_data(step) for hash_ in data: hash_['rego'] = hash_['make'][:3].upper() + "001" create_models(Harvester, data) @checks_existence(Harvester) def check_with_rego(step): data = hashes_data(step) for hash_ in data: try: hash_['rego'] = hash_['rego'].upper() except KeyError: