2022-12-04 11:58:32 -08:00
|
|
|
# Copyright 2022 Google LLC
|
2022-10-28 05:40:53 -07:00
|
|
|
#
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
# you may not use this file except in compliance with the License.
|
|
|
|
# You may obtain a copy of the License at
|
|
|
|
#
|
|
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
#
|
|
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
# See the License for the specific language governing permissions and
|
|
|
|
# limitations under the License.
|
|
|
|
|
2022-12-02 11:07:15 -08:00
|
|
|
import pytest
|
2022-11-03 03:41:53 -07:00
|
|
|
|
2022-12-02 11:07:15 -08:00
|
|
|
_params = ['boolean', 'list']
|
2022-11-03 03:41:53 -07:00
|
|
|
|
|
|
|
|
2022-12-02 11:07:15 -08:00
|
|
|
@pytest.mark.parametrize('policy_type', _params)
|
2022-12-05 02:20:20 -08:00
|
|
|
def test_policy_factory(plan_summary, tfvars_to_yaml, tmp_path, policy_type):
|
2022-11-04 10:37:23 -07:00
|
|
|
dest = tmp_path / 'policies.yaml'
|
2022-12-02 11:07:15 -08:00
|
|
|
tfvars_to_yaml(f'org_policies_{policy_type}.tfvars', dest, 'org_policies')
|
2022-12-05 02:20:20 -08:00
|
|
|
tfvars_plan = plan_summary(
|
2022-12-02 11:07:15 -08:00
|
|
|
'modules/organization',
|
|
|
|
tf_var_files=['common.tfvars', f'org_policies_{policy_type}.tfvars'])
|
2022-12-05 02:20:20 -08:00
|
|
|
yaml_plan = plan_summary('modules/organization',
|
|
|
|
tf_var_files=['common.tfvars'],
|
|
|
|
org_policies_data_path=f'{tmp_path}')
|
2022-12-02 11:07:15 -08:00
|
|
|
assert tfvars_plan.values == yaml_plan.values
|
|
|
|
|
|
|
|
|
2022-12-05 02:20:20 -08:00
|
|
|
def test_custom_constraint_factory(plan_summary, tfvars_to_yaml, tmp_path):
|
2022-11-08 09:10:13 -08:00
|
|
|
dest = tmp_path / 'constraints.yaml'
|
2022-12-02 11:07:15 -08:00
|
|
|
tfvars_to_yaml(f'org_policies_custom_constraints.tfvars', dest,
|
|
|
|
'org_policy_custom_constraints')
|
2022-12-05 02:20:20 -08:00
|
|
|
tfvars_plan = plan_summary(
|
2022-12-02 11:07:15 -08:00
|
|
|
'modules/organization',
|
|
|
|
tf_var_files=['common.tfvars', f'org_policies_custom_constraints.tfvars'])
|
2022-12-05 02:20:20 -08:00
|
|
|
yaml_plan = plan_summary(
|
2022-12-02 11:07:15 -08:00
|
|
|
'modules/organization', tf_var_files=['common.tfvars'],
|
|
|
|
org_policy_custom_constraints_data_path=f'{tmp_path}')
|
|
|
|
assert tfvars_plan.values == yaml_plan.values
|