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