deluge/tests/test_config.py

from twisted.trial import unittest
from twisted.python.failure import Failure

import common
import os

from deluge.config import Config

DEFAULTS = {"string": "foobar", "int": 1, "float": 0.435, "bool": True, "tuple": (1, 2)}

class ConfigTestCase(unittest.TestCase):
    def setUp(self):
        self.config_dir = common.set_tmp_config_dir()

    def test_init(self):
        config = Config("test.conf", defaults=DEFAULTS, config_dir=self.config_dir)
        self.assertEquals(DEFAULTS, config.config)
        
        config = Config("test.conf", config_dir=self.config_dir)
        self.assertEquals({}, config.config)
    
    def test_set_get_item(self):
        config = Config("test.conf", config_dir=self.config_dir)
        config["foo"] = 1
        self.assertEquals(config["foo"], 1)
        self.assertRaises(ValueError, config.set_item, "foo", "bar")
        config["foo"] = 2
        self.assertEquals(config.get_item("foo"), 2)
        
        config._save_timer.cancel()

    def test_load(self):
        def check_config():
            config = Config("test.conf", config_dir=self.config_dir)
            
            self.assertEquals(config["string"], "foobar")
            self.assertEquals(config["float"], 0.435)
                
        # Test loading an old config from 1.1.x
        import pickle
        pickle.dump(DEFAULTS, open(os.path.join(self.config_dir, "test.conf"), "wb"))
        
        check_config()
                
        # Test opening a previous 1.2 config file of just a json object
        import json
        json.dump(DEFAULTS, open(os.path.join(self.config_dir, "test.conf"), "wb"), indent=2)

        check_config()
        
        # Test opening a previous 1.2 config file of having the format versions
        # as ints
        f = open(os.path.join(self.config_dir, "test.conf"), "wb")
        f.write(str(1) + "\n")
        f.write(str(1) + "\n")
        json.dump(DEFAULTS, f, indent=2)
        f.close()
        
        check_config()
        
        # Test the 1.2 config format
        v = {"format": 1, "file": 1}
        f = open(os.path.join(self.config_dir, "test.conf"), "wb")
        json.dump(v, f, indent=2)
        json.dump(DEFAULTS, f, indent=2)
        f.close()
        
        check_config()

    def test_save(self):
        config = Config("test.conf", defaults=DEFAULTS, config_dir=self.config_dir)
        config["string"] = "baz"
        config["int"] = 2
        ret = config.save()
        self.assertTrue(ret)
        del config
        
        config = Config("test.conf", defaults=DEFAULTS, config_dir=self.config_dir)
        self.assertEquals(config["string"], "baz")
        self.assertEquals(config["int"], 2)

    def test_save_timer(self):
        config = Config("test.conf", defaults=DEFAULTS, config_dir=self.config_dir)
        config["string"] = "baz"
        config["int"] = 2
        self.assertTrue(config._save_timer.active())
        
        def check_config(config):
            self.assertTrue(not config._save_timer.active())
            del config
            config = Config("test.conf", defaults=DEFAULTS, config_dir=self.config_dir)
            self.assertEquals(config["string"], "baz")
            self.assertEquals(config["int"], 2)
            
        from twisted.internet.task import deferLater
        from twisted.internet import reactor
        d = deferLater(reactor, 7, check_config, config)
        return d