def test_get_dao(self):
dao = get_dao('production')
self.assertEqual(dao.username, p['username'])
self.assertEqual(dao.password, p['password'])
self.assertEqual(dao.host, p['host'])
self.assertEqual(dao.port, p['port'])
self.assertEqual(dao.db, p['db_name'])
dao = get_dao('test')
self.assertEqual(dao.username, t['username'])
self.assertEqual(dao.password, t['password'])
self.assertEqual(dao.host, t['host'])
self.assertEqual(dao.port, t['port'])
self.assertEqual(dao.db, t['db_name'])
def test_get_by_session_and_id(self):
dao = get_dao('test')
dao.create('elevators', elevator_1)
response = self.tester.get('/elevators/test/alpha/A/', content_type='application/json')
self.assertEquals(response.status_code, 200)
data = json.loads(response.data)
self.assertEquals(len(data), 1)
def test_get_by_session_and_id(self):
dao = get_dao('test')
dao.create('elevators', elevator_1)
response = self.tester.get('/elevators/test/alpha/A/',
content_type='application/json')
self.assertEquals(response.status_code, 200)
data = json.loads(response.data)
self.assertEquals(len(data), 1)
def test_get_by_session_and_id(self):
dao = get_dao('test')
dao.create('elevators', elevator_1)
dao.create('elevators', elevator_2)
items = dao.get_by_session_and_id('elevators', 'alpha', 'A')
self.assertEqual(len(items), 1)
self.assertEqual(items[0]['session'], 'alpha')
self.assertEqual(items[0]['time'], 1)
def get(environment):
"""
All the available elevators documents
:param environment: 'test' or 'production'
:return: All the available elevators documents
"""
dao = get_dao(environment)
out = json.dumps(dao.get('elevators'), sort_keys=True, indent=4, default=json_util.default)
return Response(out, content_type='application/json; charset=utf-8')
def test_create(self):
try:
payload = json.dumps(elevator_1)
response = self.tester.post('/elevators/test/', data=payload, content_type='application/json')
self.assertEquals(response.status_code, 200)
dao = get_dao('test')
items = dao.get('elevators')
self.assertEqual(len(items), 1)
except Exception, e:
print e
def create(environment):
"""
Add a new document to the DB
:param environment: 'test' or 'production'
:return: The result of the operation
"""
item = json.loads(request.data)
dao = get_dao(environment)
ack = dao.create('elevators', item)
out = json.dumps(ack, sort_keys=True, indent=4, default=json_util.default)
return Response(out, content_type='application/json; charset=utf-8')
def test_create_item(self):
try:
dao = get_dao('test')
inserted_id = dao.create('elevators', elevator_1)
self.assertIsNotNone(inserted_id)
self.assertEqual(0, inserted_id.matched_count)
inserted_id = dao.create('elevators', elevator_1)
self.assertEqual(1, inserted_id.matched_count)
self.assertIsNotNone(inserted_id)
except:
pass
def create(environment):
"""
Add a new document to the DB
:param environment: 'test' or 'production'
:return: The result of the operation
"""
item = json.loads(request.data)
dao = get_dao(environment)
ack = dao.create('elevators', item)
out = json.dumps(ack, sort_keys=True, indent=4, default=json_util.default)
return Response(out, content_type='application/json; charset=utf-8')
def get_by_session_and_id(environment, session, id):
"""
Fetch all the documents matching the given session ID and elevator's ID
:param environment: 'test' or 'production'
:param session: Session ID
:param id: Elevator ID
:return: All the matching documents
"""
dao = get_dao(environment)
out = json.dumps(dao.get_by_session_and_id('elevators', session, id), sort_keys=True, indent=4, default=json_util.default)
return Response(out, content_type='application/json; charset=utf-8')
def test_create(self):
try:
payload = json.dumps(elevator_1)
response = self.tester.post('/elevators/test/',
data=payload,
content_type='application/json')
self.assertEquals(response.status_code, 200)
dao = get_dao('test')
items = dao.get('elevators')
self.assertEqual(len(items), 1)
except Exception, e:
print e
def get(environment):
"""
All the available elevators documents
:param environment: 'test' or 'production'
:return: All the available elevators documents
"""
dao = get_dao(environment)
out = json.dumps(dao.get('elevators'),
sort_keys=True,
indent=4,
default=json_util.default)
return Response(out, content_type='application/json; charset=utf-8')
def get_by_session_and_id(environment, session, id):
"""
Fetch all the documents matching the given session ID and elevator's ID
:param environment: 'test' or 'production'
:param session: Session ID
:param id: Elevator ID
:return: All the matching documents
"""
dao = get_dao(environment)
out = json.dumps(dao.get_by_session_and_id('elevators', session, id),
sort_keys=True,
indent=4,
default=json_util.default)
return Response(out, content_type='application/json; charset=utf-8')
def test_get(self):
dao = get_dao('test')
dao.create('elevators', elevator_1)
dao.create('elevators', elevator_2)
items = dao.get('elevators')
self.assertEqual(len(items), 2)
def test_create_connection_uri(self):
dao = get_dao('test')
uri = dao.create_connection_uri()
self.assertEqual(uri, 'mongodb://*****:*****@ds023468.mlab.com:23468/elevatorsdatatest')
