diff --git a/pillar/test_attract.py b/pillar/test_attract.py deleted file mode 100644 index 4455de47..00000000 --- a/pillar/test_attract.py +++ /dev/null @@ -1,158 +0,0 @@ -import os -import json -import unittest -from pymongo import MongoClient -from bson import ObjectId -from datetime import datetime -import base64 - - -class AttractTestCase(unittest.TestCase): - - - def encodeUsrPass(self, user, password): - usrPass = "{0}:{1}".format(user, password) - b64Val = base64.b64encode(usrPass) - return b64Val - - def addUser(self, first_name, last_name, role): - return self.app.post('/users', data=dict( - first_name=first_name, - last_name=last_name, - role=role, - ), follow_redirects=True) - - def addNodeType(self, name, schema): - data = { - 'name': name, - 'dyn_schema': schema, - } - headers = { - 'content-type': 'application/json', - 'Authorization': 'Basic QU5MR05TSUVaSjoxMjM0' - } - return self.app.post( - '/node_types', - data=json.dumps(data), - headers=headers, - follow_redirects=True) - - def addNode(self, name, nodeType, properties): - data = { - 'name': name, - 'node_type': nodeType, - 'properties': properties - } - headers = { - 'content-type': 'application/json', - 'Authorization': 'Basic QU5MR05TSUVaSjoxMjM0', - } - return self.app.post( - '/nodes', - data=json.dumps(data), - headers=headers, - follow_redirects=True) - - def login(self, username, password): - headers = { - 'content-type': 'application/json', - 'Authorization': 'Basic {0}'.format( - self.encodeUsrPass(username, password)) - } - data = { - 'username': username, - } - return self.app.post( - '/tokens', - data=json.dumps(data), - headers=headers, - follow_redirects=True) - - def logout(self): - return self.app.get('/logout', follow_redirects=True) - - - # Tests - - def test_add_user(self): - rv = self.addUser('admin', 'default', 'author') - assert 201 == rv.status_code - - def test_add_node_type(self): - schema = { - 'frame_start': { - 'type':'integer', - } - } - rv = self.addNodeType('NodeName', schema) - assert 201 == rv.status_code - - def test_add_node(self): - properties = { - 'frame_start': 123 - } - rv = self.addNode('Shot01', '55016a52135d32466fc800be', properties) - assert 201 == rv.status_code - - def test_login(self): - rv = self.login('admin', 'secret') - #print (rv.data) - assert 201 == rv.status_code - - def test_empty_db(self): - rv = self.app.get('/') - assert 401 == rv.status_code - - # Test Setup - - def setUp(self): - # Setup DB - client = MongoClient() - db = client.attract_test - for col in db.collection_names(): - try: - db[col].remove({}) - except: - pass - - test_user = { - "_id": ObjectId("550171c8135d3248e477f288"), - "_updated": datetime.now(), - "firs_tname": "TestFirstname", - "last_name": "TestLastname", - "role": "author", - "_created": datetime.now(), - "_etag": "302236e27f51d2e26041ae9de49505d77332b260" - } - - test_node_type = { - "_id": ObjectId("55016a52135d32466fc800be"), - "_updated": datetime.now(), - "name": "NodeName", - "dyn_schema": {"frame_start": {"type": "integer"}}, - "_created": datetime.now(), - "_etag": "0ea3c4f684a0cda85525184d5606c4f4ce6ac5f5" - } - - test_token = { - "-id": ObjectId("5502f289135d3274cb658ba7"), - "username": "TestFirstname", - "token": "ANLGNSIEZJ", - "_etag": "1e96ed46b133b7ede5ce6ef0d6d4fc53edd9f2ba" - } - - db.users.insert(test_user) - db.node_types.insert(test_node_type) - db.tokens.insert(test_token) - - # Initialize Attract - os.environ['TEST_ATTRACT'] = '1' - import application - application.app.config['TESTING'] = True - self.app = application.app.test_client() - - def tearDown(self): - pass - -if __name__ == '__main__': - unittest.main()