67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
# Copyright 2022 Google LLC
|
|
#
|
|
# 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.
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def resources(plan_runner, version):
|
|
# convert `version` to a boolean suitable for the `v2` variable
|
|
v2 = {'v1': 'false', 'v2': 'true'}[version]
|
|
_, resources = plan_runner(v2=v2)
|
|
return resources
|
|
|
|
@pytest.fixture
|
|
def plan(plan_runner, version, count):
|
|
v2 = {'v1': 'false', 'v2': 'true'}[version]
|
|
plan, _ = plan_runner(v2=v2, instance_count=count)
|
|
return plan
|
|
|
|
|
|
@pytest.mark.parametrize('version', ['v1', 'v2'])
|
|
def test_resource_count(resources):
|
|
"Test number of resources created."
|
|
assert len(resources) == 3
|
|
|
|
|
|
@pytest.mark.parametrize('version', ['v1', 'v2'])
|
|
def test_iam(resources, version):
|
|
"Test IAM binding resources."
|
|
|
|
types = {
|
|
'v1': 'google_cloudfunctions_function_iam_binding',
|
|
'v2': 'google_cloudfunctions2_function_iam_binding'
|
|
}
|
|
|
|
bindings = [r['values'] for r in resources if r['type'] == types[version]]
|
|
assert len(bindings) == 1
|
|
assert bindings[0]['role'] == 'roles/cloudfunctions.invoker'
|
|
assert bindings[0]['members'] == ['allUsers']
|
|
|
|
|
|
@pytest.mark.parametrize('version', ['v1', 'v2'])
|
|
@pytest.mark.parametrize('count', [2])
|
|
def test_multiple_functions(plan, version, count):
|
|
"""Tests whether multiple use of functions result in use of multiple bundle files"""
|
|
|
|
# data objects are only accessible in prior_state of the plan, they are not present in root_module
|
|
archives = [
|
|
resource for child_module in plan['prior_state']["values"]["root_module"]["child_modules"]
|
|
for resource in child_module["resources"]
|
|
if resource["type"] == "archive_file"
|
|
]
|
|
file_names = set([x["values"]["output_path"] for x in archives])
|
|
|
|
assert len(archives) == len(file_names)
|