integrating special response object to make pull

This commit is contained in:
shockrah 2020-11-04 00:25:55 -08:00
parent 4d85c094cc
commit 84f04c20cd

View File

@ -1,6 +1,15 @@
import time
import subprocess, os
import json, requests
class Response:
def __init__(self, body, code):
self.body = body
self.code = code
def __str__(self):
return f'{self.code} => {self.body}'
class Test:
def __init__(self, base='http://localhost:8888', create_admin=False, admin=None):
'''
@ -54,19 +63,31 @@ class Test:
def post(self, url, **opts):
'''
@returns text of response
'''
body = self._build_req_body(**opts)
r = requests.post(self.base + url, data=body)
self.log(self.base + url, 'POST', r)
return r.text
def get(self, url, **opts):
'''
@returns text of response
'''
body = self._build_req_body(**opts)
r = requests.get(self.base + url, data=body)
self.log(self.base + url, 'GET', r)
return r.text
def delete(self, url, **opts):
'''
@returns text of response
'''
body = self._build_req_body(**opts)
r = requests.delete(self.base + url, data=body)
self.log(self.base + url, 'DELETE', r)
return r.text
def creds(self):
return self.body
@ -82,8 +103,9 @@ if __name__ == '__main__':
VOICE_CHANNEL = 1
TEXT_CHANNEL = 2
worker.post('/channels/create', name='something random', kind=TEXT_CHANNEL)
worker.get('/channels/list')
new_channel = worker.post('/channels/create', name=f'{time.time()}', kind=TEXT_CHANNEL)
print(f'Channel data: {new_channel}')
channel_list = json.loads(worker.get('/channels/list'))
worker.delete('/channels/delete', name='something random')
worker.get('/channels/list')