def __init__( self, host_base, broker_name=None, broker_passphase=None, 
                  sample_path=None, result_path=None):
        SCClientConnection.__init__(self, host_base, broker_name, broker_passphase)

        # pseudo sample path
        self.sample_path = sample_path
        self.result_path = result_path
        
        if os.path.isdir(self.result_path) == False:
            os.makedirs(self.result_path)
Пример #2
0
    def __init__(self,
                 host_base,
                 broker_name=None,
                 broker_passphase=None,
                 sample_path=None,
                 result_path=None):
        SCClientConnection.__init__(self, host_base, broker_name,
                                    broker_passphase)

        # pseudo sample path
        self.sample_path = sample_path
        self.result_path = result_path

        if os.path.isdir(self.result_path) == False:
            os.makedirs(self.result_path)
Пример #3
0
    def make_request(self,
                     action='',
                     params=None,
                     headers=None,
                     data='',
                     method='GET'):

        # local var
        fake = None

        if self.sample_path:
            reqfile = action.replace('/', '^')
            reqfile = os.path.join(self.sample_path,
                                   '[Response]%s.xml' % (reqfile))
            rf = io.open(reqfile, 'r')
            fake = urllib.addinfourl(rf, 'Fake info', regfile, 400)
            return fake

        else:
            reqfile = action.replace('/', '^')
            reqfile = os.path.join(self.result_path,
                                   '[Request(%s)]%s.xml' % (method, reqfile))
            resfile = action.replace('/', '^')
            resfile = os.path.join(self.result_path,
                                   '[Response]%s.xml' % (resfile))

            # serialize Request data
            rf = io.open(reqfile, 'w')
            if (data is not None) and (len(data) != 0):
                rf.write(self.nice_format(data))
            rf.close()

            # make request to securecloud
            response = SCClientConnection.make_request(self, action, params,
                                                       headers, data, method)
            if response:
                body = response.read()

                #make fake response
                resfp = StringIO.StringIO(body)
                fake = urllib.addinfourl(resfp, response.info(),
                                         response.geturl(), response.getcode())

                # serialize Response data
                f = io.open(resfile, 'w')
                if (body is not None) and (len(body) != 0):
                    f.write(self.nice_format(body))
                f.close()
                return fake

        return fake
    def make_request(self, action='', params=None, headers=None, data='', method='GET'):

        # local var
        fake = None

        if self.sample_path:
            reqfile = action.replace('/', '^')
            reqfile = os.path.join(self.sample_path, '[Response]%s.xml' % (reqfile) )
            rf = io.open(reqfile, 'r')
            fake = urllib.addinfourl(rf, 'Fake info', regfile, 400)
            return fake
        
        else:
            reqfile = action.replace('/', '^')
            reqfile = os.path.join(self.result_path, '[Request(%s)]%s.xml' % (method, reqfile ) )
            resfile = action.replace('/', '^')
            resfile = os.path.join(self.result_path, '[Response]%s.xml' % (resfile) )

            # serialize Request data
            rf = io.open(reqfile, 'w')
            if (data is not None) and (len(data) != 0):
                rf.write(self.nice_format(data))
            rf.close()

            # make request to securecloud
            response = SCClientConnection.make_request(self, action, params, headers, data, method)
            if response:
                body = response.read()
                
                #make fake response
                resfp = StringIO.StringIO(body)
                fake = urllib.addinfourl(resfp, response.info(), response.geturl(), response.getcode())
                
                # serialize Response data
                f = io.open(resfile, 'w')
                if (body is not None) and (len(body) != 0):
                    f.write(self.nice_format(body))
                f.close()
                return fake

        return fake