def test_get_owners(self): """Test for app.models.LogEntry.get_entries""" logentry = LogEntry('test level', 'test get_owners owner', 'test data', 'test tags') logentry_id = logentry.save() self._drop.append(logentry_id) result = logentry.get_owners() self.assertIn('test get_owners owner', result)
def test_count_api_page(self): """Test for app.views.list""" logentry = LogEntry('test level', 'test count owner', 'test data', ['tag1', 'tag2']) logentry_id = logentry.save() logentry1 = LogEntry('test level1', 'test count owner', 'test data1', ['tag1', 'tag2']) logentry1_id = logentry1.save() self._drop.append(logentry_id) self._drop.append(logentry1_id) with app.test_client() as test_app: find = dict(owner='test count owner') response = test_app.post('/api/count/', content_type='application/json', data=json.dumps(dict(find=find))) self.assertEqual(response.status_code, 200) result = json.loads(response.get_data()) ok = result.get('OK') self.assertTrue(ok) data = result['result'] self.assertEqual(data, 2)
def test_owners_api_page(self): """Test for app.views.list""" logentry = LogEntry('test level', 'test owners owner', 'test data', 'test tags') logentry_id = logentry.save() logentry1 = LogEntry('test level1', 'test owners owner1', 'test data1', 'test tags1') logentry1_id = logentry1.save() self._drop.append(logentry_id) self._drop.append(logentry1_id) with app.test_client() as test_app: response = test_app.get('/api/owners/', content_type='application/json') self.assertEqual(response.status_code, 200) result = json.loads(response.get_data()) ok = result.get('OK') self.assertTrue(ok) data = result['result'] self.assertGreaterEqual(len(data), 2) self.assertIn('test owners owner', data) self.assertIn('test owners owner1', data)
def test_count_from_db(self): """Test for app.models.LogEntry.count""" logentry = LogEntry('test level', 'test owner', 'test count data', 'test tags') logentry_id = logentry.save() self._drop.append(logentry_id) result = logentry.count(dict(data='test count data')) self.assertEqual(result, 1) result = logentry.count(dict(data='test count data1')) self.assertEqual(result, 0)
def test_list_api_page(self): """Test for app.views.list""" logentry = LogEntry('test level', 'test list owner', 'test data', 'test tags') logentry_id = logentry.save() logentry1 = LogEntry('test level1', 'test list owner', 'test data1', 'test tags1') logentry1_id = logentry1.save() self._drop.append(logentry_id) self._drop.append(logentry1_id) with app.test_client() as test_app: find = dict(owner='test list owner') response = test_app.post('/api/list/', content_type='application/json', data=json.dumps(dict(find=find))) self.assertEqual(response.status_code, 200) result = json.loads(response.get_data()) ok = result.get('OK') self.assertTrue(ok) data = result['result'] self.assertEqual(len(data), 2) # By default sorting by timestamp DESCENDING self.assertEqual(data[0]['data'], 'test data1') self.assertEqual(data[1]['data'], 'test data')
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_get_from_db(self): """Test for app.models.LogEntry.get_entries""" logentry = LogEntry('test level', 'test get_entries owner', 'test data', 'test tags') logentry_id = logentry.save() self._drop.append(logentry_id) cursor = logentry.get_entries(dict(owner='test get_entries owner')) result = list(cursor) logentry_from_db = result[0] self.assertEqual(len(result), 1) self.assertEqual(result[0]['_id'], logentry_id) self.assertTrue(logentry_from_db['level'] == 'test level' and logentry_from_db['owner'] == 'test get_entries 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.")