class AuthTests(TestCase):
fixtures = ["test_fixture.json"]
def setUp(self):
self.client = MyTestClient()
def test_login(self):
# Should not be logged in
#print ( ' test not log in ? ')
r = self.client.get(newt_base_url + "/auth")
#print ('get ok ?' ,r.status_code )
self.assertEquals(r.status_code, 200)
json_response = r.json()
#print ( 'json' , json_response )
self.assertEquals(json_response['output']['auth'], False)
# Should be logged in
r = self.client.post(newt_base_url + "/auth", data=login)
self.assertEquals(r.status_code, 200)
json_response = r.json()
#print('cookies 1 : ' , self.client.cookies )
self.assertEquals(json_response['output']['auth'], True)
self.assertEquals(json_response['output']['username'],
login['username'])
cookies = self.client.cookies
# Loggen in self.client should return user info
#print( 'session id :' , json_response["output"]["newt_sessionid"])
#self.client.session['newt_sessionid'] = json_response["output"]["newt_sessionid"]
r = self.client.get(newt_base_url + "/auth")
self.assertEquals(r.status_code, 200)
json_response = r.json()
#print( 'cookies 2 : ' , self.client.cookies )
self.assertEquals(json_response['output']['auth'], True)
self.assertEquals(json_response['output']['username'],
login['username'])
def test_logout(self):
# Should be logged in
r = self.client.post(newt_base_url + "/auth", data=login)
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], True)
self.assertEquals(json_response['output']['username'],
login['username'])
r = self.client.delete(newt_base_url + "/auth")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], False)
r = self.client.get(newt_base_url + "/auth")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], False)
class AuthTests(TestCase):
fixtures = ["test_fixture.json"]
def setUp(self):
self.client = MyTestClient()
def test_login(self):
# Should not be logged in
r = self.client.get(newt_base_url + "/auth")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], False)
# Should be logged in
r = self.client.post(newt_base_url + "/auth", data=login)
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], True)
self.assertEquals(json_response['output']['username'],
login['username'])
# Loggen in self.client should return user info
r = self.client.get(newt_base_url + "/auth")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], True)
self.assertEquals(json_response['output']['username'],
login['username'])
def test_logout(self):
# Should be logged in
r = self.client.post(newt_base_url + "/auth", data=login)
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], True)
self.assertEquals(json_response['output']['username'],
login['username'])
r = self.client.delete(newt_base_url + "/auth")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], False)
r = self.client.get(newt_base_url + "/auth")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], False)
class AuthTests(TestCase):
fixtures = ["test_fixture.json"]
def setUp(self):
self.client = MyTestClient()
def test_login(self):
# Should not be logged in
r = self.client.get(newt_base_url + "/auth")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], False)
# Should be logged in
r = self.client.post(newt_base_url + "/auth", data=login)
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], True)
self.assertEquals(json_response['output']['username'], login['username'])
# Loggen in self.client should return user info
r = self.client.get(newt_base_url + "/auth")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], True)
self.assertEquals(json_response['output']['username'], login['username'])
def test_logout(self):
# Should be logged in
r = self.client.post(newt_base_url + "/auth", data=login)
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], True)
self.assertEquals(json_response['output']['username'], login['username'])
r = self.client.delete(newt_base_url + "/auth")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], False)
r = self.client.get(newt_base_url + "/auth")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['auth'], False)
class JobTests(TestCase):
fixtures = ["test_fixture.json"]
def setUp(self):
self.client = MyTestClient()
self.client.post(newt_base_url + "/auth", data=login)
def test_get_queues(self):
# Tests getting queues
r = self.client.get(newt_base_url + "/job/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertTrue(len(json_response['output']) > 0)
self.assertIn(machine, json_response['output'].keys())
def test_running_cmds(self):
# Tests submitting a job
payload = {
"jobscript": "/bin/hostname\nsleep 10"
}
r = self.client.post(newt_base_url + "/job/"+machine+"/", data=payload)
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertIsNot(json_response['output']['jobid'], None)
# Get job id from submitting the job
job_id = json_response['output']['jobid']
# Give the process time to register
time.sleep(1)
# Tests getting job info
r = self.client.get(newt_base_url + "/job/"+machine+"/%s/" % job_id)
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['jobid'], job_id)
self.assertEquals(json_response['output']['user'], login['username'])
# Delete job from queue
r = self.client.delete(newt_base_url + "/job/"+machine+"/%s/" % job_id)
self.assertEquals(r.status_code, 200)
class JobTests(TestCase):
fixtures = ["test_fixture.json"]
def setUp(self):
self.client = MyTestClient()
self.client.post(newt_base_url + "/auth", data=login)
def test_get_queues(self):
# Tests getting queues
r = self.client.get(newt_base_url + "/job/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertTrue(len(json_response['output']) > 0)
self.assertIn(machine, json_response['output'].keys())
def test_running_cmds(self):
# Tests submitting a job
payload = {"jobscript": "/bin/hostname\nsleep 10"}
r = self.client.post(newt_base_url + "/job/" + machine + "/",
data=payload)
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertIsNot(json_response['output']['jobid'], None)
# Get job id from submitting the job
job_id = json_response['output']['jobid']
# Give the process time to register
time.sleep(1)
# Tests getting job info
r = self.client.get(newt_base_url + "/job/" + machine +
"/%s/" % job_id)
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['jobid'], job_id)
self.assertEquals(json_response['output']['user'], login['username'])
# Delete job from queue
r = self.client.delete(newt_base_url + "/job/" + machine +
"/%s/" % job_id)
self.assertEquals(r.status_code, 200)
class StoreTests(TestCase):
fixtures = ["test_fixture.json"]
def setUp(self):
self.client = MyTestClient()
# Hacky: Need to figure out a way around this...
try:
from pymongo import MongoClient
db = MongoClient()['store']
db.test_store_1.drop()
db.permissions.remove({"name": "test_store_1"})
except Exception:
pass
try:
import redis
storedb = redis.Redis(host="localhost", db=0)
storedb.flushdb()
except Exception:
pass
def test_store_basic(self):
self.client.post(newt_base_url + "/auth", data=login)
r = self.client.get(newt_base_url + "/store")
self.assertEquals(r.status_code, 200)
def test_store_manipulation(self):
self.client.post(newt_base_url + "/auth", data=login)
# Creates a new store (create_store)
r = self.client.post(newt_base_url + "/store")
self.assertEquals(r.status_code, 200)
json_response = r.json()
# Ensures that no data was added to the store
self.assertEquals(json_response['output']['oid'], [])
store_id = json_response['output']['id']
# Ensures that new store is empty (get_store_contents)
r = self.client.get(newt_base_url + "/store/" + store_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], [])
# Tests insertion (store_insert)
payload = {"data": json.dumps({"foo": "bar"})}
r = self.client.post(newt_base_url + "/store/" + store_id + "/",
data=payload)
self.assertEquals(r.status_code, 200)
json_response = r.json()
obj_id = json_response['output']
# Checks insertion by checking all of the store's objects (get_store_contents)
r = self.client.get(newt_base_url + "/store/" + store_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'][0]['data'], payload['data'])
self.assertEquals(json_response['output'][0]['oid'], obj_id)
# Checks insertion by checking the individual object (store_get_obj)
r = self.client.get(newt_base_url + "/store/" + store_id + "/" +
obj_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], payload['data'])
# Tests update (store_update)
updated_payload = {"data": json.dumps({"foo": "baz"})}
r = self.client.put(newt_base_url + "/store/" + store_id + "/" +
obj_id + "/",
data=json.dumps(updated_payload),
content_type="application/json")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], obj_id)
# Checks updated data
r = self.client.get(newt_base_url + "/store/" + store_id + "/" +
obj_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], updated_payload['data'])
# Tests delete
r = self.client.delete(newt_base_url + "/store/" + store_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEqual(json_response['output'], store_id)
# Ensures that getting the deleted store will error
r = self.client.get(newt_base_url + "/store/" + store_id + "/")
self.assertEquals(r.status_code, 404)
def test_store_creation_with_initial(self):
self.client.post(newt_base_url + "/auth", data=login)
payload = {"data": json.dumps({"x": 5})}
# Without an initial name
r = self.client.post(newt_base_url + "/store/", data=payload)
self.assertEquals(r.status_code, 200)
json_response = r.json()
store_id = json_response['output']['id']
self.assertEquals(len(json_response['output']['oid']), 1)
obj_id = json_response['output']['oid'][0]
# Checks the data
r = self.client.get(newt_base_url + "/store/" + store_id + "/" +
obj_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], payload['data'])
# With an initial name
r = self.client.post(newt_base_url + "/store/teststore1/",
data=payload)
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['id'], "teststore1")
self.assertEquals(len(json_response['output']['oid']), 1)
obj_id = json_response['output']['oid'][0]
# Checks the data
r = self.client.get(newt_base_url + "/store/teststore1/" + obj_id +
"/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], payload['data'])
# Deletes the store
r = self.client.delete(newt_base_url + "/store/teststore1/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEqual(json_response['output'], "teststore1")
r = self.client.delete(newt_base_url + "/store/" + store_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEqual(json_response['output'], store_id)
def test_store_perms(self):
self.client.post(newt_base_url + "/auth", data=login)
# Create a store
r = self.client.post(newt_base_url + "/store/test_store_1/")
self.assertEquals(r.status_code, 200)
r = self.client.get(newt_base_url + "/store/test_store_1/perms/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
# Checks that the creator has appropriate permissions
self.assertEquals(json_response['output']['name'], "test_store_1")
self.assertEquals(json_response['output']['perms'][0]['name'],
login['username'])
self.assertIn("r", json_response['output']['perms'][0]['perms'])
self.assertIn("w", json_response['output']['perms'][0]['perms'])
payload = {
"data":
json.dumps([{
"name": login['username'],
"perms": ["r", "w", "x"]
}])
}
r = self.client.post(newt_base_url + "/store/test_store_1/perms/",
data=payload)
self.assertEqual(r.status_code, 200)
r = self.client.get(newt_base_url + "/store/test_store_1/perms/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['name'], "test_store_1")
self.assertEquals(json_response['output']['perms'][0]['name'],
login['username'])
self.assertIn("r", json_response['output']['perms'][0]['perms'])
self.assertIn("w", json_response['output']['perms'][0]['perms'])
self.assertIn("x", json_response['output']['perms'][0]['perms'])
# Deletes the store
r = self.client.delete(newt_base_url + "/store/test_store_1/")
self.assertEquals(r.status_code, 200)
r = self.client.get(newt_base_url + "/store/test_store_1/perms/")
self.assertEquals(r.status_code, 404)
class StoresTests(TestCase):
fixtures = ["test_fixture.json"]
def setUp(self):
self.client = MyTestClient()
# Hacky: Need to figure out a way around this...
try:
from pymongo import MongoClient
db = MongoClient()['stores']
db.test_store_1.drop()
db.permissions.remove({"name":"test_store_1"})
except Exception:
pass
try:
import redis
storedb = redis.Redis(host="localhost", db=0)
storedb.flushdb()
except Exception:
pass
def test_stores_basic(self):
r = self.client.get(newt_base_url + "/stores")
self.assertEquals(r.status_code, 200)
def test_stores_manipulation(self):
# Creates a new store (create_store)
r = self.client.post(newt_base_url + "/stores")
self.assertEquals(r.status_code, 200)
json_response = r.json()
# Ensures that no data was added to the store
self.assertEquals(json_response['output']['oid'], [])
store_id = json_response['output']['id']
# Ensures that new store is empty (get_store_contents)
r = self.client.get(newt_base_url + "/stores/" + store_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], [])
# Tests insertion (store_insert)
payload = {"data": json.dumps({"foo":"bar"})}
r = self.client.post(newt_base_url + "/stores/" + store_id + "/",
data=payload)
self.assertEquals(r.status_code, 200)
json_response = r.json()
obj_id = json_response['output']
# Checks insertion by checking all of the store's objects (get_store_contents)
r = self.client.get(newt_base_url + "/stores/" + store_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'][0]['data'], payload['data'])
self.assertEquals(json_response['output'][0]['oid'], obj_id)
# Checks insertion by checking the individual object (store_get_obj)
r = self.client.get(newt_base_url + "/stores/" + store_id + "/" + obj_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], payload['data'])
# Tests update (store_update)
updated_payload = {"data": json.dumps({"foo": "baz"})}
r = self.client.put(newt_base_url + "/stores/" + store_id + "/" + obj_id + "/",
data=json.dumps(updated_payload), content_type="application/json")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], obj_id)
# Checks updated data
r = self.client.get(newt_base_url + "/stores/" + store_id + "/" + obj_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], updated_payload['data'])
# Tests delete
r = self.client.delete(newt_base_url + "/stores/" + store_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEqual(json_response['output'], store_id)
# Ensures that getting the deleted store will error
r = self.client.get(newt_base_url + "/stores/" + store_id + "/")
self.assertEquals(r.status_code, 404)
def test_stores_creation_with_initial(self):
payload = {"data": json.dumps({"x":5})}
# Without an initial name
r = self.client.post(newt_base_url + "/stores/", data=payload)
self.assertEquals(r.status_code, 200)
json_response = r.json()
store_id = json_response['output']['id']
self.assertEquals(len(json_response['output']['oid']), 1)
obj_id = json_response['output']['oid'][0]
# Checks the data
r = self.client.get(newt_base_url + "/stores/" + store_id + "/" + obj_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], payload['data'])
# With an initial name
r = self.client.post(newt_base_url + "/stores/teststore1/", data=payload)
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['id'], "teststore1")
self.assertEquals(len(json_response['output']['oid']), 1)
obj_id = json_response['output']['oid'][0]
# Checks the data
r = self.client.get(newt_base_url + "/stores/teststore1/" + obj_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output'], payload['data'])
# Deletes the stores
r = self.client.delete(newt_base_url + "/stores/teststore1/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEqual(json_response['output'], "teststore1")
r = self.client.delete(newt_base_url + "/stores/" + store_id + "/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEqual(json_response['output'], store_id)
def test_store_perms(self):
self.client.post(newt_base_url + "/auth", data=login)
# Create a store
r = self.client.post(newt_base_url + "/stores/test_store_1/")
self.assertEquals(r.status_code, 200)
r = self.client.get(newt_base_url + "/stores/test_store_1/perms/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
# Checks that the creator has appropriate permissions
self.assertEquals(json_response['output']['name'], "test_store_1")
self.assertEquals(json_response['output']['perms'][0]['name'], login['username'])
self.assertIn("r", json_response['output']['perms'][0]['perms'])
self.assertIn("w", json_response['output']['perms'][0]['perms'])
payload = {"data": json.dumps([{"name": login['username'], "perms": ["r", "w", "x"]}])}
r = self.client.post(newt_base_url + "/stores/test_store_1/perms/", data=payload)
self.assertEqual(r.status_code, 200)
r = self.client.get(newt_base_url + "/stores/test_store_1/perms/")
self.assertEquals(r.status_code, 200)
json_response = r.json()
self.assertEquals(json_response['output']['name'], "test_store_1")
self.assertEquals(json_response['output']['perms'][0]['name'], login['username'])
self.assertIn("r", json_response['output']['perms'][0]['perms'])
self.assertIn("w", json_response['output']['perms'][0]['perms'])
self.assertIn("x", json_response['output']['perms'][0]['perms'])
# Deletes the stores
r = self.client.delete(newt_base_url + "/stores/test_store_1/")
self.assertEquals(r.status_code, 200)
r = self.client.get(newt_base_url + "/stores/test_store_1/perms/")
self.assertEquals(r.status_code, 404)
