def setUp(self):
self.model = ServerModel()
self.collection = self.model.mongo.get_collection('servers')
def setUp(self):
self.model = ServerModel()
self.collection = self.model.mongo.get_collection('servers')
class ServerModelTest(unittest.TestCase):
def setUp(self):
self.model = ServerModel()
self.collection = self.model.mongo.get_collection('servers')
def get_servers_for_group_test(self):
self.collection.remove()
self.collection.insert({"name": "test", "alert_group": 'test'})
result = self.model.get_servers_for_group('test')
eq_(result.count(), 1)
def check_server_exists_test(self):
self.collection.remove()
self.collection.insert({"name": "test"})
result = self.model.server_exists('test')
eq_(result, 1)
def update_server_test(self):
self.collection.remove()
self.collection.insert({"name": "test"})
server = self.collection.find_one()
self.model.update({
"name": "test_updated",
"default": 1
}, server['_id'])
result = self.collection.find_one()
eq_(result['name'], 'test_updated')
def add_server_test(self):
self.collection.remove()
self.model.add('test', 'note', '')
result = self.collection.find_one()
eq_(result['name'], 'test')
if result['key']:
assert True
def get_server_test(self):
self.collection.remove()
self.collection.insert({"name": "test"})
server = self.collection.find_one()
result = self.model.get_by_id(server['_id'])
eq_(result['name'], 'test')
eq_(result['_id'], server['_id'])
def get_server_by_key_test(self):
self.collection.remove()
self.collection.insert({"name": "test", "key": "test_me"})
server = self.collection.find_one()
result = self.model.get_server_by_key('test_me')
eq_(result['name'], 'test')
eq_(result['key'], 'test_me')
eq_(result['_id'], server['_id'])
def delete_server_test(self):
self.collection.remove()
self.collection.insert({"name": "test", "key": "test_me"})
server = self.collection.find_one()
self.model.delete(server['_id'])
result = self.collection.count()
eq_(result, 0)
def servers_dict_test(self):
self.collection.remove()
self.collection.insert({"name": "test", "key": "test_me"})
self.collection.insert({"name": "test2", "key": "test_me2"})
servers = self.collection.find()
result = self.model.get_all_dict()
for server in servers:
key = unicode(server['_id'])
if key in result.keys():
assert True
def get_all_servers_test(self):
self.collection.remove()
self.collection.insert({"name": "test", "key": "test_me"})
self.collection.insert({"name": "test1", "key": "test_me_again"})
result = self.model.get_all()
eq_(result.count(), 2)
self.collection.remove()
def get_filtered_test(self):
self.collection.remove()
self.collection.insert({"name": "test", "key": "test_me"})
self.collection.insert({"name": "test1", "key": "test_me_again"})
self.collection.insert({
"name": "test2",
"key": "test_me_one_more_time"
})
result = self.model.get_filtered(filter='all')
eq_(result.count(), 3)
# Get app ids
server_ids = []
for i in self.collection.find():
server_ids.append(str(i['_id']))
result = self.model.get_filtered(filter=server_ids)
eq_(result.count(), 3)
result = self.model.get_filtered(filter=server_ids[0:2])
eq_(result.count(), 2)
for r in result.clone():
assert str(r['_id']) in server_ids[0:2]
self.collection.remove()
class ServerModelTest(unittest.TestCase):
def setUp(self):
self.model = ServerModel()
self.collection = self.model.mongo.get_collection('servers')
def get_servers_for_group_test(self):
self.collection.remove()
self.collection.insert({"name" : "test", "alert_group": 'test'})
result = self.model.get_servers_for_group('test')
eq_(result.count(), 1)
def check_server_exists_test(self):
self.collection.remove()
self.collection.insert({"name" : "test"})
result = self.model.server_exists('test')
eq_(result, 1)
def update_server_test(self):
self.collection.remove()
self.collection.insert({"name" : "test"})
server = self.collection.find_one()
self.model.update({"name": "test_updated", "default": 1 }, server['_id'])
result = self.collection.find_one()
eq_(result['name'],'test_updated')
def add_server_test(self):
self.collection.remove()
self.model.add('test', 'note','')
result = self.collection.find_one()
eq_(result['name'],'test')
if result['key']:
assert True
def get_server_test(self):
self.collection.remove()
self.collection.insert({"name" : "test"})
server = self.collection.find_one()
result = self.model.get_by_id(server['_id'])
eq_(result['name'],'test')
eq_(result['_id'],server['_id'])
def get_server_by_key_test(self):
self.collection.remove()
self.collection.insert({"name" : "test", "key": "test_me"})
server = self.collection.find_one()
result = self.model.get_server_by_key('test_me')
eq_(result['name'],'test')
eq_(result['key'],'test_me')
eq_(result['_id'],server['_id'])
def delete_server_test(self):
self.collection.remove()
self.collection.insert({"name" : "test", "key": "test_me"})
server = self.collection.find_one()
self.model.delete(server['_id'])
result = self.collection.count()
eq_(result,0)
def servers_dict_test(self):
self.collection.remove()
self.collection.insert({"name" : "test", "key": "test_me"})
self.collection.insert({"name" : "test2", "key": "test_me2"})
servers = self.collection.find()
result = self.model.get_all_dict()
for server in servers:
key = unicode(server['_id'])
if key in result.keys():
assert True
def get_all_servers_test(self):
self.collection.remove()
self.collection.insert({"name" : "test", "key": "test_me"})
self.collection.insert({"name" : "test1", "key": "test_me_again"})
result = self.model.get_all()
eq_(result.count(), 2)
self.collection.remove()
def get_filtered_test(self):
self.collection.remove()
self.collection.insert({"name" : "test", "key": "test_me"})
self.collection.insert({"name" : "test1", "key": "test_me_again"})
self.collection.insert({"name" : "test2", "key": "test_me_one_more_time"})
result = self.model.get_filtered(filter='all')
eq_(result.count(), 3)
# Get app ids
server_ids = []
for i in self.collection.find():
server_ids.append(str(i['_id']))
result = self.model.get_filtered(filter=server_ids)
eq_(result.count(), 3)
result = self.model.get_filtered(filter=server_ids[0:2])
eq_(result.count(), 2)
for r in result.clone():
assert str(r['_id']) in server_ids[0:2]
self.collection.remove()
