def query_user(self):
"""
Launch request on "user" endpoint. Only for current App user.
"""
request_model = User.get_request_model(self.backend.token)
request = self.get(
request_model['endpoint'],
request_model['params'],
request_model['projection']
)
if request:
if request['_items']:
user = User()
user.create(
request['_items'][0]['_id'],
request['_items'][0],
request['_items'][0]['name']
)
data_manager.update_database('user', user)
if 'OK' in request['_status']:
data_manager.db_is_ready[request_model['endpoint']] = True
def test_get_request_user_model(self):
"""Get User Request Model"""
under_test = User.get_request_model('')
self.assertTrue('endpoint' in under_test)
self.assertEqual('user', under_test['endpoint'])
self.assertTrue('params' in under_test)
self.assertTrue('projection' in under_test)
def test_get_request_user_model(self):
"""Get User Request Model"""
under_test = User.get_request_model('')
self.assertTrue('endpoint' in under_test)
self.assertEqual('user', under_test['endpoint'])
self.assertTrue('params' in under_test)
self.assertTrue('projection' in under_test)
def query_user(self):
"""
Launch request on "user" endpoint. Only for current App user.
"""
request_model = User.get_request_model(self.backend.token)
request = self.get(request_model['endpoint'], request_model['params'],
request_model['projection'])
if request:
if request['_items']:
user = User()
user.create(request['_items'][0]['_id'], request['_items'][0],
request['_items'][0]['name'])
data_manager.update_database('user', user)
if 'OK' in request['_status']:
data_manager.db_is_ready[request_model['endpoint']] = True
class TestAlignakQWidget(unittest2.TestCase):
"""
This file test methods of AlignakQWidget class object
"""
settings.init_config()
init_localization()
app_backend.login()
# User data test
user = User()
user_keys = User.get_request_model('test')['projection']
user_data_test = {}
for key in user_keys:
if 'host_notifications_enabled' in key or 'service_notifications_enabled' in key:
user_data_test[key] = True
else:
user_data_test[key] = 'test'
user.create('_id', user_data_test, 'admin')
@classmethod
def setUpClass(cls):
"""Create QApplication"""
try:
cls.app = QApplication(sys.argv)
except:
pass
def test_initialize_alignak_qwidget(self):
"""Initialize AlignakQWidget"""
under_test = AlignakQWidget()
self.assertIsNotNone(under_test.backend_connected)
self.assertIsNotNone(under_test.status_btn)
self.assertIsNotNone(under_test.status_dialog)
self.assertIsNotNone(under_test.profile_btn)
self.assertIsNotNone(under_test.profile_widget)
self.assertFalse(under_test.refresh_timer.isActive())
self.assertIsInstance(under_test, QWidget)
data_manager.update_database('user', self.user)
under_test.initialize()
self.assertIsNotNone(under_test.backend_connected)
self.assertIsNotNone(under_test.status_btn)
self.assertFalse(under_test.status_btn.isEnabled())
self.assertIsNotNone(under_test.status_dialog)
self.assertIsNotNone(under_test.profile_btn)
self.assertIsNotNone(under_test.profile_widget)
self.assertTrue(under_test.refresh_timer.isActive())
def test_update_status(self):
"""Update Alignak QWidget"""
under_test = AlignakQWidget()
under_test.initialize()
self.assertFalse(under_test.status_btn.isEnabled())
daemon_test = Daemon()
daemon_test.create(
'_id1',
{
'alive': True,
'address': '127.0.0.1',
'port': '8888',
'reachable': True,
'spare': True,
'passive': True,
'last_check': 123456789,
},
'arbiter'
)
data_manager.update_database('alignakdaemon', [daemon_test])
under_test.update_status()
# Status button is enabled when alignakdaemon is filled
self.assertTrue(under_test.status_btn.isEnabled())