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)
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)
