115 lines
2.8 KiB
Python
115 lines
2.8 KiB
Python
# Copyright 2017 IBM
|
|
#
|
|
# 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.
|
|
|
|
# Implementation of ini add / remove for devstack. We don't use the
|
|
# python ConfigFile parser because that ends up rewriting the entire
|
|
# file and doesn't ensure comments remain.
|
|
|
|
import fixtures
|
|
import testtools
|
|
|
|
from devstack import dsconf
|
|
|
|
|
|
BASIC = """
|
|
[[local|localrc]]
|
|
a = b
|
|
c = d
|
|
f = 1
|
|
[[post-config|$NEUTRON_CONF]]
|
|
[DEFAULT]
|
|
global_physnet_mtu=1450
|
|
[[post-config|$NOVA_CONF]]
|
|
[upgrade_levels]
|
|
compute = auto
|
|
"""
|
|
|
|
RESULT1 = """
|
|
[[local|localrc]]
|
|
a = b
|
|
c = d
|
|
f = 1
|
|
[[post-config|$NEUTRON_CONF]]
|
|
[DEFAULT]
|
|
global_physnet_mtu = 1400
|
|
[[post-config|$NOVA_CONF]]
|
|
[upgrade_levels]
|
|
compute = auto
|
|
"""
|
|
|
|
RESULT2 = """
|
|
[[local|localrc]]
|
|
a = b
|
|
c = d
|
|
f = 1
|
|
[[post-config|$NEUTRON_CONF]]
|
|
[DEFAULT]
|
|
global_physnet_mtu=1450
|
|
[oslo_policy]
|
|
policy_file = /etc/neutron/policy.json
|
|
[[post-config|$NOVA_CONF]]
|
|
[upgrade_levels]
|
|
compute = auto
|
|
"""
|
|
|
|
RESULT3 = """
|
|
[[local|localrc]]
|
|
a = b
|
|
c = d
|
|
f = 1
|
|
[[post-config|$NEUTRON_CONF]]
|
|
[DEFAULT]
|
|
global_physnet_mtu=1450
|
|
[[post-config|$NOVA_CONF]]
|
|
[upgrade_levels]
|
|
compute = auto
|
|
[[post-config|$GLANCE_CONF]]
|
|
[DEFAULT]
|
|
graceful_shutdown_timeout = 5
|
|
"""
|
|
|
|
|
|
class TestLcSet(testtools.TestCase):
|
|
|
|
def setUp(self):
|
|
super(TestLcSet, self).setUp()
|
|
self._path = self.useFixture(fixtures.TempDir()).path
|
|
self._path += "/local.conf"
|
|
with open(self._path, "w") as f:
|
|
f.write(BASIC)
|
|
|
|
def test_set_existing(self):
|
|
conf = dsconf.LocalConf(self._path)
|
|
conf.set("post-config", "$NEUTRON_CONF", "DEFAULT",
|
|
"global_physnet_mtu", "1400")
|
|
with open(self._path) as f:
|
|
content = f.read()
|
|
self.assertEqual(content, RESULT1)
|
|
|
|
def test_set_new(self):
|
|
conf = dsconf.LocalConf(self._path)
|
|
conf.set("post-config", "$NEUTRON_CONF", "oslo_policy",
|
|
"policy_file", "/etc/neutron/policy.json")
|
|
with open(self._path) as f:
|
|
content = f.read()
|
|
self.assertEqual(content, RESULT2)
|
|
|
|
def test_set_new_section(self):
|
|
conf = dsconf.LocalConf(self._path)
|
|
conf.set("post-config", "$GLANCE_CONF", "DEFAULT",
|
|
"graceful_shutdown_timeout", "5")
|
|
with open(self._path) as f:
|
|
content = f.read()
|
|
self.assertEqual(content, RESULT3)
|