1.1 --- a/OpenSecurity/bin/test_vmmanager.pyw	Thu Jul 17 10:20:10 2014 +0100
     1.2 +++ b/OpenSecurity/bin/test_vmmanager.pyw	Fri Jul 18 13:45:09 2014 +0100
     1.3 @@ -35,10 +35,12 @@
     1.4  import unittest
     1.5  
     1.6  import os
     1.7 +import re
     1.8  import os.path
     1.9  import sys
    1.10  import cygwin
    1.11  import vmmanager
    1.12 +import _winreg
    1.13  gvm_mgr = None
    1.14  
    1.15  class TestVMManager(unittest.TestCase):
    1.16 @@ -57,11 +59,43 @@
    1.17          template = vmmanager.VMManager.getTemplateUUID()
    1.18          self.assertIsNotNone(template,  "returned no UUID for template")
    1.19      
    1.20 -    #@unittest.skip("skipping (requires running vmmanager)")    
    1.21 +    @unittest.skip("skipping")    
    1.22      def testUpdateTemplate(self):
    1.23          gvm_mgr.updateTemplate()
    1.24          pass
    1.25      
    1.26 +    def setKey(self, key, name, value):
    1.27 +        _, reg_type = _winreg.QueryValueEx(key, name)
    1.28 +        _winreg.SetValueEx(key, name, 0, reg_type, value)
    1.29 +    
    1.30 +    @unittest.skip("skipping")    
    1.31 +    def testGetProxySettings(self):
    1.32 +        #sudo echo "http_proxy=http://80.122.169.38:8080/" >> /etc/environment
    1.33 +        aReg = _winreg.ConnectRegistry(None,_winreg.HKEY_CURRENT_USER)
    1.34 +        aKey = _winreg.OpenKey(aReg, r"Software\Microsoft\Windows\CurrentVersion\Internet Settings")
    1.35 +        subCount, valueCount, lastModified = _winreg.QueryInfoKey(aKey)
    1.36 +        proxy = dict()
    1.37 +        for i in range(valueCount):                                           
    1.38 +            try:
    1.39 +                n,v,t = _winreg.EnumValue(aKey,i)
    1.40 +                proxy[n] = v
    1.41 +            except EnvironmentError:                                               
    1.42 +                break
    1.43 +        _winreg.CloseKey(aKey)
    1.44 +        print proxy
    1.45 +        if 'ProxyEnable' in proxy.keys() and proxy['ProxyEnable'] == 1:
    1.46 +            print proxy['ProxyServer']
    1.47 +            return proxy['ProxyServer']
    1.48 +        else: 
    1.49 +            return ""
    1.50 +        
    1.51 +    def testMatchProxy(self):
    1.52 +        #http=212.17.86.109:8080;https=212.17.86.109:8080;ftp=212.17.86.109:8080
    1.53 +        #212.17.86.109:8080
    1.54 +        text = 'http=212.17.86.109:8080;https=212.17.86.109:8080;ftp=212.17.86.109:8080'
    1.55 +        print re.search(r"(?<=http=)(?P<HttpProxy>.*?)(?=;)", text).groupdict()
    1.56 +        print re.search(r"(?<=http=)(.*?)(?=;)", text)
    1.57 +        
    1.58      #@classmethod
    1.59      #def tearOffClass(self):
    1.60      #    gvm_mgr.stop()