mirror of
https://github.com/tahoe-lafs/tahoe-lafs.git
synced 2024-12-28 08:48:53 +00:00
102 lines
3.5 KiB
Python
102 lines
3.5 KiB
Python
import os.path
|
|
|
|
from twisted.trial import unittest
|
|
|
|
from allmydata.util import configutil
|
|
from allmydata.test.no_network import GridTestMixin
|
|
from ..scripts import create_node
|
|
from .. import client
|
|
|
|
|
|
class ConfigUtilTests(GridTestMixin, unittest.TestCase):
|
|
|
|
def test_config_utils(self):
|
|
self.basedir = "cli/ConfigUtilTests/test-config-utils"
|
|
self.set_up_grid(oneshare=True)
|
|
tahoe_cfg = os.path.join(self.get_clientdir(i=0), "tahoe.cfg")
|
|
|
|
# test that at least one option was read correctly
|
|
config = configutil.get_config(tahoe_cfg)
|
|
self.failUnlessEqual(config.get("node", "nickname"), "client-0")
|
|
|
|
# test that set_config can mutate an existing option
|
|
configutil.set_config(config, "node", "nickname", "Alice!")
|
|
configutil.write_config(tahoe_cfg, config)
|
|
|
|
config = configutil.get_config(tahoe_cfg)
|
|
self.failUnlessEqual(config.get("node", "nickname"), "Alice!")
|
|
|
|
# test that set_config can set a new option
|
|
descriptor = "Twas brillig, and the slithy toves Did gyre and gimble in the wabe"
|
|
configutil.set_config(config, "node", "descriptor", descriptor)
|
|
configutil.write_config(tahoe_cfg, config)
|
|
|
|
config = configutil.get_config(tahoe_cfg)
|
|
self.failUnlessEqual(config.get("node", "descriptor"), descriptor)
|
|
|
|
def test_config_validation_success(self):
|
|
d = self.mktemp()
|
|
os.mkdir(d)
|
|
fname = os.path.join(d, 'tahoe.cfg')
|
|
|
|
with open(fname, 'w') as f:
|
|
f.write('[node]\nvalid = foo\n')
|
|
|
|
config = configutil.get_config(fname)
|
|
# should succeed, no exceptions
|
|
configutil.validate_config(fname, config, dict(node=['valid']))
|
|
|
|
def test_config_validation_invalid_item(self):
|
|
d = self.mktemp()
|
|
os.mkdir(d)
|
|
fname = os.path.join(d, 'tahoe.cfg')
|
|
|
|
with open(fname, 'w') as f:
|
|
f.write('[node]\nvalid = foo\ninvalid = foo\n')
|
|
|
|
config = configutil.get_config(fname)
|
|
e = self.assertRaises(
|
|
configutil.UnknownConfigError,
|
|
configutil.validate_config,
|
|
fname, config, dict(node=['valid']),
|
|
)
|
|
self.assertIn("section [node] contains unknown option 'invalid'", str(e))
|
|
|
|
def test_config_validation_invalid_section(self):
|
|
d = self.mktemp()
|
|
os.mkdir(d)
|
|
fname = os.path.join(d, 'tahoe.cfg')
|
|
|
|
with open(fname, 'w') as f:
|
|
f.write('[node]\nvalid = foo\n[invalid]\n')
|
|
|
|
config = configutil.get_config(fname)
|
|
e = self.assertRaises(
|
|
configutil.UnknownConfigError,
|
|
configutil.validate_config,
|
|
fname, config, dict(node=['valid']),
|
|
)
|
|
self.assertIn("contains unknown section [invalid]", str(e))
|
|
|
|
def test_create_client_config(self):
|
|
d = self.mktemp()
|
|
os.mkdir(d)
|
|
fname = os.path.join(d, 'tahoe.cfg')
|
|
|
|
with open(fname, 'w') as f:
|
|
opts = {"nickname": "nick",
|
|
"webport": "tcp:3456",
|
|
"hide-ip": False,
|
|
"listen": "none",
|
|
"shares-needed": "1",
|
|
"shares-happy": "1",
|
|
"shares-total": "1",
|
|
}
|
|
create_node.write_node_config(f, opts)
|
|
create_node.write_client_config(f, opts)
|
|
|
|
config = configutil.get_config(fname)
|
|
# should succeed, no exceptions
|
|
configutil.validate_config(fname, config,
|
|
client._valid_config_sections())
|