#!/usr/bin/python

import mock
import os
import sys

class SimpleconfigTest(mock.TestCase):
    def setUp(self):
        self.setupModules(["_isys", "block", "ConfigParser"])        
        self.fs = mock.DiskIO()
        import pyanaconda.simpleconfig
        pyanaconda.simpleconfig.open = self.fs.open
        pyanaconda.simpleconfig.os = mock.Mock()    # Mock os module
        pyanaconda.simpleconfig.os.path = os.path   # except os.path part
        
        # Stuff for IfcfgFile class tests
        self.DIR = '/etc/sysconfig/network-scripts/'
        self.IFACE = 'eth0'
        self.PATH = "%sifcfg-%s" % (self.DIR, self.IFACE)
        self.CONTENT = '# Broadcom Corporation NetXtreme BCM5761 Gigabit Ethernet\n'
        self.CONTENT += 'DEVICE=eth0\n'
        self.CONTENT += 'HWADDR=00:10:18:61:35:98\n'
        self.CONTENT += 'ONBOOT=no\n'      
        self.fs.open(self.PATH, 'w').write(self.CONTENT)
    
    def tearDown(self):
        self.tearDownModules()
    
    def uppercase_ascii_string_letters_test(self):
        """Converting to uppercase (letters)"""
        import pyanaconda.simpleconfig
        ret = pyanaconda.simpleconfig.uppercase_ASCII_string('abcd')
        self.assertEqual(ret, 'ABCD')
        ret = pyanaconda.simpleconfig.uppercase_ASCII_string('aBCD')
        self.assertEqual(ret, 'ABCD')
        ret = pyanaconda.simpleconfig.uppercase_ASCII_string('ABCD')
        self.assertEqual(ret, 'ABCD')
        
    def uppercase_ascii_string_numbers_test(self):
        """Converting to uppercase (numbers)"""
        import pyanaconda.simpleconfig
        ret = pyanaconda.simpleconfig.uppercase_ASCII_string('123')
        self.assertEqual(ret, '123')
        
    def uppercase_ascii_string_others_test(self):
        """Converting to uppercase (special chars)"""
        import pyanaconda.simpleconfig
        ret = pyanaconda.simpleconfig.uppercase_ASCII_string('--')
        self.assertEqual(ret, '--')
        ret = pyanaconda.simpleconfig.uppercase_ASCII_string(' ')
        self.assertEqual(ret, ' ')
        ret = pyanaconda.simpleconfig.uppercase_ASCII_string('')
        self.assertEqual(ret, '')
        
    def set_and_get_test(self):
        """Setting and getting values"""
        import pyanaconda.simpleconfig
        scf = pyanaconda.simpleconfig.SimpleConfigFile()
        scf.set(('key1', 'value1'))
        self.assertEqual(scf.get('key1'), 'value1')
        scf.set(('KEY2', 'value2'))
        self.assertEqual(scf.get('key2'), 'value2')
        scf.set(('KEY3', 'value3'))
        self.assertEqual(scf.get('KEY3'), 'value3')
        scf.set(('key4', 'value4'))
        self.assertEqual(scf.get('KEY4'), 'value4')
        
    def unset_test(self):
        import pyanaconda.simpleconfig
        scf = pyanaconda.simpleconfig.SimpleConfigFile()
        scf.set(('key1', 'value1'))
        scf.unset(('key1'))
        self.assertEqual(scf.get('key1'), '')

    def write_test(self):
        import pyanaconda.simpleconfig
        scf = pyanaconda.simpleconfig.SimpleConfigFile()
        scf.set(('key1', 'value1'))
        scf.write('/tmp/file')
        self.assertEqual(self.fs['/tmp/file'], 'KEY1="value1"\n')
        
    def read_test(self):
        import pyanaconda.simpleconfig
        scf = pyanaconda.simpleconfig.SimpleConfigFile()
        self.fs.open('/tmp/file', 'w').write('KEY1="value1"\n')
        scf.read('/tmp/file')
        self.assertEqual(scf.get('key1'), 'value1')
    
    def ifcfgfile_path_property_test(self):
        import pyanaconda.simpleconfig
        scf = pyanaconda.simpleconfig.IfcfgFile(self.DIR, self.IFACE)        
        self.assertEqual(scf.path, self.PATH)        
    
    def ifcfgfile_read_test(self):
        import pyanaconda.simpleconfig        
        scf = pyanaconda.simpleconfig.IfcfgFile(self.DIR, self.IFACE)        
        scf.read()
        self.assertEqual(scf.get('device'), 'eth0')
        self.assertEqual(scf.get('hwaddr'), '00:10:18:61:35:98')
        self.assertEqual(scf.get('onboot'), 'no')

    def ifcfgfile_read_and_clear_test(self):
        import pyanaconda.simpleconfig        
        scf = pyanaconda.simpleconfig.IfcfgFile(self.DIR, self.IFACE)
        scf.read()
        scf.clear()
        self.assertEqual(scf.get('device'), '')
        self.assertEqual(scf.get('hwaddr'), '')
        self.assertEqual(scf.get('onboot'), '')
        
    def ifcfgfile_write_test(self):
        import pyanaconda.simpleconfig
        scf = pyanaconda.simpleconfig.IfcfgFile(self.DIR, self.IFACE)
        scf.set(('device', 'eth0'))
        scf.set(('hwaddr', '00:11:22:33:44:55'))
        scf.set(('onboot', 'no'))
        scf.write()
        self.assertEqual(self.fs[self.PATH], 
            'DEVICE="eth0"\nHWADDR="00:11:22:33:44:55"\nONBOOT="no"\n')