120 lines
3.5 KiB
Python
120 lines
3.5 KiB
Python
from .context import easymapping
|
|
import pytest
|
|
import os
|
|
import yaml
|
|
|
|
|
|
def load_fixture(file):
|
|
path = os.path.dirname(os.path.realpath(__file__))
|
|
with open(path + "/fixtures/" + file, 'r') as content_file:
|
|
lineList = content_file.readlines()
|
|
|
|
return lineList
|
|
|
|
|
|
def test_parser_doesnt_crash():
|
|
lineList = load_fixture("no-services")
|
|
|
|
result = {
|
|
"customerrors": False
|
|
}
|
|
|
|
cfg = easymapping.HaproxyConfigGenerator(result, "/tmp")
|
|
haproxy_config = cfg.generate(lineList)
|
|
|
|
assert len(haproxy_config) > 0
|
|
path = os.path.dirname(os.path.realpath(__file__))
|
|
with open(path + "/expected/no-services.txt", 'r') as expected_file:
|
|
assert expected_file.read() == haproxy_config
|
|
|
|
|
|
def test_parser_finds_services():
|
|
lineList = load_fixture("services")
|
|
|
|
result = {
|
|
"customerrors": False
|
|
}
|
|
|
|
cert_file = "/tmp/www.somehost.com.br.1.pem"
|
|
if os.path.exists(cert_file):
|
|
os.remove(cert_file)
|
|
|
|
cfg = easymapping.HaproxyConfigGenerator(result, "/tmp")
|
|
haproxy_config = cfg.generate(lineList)
|
|
|
|
assert len(haproxy_config) > 0
|
|
path = os.path.dirname(os.path.realpath(__file__))
|
|
with open(path + "/expected/services.txt", 'r') as expected_file:
|
|
assert expected_file.read() == haproxy_config
|
|
|
|
with open(cert_file, 'r') as expected_file:
|
|
assert expected_file.read() == "Some PEM Certificate"
|
|
|
|
def test_parser_finds_services_changed_label():
|
|
lineList = load_fixture("services-changed-label")
|
|
|
|
result = {
|
|
"customerrors": False,
|
|
"lookup_label": "haproxy"
|
|
}
|
|
|
|
cert_file = "/tmp/www.somehost.com.br.1.pem"
|
|
if os.path.exists(cert_file):
|
|
os.remove(cert_file)
|
|
|
|
cfg = easymapping.HaproxyConfigGenerator(result, "/tmp")
|
|
haproxy_config = cfg.generate(lineList)
|
|
|
|
assert len(haproxy_config) > 0
|
|
path = os.path.dirname(os.path.realpath(__file__))
|
|
with open(path + "/expected/services.txt", 'r') as expected_file:
|
|
assert expected_file.read() == haproxy_config
|
|
|
|
with open(cert_file, 'r') as expected_file:
|
|
assert expected_file.read() == "Some PEM Certificate"
|
|
|
|
|
|
def test_parser_static():
|
|
path = os.path.dirname(os.path.realpath(__file__))
|
|
with open(path + "/fixtures/static.yml", 'r') as content_file:
|
|
parsed = yaml.load(content_file.read(), Loader=yaml.FullLoader)
|
|
|
|
cfg = easymapping.HaproxyConfigGenerator(parsed, "/tmp")
|
|
haproxy_config = cfg.generate()
|
|
assert len(haproxy_config) > 0
|
|
|
|
with open(path + "/expected/static.txt", 'r') as expected_file:
|
|
assert expected_file.read() == haproxy_config
|
|
|
|
|
|
def test_parser_tcp():
|
|
lineList = load_fixture("services-tcp")
|
|
|
|
result = {
|
|
"customerrors": False
|
|
}
|
|
|
|
cfg = easymapping.HaproxyConfigGenerator(result, "/tmp")
|
|
haproxy_config = cfg.generate(lineList)
|
|
# print(haproxy_config)
|
|
|
|
assert len(haproxy_config) > 0
|
|
path = os.path.dirname(os.path.realpath(__file__))
|
|
with open(path + "/expected/services-tcp.txt", 'r') as expected_file:
|
|
assert expected_file.read() == haproxy_config
|
|
|
|
def test_parser_multi_containers():
|
|
lineList = load_fixture("services-multi-containers")
|
|
|
|
result = {
|
|
"customerrors": False
|
|
}
|
|
|
|
cfg = easymapping.HaproxyConfigGenerator(result, "/tmp")
|
|
haproxy_config = cfg.generate(lineList)
|
|
|
|
assert len(haproxy_config) > 0
|
|
path = os.path.dirname(os.path.realpath(__file__))
|
|
with open(path + "/expected/services-multi-containers.txt", 'r') as expected_file:
|
|
assert expected_file.read() == haproxy_config
|
|
|