87 lines
2.6 KiB
Python
Executable File
87 lines
2.6 KiB
Python
Executable File
# Copyright 2016, Rackspace US, Inc.
|
|
#
|
|
# 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.
|
|
from migrate_openstack_vars import main
|
|
from migrate_openstack_vars import VAR_MAPPINGS
|
|
|
|
import os
|
|
import sys
|
|
|
|
|
|
FILE_NAME = 'test_user_variables.yml'
|
|
|
|
|
|
def set_up():
|
|
# Create an example file with key/value pairs, as well as a comment
|
|
# The old to new value mappings are written to a file, then later the
|
|
# file is inspected to ensure no old values remain.
|
|
var_lines = ["{}: {}".format(key, val) for
|
|
key, val in VAR_MAPPINGS.items()]
|
|
var_lines.append('# A test comment')
|
|
sample = VAR_MAPPINGS.items()[0]
|
|
var_lines.append('# {} / {}'.format(*sample))
|
|
with open(FILE_NAME, 'w') as f:
|
|
f.write('\n'.join(var_lines))
|
|
|
|
|
|
def teardown():
|
|
# Remove files so they don't pollute the directories.
|
|
os.remove(FILE_NAME)
|
|
|
|
def test():
|
|
main(FILE_NAME)
|
|
|
|
with open(FILE_NAME, 'r') as f:
|
|
contents = f.readlines()
|
|
|
|
for line in contents:
|
|
# only split lines that look like a key/value pair.
|
|
if ':' in line:
|
|
var, value = line.split(':', 1)
|
|
value = value.strip()
|
|
elif '/' in line:
|
|
# For the comment containing a variable, clean up the list
|
|
# contents before assigning the parts we want to test.
|
|
parts = line.split()
|
|
parts.remove('#')
|
|
parts.remove('/')
|
|
var, value = parts
|
|
else:
|
|
var = value = line
|
|
|
|
|
|
# Once run through the 'main' function, the keys and values should
|
|
# match
|
|
if not value == var:
|
|
import pdb; pdb.set_trace() # NOQA
|
|
print("Var and value don't match.")
|
|
print("Var: {}, Value: {}".format(var, value))
|
|
sys.exit()
|
|
|
|
invalid_variable = var not in VAR_MAPPINGS.values()
|
|
# Comments aren't in our test mapping, so make sure we ignore them
|
|
is_comment = line.startswith('#')
|
|
|
|
if invalid_variable and not is_comment:
|
|
err = "Variable {} doesn't appear to be a valid new name."
|
|
sys.exit(err.format(var))
|
|
|
|
|
|
print("Tests passed")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
set_up()
|
|
test()
|
|
teardown()
|