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)