Phoenix/unittests/test_config.py

import imp_unittest, unittest
import wtc
import wx
import os

cfgFilename = os.path.join(os.path.dirname(__file__), 'cfgtest')

#---------------------------------------------------------------------------

class ConfigTests(wtc.WidgetTestCase):
    
    def writeStuff(self, cfg):
        cfg.SetPath('/one/two/three')
        cfg.Write('key', 'value')
        cfg.WriteInt('int', 123)
        cfg.WriteFloat('float', 123.456)
        cfg.WriteBool('bool', True)
        cfg.Flush()


    def test_Config1(self):
        null = wx.LogNull()
        name = cfgFilename + '_1'
        cfg = wx.Config('unittest_ConfigTests', localFilename=name)
        self.writeStuff(cfg)
        del cfg
        if os.path.exists(name):
            os.remove(name)
        
        
    def test_Config2(self):
        null = wx.LogNull()
        name = cfgFilename + '_2'
        
        cfg = wx.Config('unittest_ConfigTests', localFilename=name)
        self.writeStuff(cfg)

        cfg.SetPath('/before')
        self.assertTrue(cfg.GetPath() == '/before')
        changer = wx.ConfigPathChanger(cfg, '/one/two/three/')
        self.assertTrue(cfg.GetPath() == '/one/two/three')
        del changer
        self.assertTrue(cfg.GetPath() == '/before')
                    
        del cfg
        if os.path.exists(name):
            os.remove(name)


    def test_Config3(self):
        null = wx.LogNull()
        name = cfgFilename + '_3'
        
        cfg = wx.Config('unittest_ConfigTests', localFilename=name)
        self.writeStuff(cfg)

        cfg.SetPath('/before')
        self.assertTrue(cfg.GetPath() == '/before')
        with wx.ConfigPathChanger(cfg, '/one/two/three/'):
            self.assertTrue(cfg.GetPath() == '/one/two/three')
        self.assertTrue(cfg.GetPath() == '/before')
                    
        del cfg
        if os.path.exists(name):
            os.remove(name)


    def test_Config4(self): 
        null = wx.LogNull()
        name = cfgFilename + '_4'
        
        cfg0 = wx.Config('unittest_ConfigTests', localFilename=name)
        wx.Config.Set(cfg0)

        cfg = wx.Config.Get(False)
        #self.assertTrue(cfg is cfg0)
        self.writeStuff(cfg)
        del cfg
        
        cfg = wx.Config.Get()
        cfg.SetPath('/one/two/three')
        self.assertTrue(cfg.GetPath() == '/one/two/three')
        self.assertTrue(cfg.GetNumberOfEntries() == 4)
        self.assertTrue(cfg.Read('key') == 'value')
        self.assertTrue(cfg.ReadInt('int') == 123)
        self.assertTrue(cfg.ReadFloat('float') == 123.456)
        self.assertTrue(cfg.ReadBool('bool') == True)
        self.assertTrue(cfg.Read('no-value', 'defvalue') == 'defvalue')

        wx.Config.Set(None)
        self.myYield()
        if os.path.exists(name):
            os.remove(name)


    def test_Config5(self):
        null = wx.LogNull()
        name = cfgFilename + '_5'
        
        cfg = wx.Config('unittest_ConfigTests', localFilename=name)
        cfg.SetPath('/zero')
        cfg.Write('key1', 'value')
        cfg.Write('key2', 'value')
        cfg.Write('key3', 'value')
        
        cfg.SetPath('/one')
        cfg.Write('key1', 'value')
        cfg.Write('key2', 'value')
        cfg.Write('key3', 'value')

        cfg.SetPath('/two')
        cfg.Write('key1', 'value')
        cfg.Write('key2', 'value')
        cfg.Write('key3', 'value')

        cfg.SetPath('/three')
        cfg.Write('key1', 'value')
        cfg.Write('key2', 'value')
        cfg.Write('key3', 'value')

        cfg.Flush()
        
        cfg.SetPath('/')
        count = 0
        more, group, index = cfg.GetFirstGroup()
        while more:
            count += 1
            more, group, index = cfg.GetNextGroup(index)
        self.assertEqual(count, 4)
        
        cfg.SetPath('/two')
        count = 0
        more, entry, index = cfg.GetFirstEntry()
        while more:
            count += 1
            more, entry, index = cfg.GetNextEntry(index)
        self.assertEqual(count, 3)
                    
        del cfg
        if os.path.exists(name):
            os.remove(name)


    def test_Config6(self):
        null = wx.LogNull()
        name = cfgFilename + '_6'
        
        cfg = wx.FileConfig('unittest_ConfigTests', localFilename=name)
        cfg.SetPath('/zero')
        cfg.Write('key1', 'value')
        cfg.Write('key2', 'value')
        cfg.Write('key3', 'value')
        
        cfg.SetPath('/one')
        cfg.Write('key1', 'value')
        cfg.Write('key2', 'value')
        cfg.Write('key3', 'value')

        cfg.SetPath('/two')
        cfg.Write('key1', 'value')
        cfg.Write('key2', 'value')
        cfg.Write('key3', 'value')

        cfg.SetPath('/three')
        cfg.Write('key1', 'value')
        cfg.Write('key2', 'value')
        cfg.Write('key3', 'value')

        cfg.Flush()
        
        cfg.SetPath('/')
        count = 0
        more, group, index = cfg.GetFirstGroup()
        while more:
            count += 1
            more, group, index = cfg.GetNextGroup(index)
        self.assertEqual(count, 4)
        
        cfg.SetPath('/two')
        count = 0
        more, entry, index = cfg.GetFirstEntry()
        while more:
            count += 1
            more, entry, index = cfg.GetNextEntry(index)
        self.assertEqual(count, 3)
                    
        del cfg
        if os.path.exists(name):
            os.remove(name)
       
#---------------------------------------------------------------------------


if __name__ == '__main__':
    unittest.main()