def test_insert_into_db(self):
"""Test for app.models.LogEntry.save"""
logentry = LogEntry('test level', 'test owner', 'test data', 'test tags')
logentry_id = logentry.save()
configs = config()
db = MongoDBConnection.connect(configs)
collection = unicode(configs['mongo']['collection'])
logentry_from_db = db[collection].find_one({'_id': ObjectId(logentry_id)})
db[collection].remove({'_id': ObjectId(logentry_id)})
self.assertTrue(logentry_from_db['level'] == 'test level' and
logentry_from_db['owner'] == 'test owner' and
logentry_from_db['data'] == 'test data' and
logentry_from_db['tags'] == 'test tags',
msg="New entry does not match the expected or does not exist.")
def test_connection(self):
"""Test for app.connectors.MongoDBConnection.connect"""
configs = config()
db = MongoDBConnection.connect(configs)
new_collection = 'testcollection'
try:
db.create_collection(new_collection)
except:
self.assertTrue(False, msg="Can't create collection.")
try:
db[new_collection].insert({'message': "Test message"})
except:
self.assertTrue(False, msg="Can't insert records to collection.")
if not db[new_collection].find_one({'message': "Test message"}):
self.assertTrue(False, msg="Can't find inserted record.")
try:
db.drop_collection(new_collection)
except:
self.assertTrue(False, msg="Can't drop collection.")
def test_add_logentry(self):
"""Test for app.views.add_logentry"""
with app.test_client() as logentry_view:
data = {'level': 'info',
'owner': 'test owner',
'data': 'test data',
'tags': ['tag1', 'tag2']}
json_data = json.dumps(data)
response = logentry_view.post('/api/entry/', content_type='application/json', data=json_data)
resp = json.loads(response.data)
configs = config()
db = MongoDBConnection.connect(configs)
collection = unicode(configs['mongo']['collection'])
entry_from_db = db[collection].find_one({'_id': ObjectId(resp['id'])})
db[collection].remove({'_id': ObjectId(resp['id'])})
self.assertTrue('OK' in resp and
resp['OK'] and
entry_from_db['level'] == 'info' and
entry_from_db['owner'] == 'test owner' and
entry_from_db['data'] == 'test data' and
entry_from_db['tags'] == ['tag1', 'tag2'],
msg="Server response is not the same as entry in DB.")
def tearDown(self):
configs = config()
db = MongoDBConnection.connect(configs)
collection = unicode(configs['mongo']['collection'])
for logentry_id in self._drop:
db[collection].remove({'_id': ObjectId(logentry_id)})