Exemplo n.º 1
0
        # 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)
Exemplo n.º 2
0
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)
Exemplo n.º 3
0
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)
Exemplo n.º 4
0
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: