def retrieve_token(sso_server = sso_server, partner = partner, payload = payload, another = False):
""" Retrieve token grant by posting to /oauth/saml_grant/<partner_subdomain>
Parameters:
sso_server
partner subdomain
payload includes client_id and assertion
another True - Retrieve a new SAML. False - Return the existing one
Return:
SAML
"""
global token
if token and not another:
# Not need another and return the existing one
return token
elif not token and not another:
# Not need another but not have an existing one, retrieve one
res = Resource("http://"+sso_server)
r = res.post('/oauth/saml_grant/'+partner, payload=payload, headers=headers)
token = ast.literal_eval(r)
logging.info("The token retrieved is '%s'" % str(token))
return token
else:
# Retrieve another one
res = Resource("http://"+sso_server)
r = res.post('/oauth/saml_grant/'+partner, payload=payload, headers=headers)
ntoken = ast.literal_eval(r)
logging.info("The another token retrieved is '%s'" % str(ntoken))
return ntoken
class TestNodeTasks(unittest.TestCase):
def setUp(self):
self.client=Resource('http://127.0.0.1:8000')
self.client.post('/environments',headers={'Content-Type': 'application/json'},payload='{"name":"test"}')
self.client.post('/environments/0/nodes',headers={'Content-Type': 'application/json'},payload='{"name":"testNode","type":"reprap"}')
def test_TaskCreate(self):
response=self.client.post('/environments/0/nodes/0/tasks',headers={'Content-Type': 'application/json'},payload='{"name":"testTask","type":"print"}')
self.failUnlessEqual(200, self.client.status)
def test_TaskRead(self):
self.client.get('/environments/0/nodes/0/connector',headers={'Content-Type': 'application/json'})
self.failUnlessEqual(200, self.client.status)
def test_ConnectorUpdate(self):
self.client.put('/environments/0/nodes/0/connector',headers={'Content-Type': 'application/json'},payload='{"driverType":"teacup","driverParams":{"speed":115200}}')
self.failUnlessEqual(200, self.client.status)
def test_ConnectorDelete(self):
self.client.delete('/environments/0/nodes/0/connector',headers={'Content-Type': 'application/json'})
self.failUnlessEqual(200, self.client.status)
def test_ConnectorStatusUpdate(self):
self.client.put('/environments/0/nodes/0/connector',headers={'Content-Type': 'application/json'},payload='{"driverType":"teacup","driverParams":{"speed":115200}}')
self.failUnlessEqual(200, self.client.status)
self.client.post('/environments/0/nodes/0/connector/status',headers={'Content-Type': 'application/json'},payload='{"connected":"True"}')
self.failUnlessEqual(200, self.client.status)
def tearDown(self):
self.client.delete('/environments',headers={'Content-Type': 'application/json'})
class TestNodeConnector(unittest.TestCase):
def setUp(self):
self.client=Resource('http://127.0.0.1:8000')
self.client.post('/environments',headers={'Content-Type': 'application/json'},payload='{"name":"test"}')
self.client.post('/environments/0/nodes',headers={'Content-Type': 'application/json'},payload='{"name":"testNode","type":"reprap"}')
def test_ConnectorCreate(self):
self.client.put('/environments/0/nodes/0/connector',headers={'Content-Type': 'application/json'},payload='{"driverType":"teacup","driverParams":{"speed":115200}}')
self.failUnlessEqual(200, self.client.status)
def test_ConnectorRead(self):
self.client.get('/environments/0/nodes/0/connector',headers={'Content-Type': 'application/json'})
self.failUnlessEqual(200, self.client.status)
def test_ConnectorUpdate(self):
self.client.put('/environments/0/nodes/0/connector',headers={'Content-Type': 'application/json'},payload='{"driverType":"teacup","driverParams":{"speed":115200}}')
self.failUnlessEqual(200, self.client.status)
def test_ConnectorDelete(self):
self.client.delete('/environments/0/nodes/0/connector',headers={'Content-Type': 'application/json'})
self.failUnlessEqual(200, self.client.status)
def test_ConnectorStatusUpdate(self):
self.client.put('/environments/0/nodes/0/connector',headers={'Content-Type': 'application/json'},payload='{"driverType":"teacup","driverParams":{"speed":115200}}')
self.failUnlessEqual(200, self.client.status)
self.client.post('/environments/0/nodes/0/connector/status',headers={'Content-Type': 'application/json'},payload='{"connected":"True"}')
self.failUnlessEqual(200, self.client.status)
def tearDown(self):
self.client.delete('/environments',headers={'Content-Type': 'application/json'})
class TestNodes(unittest.TestCase):
def setUp(self):
self.client=Resource('http://127.0.0.1:8000')
response=self.client.put('/environments',headers={'Content-Type': 'application/json'},payload='{"name":"test"}')
self.failUnlessEqual(200, self.client.status)
def test_NodeCreate(self):
response=self.client.post('/environments/0/nodes',headers={'Content-Type': 'application/json'},payload='{"name":"testNode","type":"reprap"}')
self.failUnlessEqual(200, self.client.status)
def test_NodesRead(self):
self.client.get('/environments/0/nodes',headers={'Content-Type': 'application/json'})
self.failUnlessEqual(200, self.client.status)
def test_NodeUpdate(self):
pass
def test_NodesDelete(self):
self.client.delete('/environments/0/nodes',headers={'Content-Type': 'application/json'})
self.failUnlessEqual(200, self.client.status)
def tearDown(self):
self.client.delete('/environments',headers={'Content-Type': 'application/json'})
from restclient import Resource
import random
import json
s = 'http://localhost:8089/'
u0 = str(random.randint(0, 10000))
u1 = str(random.randint(0, 10000))
print(u0, u1)
res = Resource(s)
print res.post('api/%s/' % u0)
print res.post('api/%s/msg' % u0, payload='Immmmm')
msg2_id = json.loads(res.post('api/%s/msg' % u0, payload='sailing'))['msg_id']
print res.post('api/%s/msg' % u0, payload='away')
print res.get('api/%s/' % u0)
print res.delete('api/%s/msg/%s/' % (u0, msg2_id))
print res.post('api/%s/' % u1)
r = res.get()
print(r)
print res.delete('api/%s/' % u0)
r = res.get()
print(r)
from restclient import Resource
import random
import json
s = 'http://localhost:8089/'
u0 = str(random.randint(0, 10000))
u1 = str(random.randint(0, 10000))
print (u0, u1)
res = Resource(s)
print res.post('api/%s/' % u0)
print res.post('api/%s/msg' % u0, payload='Immmmm')
msg2_id = json.loads(res.post('api/%s/msg' % u0, payload='sailing'))['msg_id']
print res.post('api/%s/msg' % u0, payload='away')
print res.get('api/%s/' % u0)
print res.delete('api/%s/msg/%s/' % (u0, msg2_id))
print res.post('api/%s/' % u1)
r = res.get()
print(r)
print res.delete('api/%s/' % u0)
r = res.get()
print(r)
