view tests/test-config.py @ 89:351d2799f145

client: added context manager protocol methods

This allows client objects to be used with the 'with' statement and be closed
automatically when the 'with' context exits.

import os, common, hglib

class test_config(common.basetest):
    def setUp(self):
        common.basetest.setUp(self)
        open('.hg/hgrc', 'a').write('[section]\nkey=value\n')
        self.client = hglib.open()

    def test_basic(self):
        config = self.client.config()

        self.assertTrue(('section', 'key', 'value') in self.client.config())

        self.assertTrue([('section', 'key', 'value')],
                        self.client.config('section'))
        self.assertTrue([('section', 'key', 'value')],
                        self.client.config(['section', 'foo']))
        self.assertRaises(hglib.error.CommandError,
                          self.client.config, ['a.b', 'foo'])

    def test_show_source(self):
        config = self.client.config(showsource=True)

        self.assertTrue((os.path.abspath('.hg/hgrc') + ':2',
                         'section', 'key', 'value') in config)