71 lines
2.7 KiB
Python
71 lines
2.7 KiB
Python
# Copyright 2021 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 os
|
|
import pytest
|
|
|
|
|
|
FIXTURES_DIR = os.path.join(os.path.dirname(__file__), 'fixture')
|
|
|
|
|
|
def test_external_addresses(plan_runner):
|
|
addresses = '{one = "europe-west1", two = "europe-west2"}'
|
|
_, resources = plan_runner(FIXTURES_DIR, external_addresses=addresses)
|
|
assert [r['values']['name'] for r in resources] == ['one', 'two']
|
|
assert set(r['values']['address_type']
|
|
for r in resources) == set(['EXTERNAL'])
|
|
assert [r['values']['region']
|
|
for r in resources] == ['europe-west1', 'europe-west2']
|
|
|
|
|
|
def test_global_addresses(plan_runner):
|
|
_, resources = plan_runner(FIXTURES_DIR, global_addresses='["one", "two"]')
|
|
assert [r['values']['name'] for r in resources] == ['one', 'two']
|
|
assert set(r['values']['address_type'] for r in resources) == set([None])
|
|
|
|
|
|
def test_internal_addresses(plan_runner):
|
|
addresses = (
|
|
'{one = {region = "europe-west1", subnetwork = "foobar"}, '
|
|
'two = {region = "europe-west2", subnetwork = "foobarz"}}'
|
|
)
|
|
_, resources = plan_runner(FIXTURES_DIR, internal_addresses=addresses)
|
|
assert [r['values']['name'] for r in resources] == ['one', 'two']
|
|
assert set(r['values']['address_type']
|
|
for r in resources) == set(['INTERNAL'])
|
|
assert [r['values']['region']
|
|
for r in resources] == ['europe-west1', 'europe-west2']
|
|
|
|
|
|
def test_internal_addresses_config(plan_runner):
|
|
addresses = (
|
|
'{one = {region = "europe-west1", subnetwork = "foobar"}, '
|
|
'two = {region = "europe-west2", subnetwork = "foobarz"}}'
|
|
)
|
|
config = (
|
|
'{one = {address = "10.0.0.2", purpose = "SHARED_LOADBALANCER_VIP", '
|
|
'tier=null}}'
|
|
)
|
|
_, resources = plan_runner(FIXTURES_DIR,
|
|
internal_addresses=addresses,
|
|
internal_addresses_config=config)
|
|
assert [r['values']['name'] for r in resources] == ['one', 'two']
|
|
assert set(r['values']['address_type']
|
|
for r in resources) == set(['INTERNAL'])
|
|
assert [r['values'].get('address')
|
|
for r in resources] == ['10.0.0.2', None]
|
|
assert [r['values'].get('purpose')
|
|
for r in resources] == ['SHARED_LOADBALANCER_VIP', None]
|