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)
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)
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