pillar/attract/test_attract.py

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, firstname, lastname, role):
        return self.app.post('/users', data=dict(
            firstname=firstname,
            lastname=lastname,
            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(),
            "firstname": "TestFirstname",
            "lastname": "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()