Пример #1
0
def test_diff2():
    test_dir = os.path.join(data_dir, 'diff_test2')

    old_policy_dir = os.path.join(test_dir, 'old_policy')
    old_globals_path = os.path.join(old_policy_dir, 'old_prefix.json')
    old_rules_path = os.path.join(old_policy_dir, 'old_rules.json')
    old_apps_dir = os.path.join(old_policy_dir, 'apps')
    ensure_dir(old_apps_dir)

    new_policy_dir = os.path.join(test_dir, 'new_policy')
    new_globals_path = os.path.join(new_policy_dir, 'new_prefix.json')
    new_rules_path = os.path.join(new_policy_dir, 'new_rules.json')
    new_apps_dir = os.path.join(new_policy_dir, 'apps')
    ensure_dir(new_apps_dir)

    diff_path_pickle = os.path.join(test_dir, 'diff/diff2.pickle')
    diff_path_json = os.path.join(test_dir, 'diff/diff2.json')

    all_apps_old = scripts.run(old_rules_path, old_globals_path, old_apps_dir)
    all_apps_new = scripts.run(new_rules_path, new_globals_path, new_apps_dir)

    policy_diff = diff.make_diff(old_apps_dir, new_apps_dir)
    with open(diff_path_pickle, 'wb') as f:
        pickle.dump(policy_diff, f)

    policy_jsonable = massage_to_jsonable(policy_diff)
    with open(diff_path_json, 'wb') as f:
        json.dump(policy_jsonable, f, indent=4)

    eq_(gen_diff_diff2_expect, policy_diff)
Пример #2
0
def test_json_gen_allow_simple3():
    test_dir = os.path.join(data_dir, 'gen_allow_simple3')
    policies_path = os.path.join(test_dir, 'gen_allow_simple3-rules.json')
    globals_path = os.path.join(test_dir, 'gen_allow_simple3-prefix.json')
    apps_dir = os.path.join(test_dir, 'apps')
    ensure_dir(apps_dir)

    all_apps = scripts.run(policies_path, globals_path, apps_dir)

    my_result = get_apps_dict(all_apps, apps_dir)

    eq_(gen_allow_simple3_expect, my_result)