3b568b70a0
The Synergy's methods startManager() stopManager() getManagerStatus() and executeCommand() require the parsing of the parameters included into the user request and make some checks for the mandatory parameters. This commit enhances this parsing and checking steps by using the Python decorators. Moreover it simplifies the execution of the methods startManager() stopManager() and getManagerStatus() by handling requests for only one manager at a time. Change-Id: Ic295ad20f03459cf3e512e795529c35bd6969ed3 Sem-Ver: bugfix
156 lines
4.8 KiB
Python
156 lines
4.8 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest
|
|
|
|
import json
|
|
import logging
|
|
import mock
|
|
import sys
|
|
import time
|
|
|
|
from mock import Mock
|
|
from synergy.common.utils import objectHookHandler
|
|
from synergy.service import Synergy
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
|
|
LOG = logging.getLogger("SynergyTests")
|
|
LOG.setLevel(logging.DEBUG)
|
|
|
|
|
|
def getLogger(name):
|
|
formatter = logging.Formatter("%(asctime)s - %(name)s - "
|
|
"%(levelname)s - %(message)s")
|
|
|
|
ch = logging.StreamHandler(sys.stdout)
|
|
ch.setLevel(logging.DEBUG)
|
|
ch.setFormatter(formatter)
|
|
|
|
logger = logging.getLogger(name)
|
|
logger.setLevel(logging.DEBUG)
|
|
logger.propagate = False
|
|
logger.addHandler(ch)
|
|
|
|
return logger
|
|
|
|
|
|
class SynergyTests(unittest.TestCase):
|
|
|
|
@mock.patch('synergy.service.LOG', LOG)
|
|
def setUp(self):
|
|
super(SynergyTests, self).setUp()
|
|
|
|
self.synergy = Synergy()
|
|
self.synergy.managers["TimerManager"].start()
|
|
time.sleep(1)
|
|
|
|
@mock.patch('synergy.service.LOG', LOG)
|
|
def test_managers(self):
|
|
self.assertEqual(self.synergy.managers.keys(), ["TimerManager"])
|
|
|
|
@mock.patch('synergy.service.LOG', LOG)
|
|
def test_listManagers(self):
|
|
start_response = Mock()
|
|
result = self.synergy.listManagers(environ={},
|
|
start_response=start_response)
|
|
|
|
result = json.loads(result[0], object_hook=objectHookHandler)
|
|
self.assertEqual(len(result), 1)
|
|
self.assertEqual(result[0].getName(), "TimerManager")
|
|
|
|
@mock.patch('synergy.service.LOG', LOG)
|
|
def test_getManagerStatus(self):
|
|
start_response = Mock()
|
|
environ = {}
|
|
result = self.synergy.getManagerStatus(environ,
|
|
start_response=start_response)
|
|
|
|
result = json.loads(result[0], object_hook=objectHookHandler)
|
|
|
|
self.assertEqual(result[0].getStatus(), 'ACTIVE')
|
|
|
|
environ = {'QUERY_STRING': 'manager=TimerManager'}
|
|
|
|
result = self.synergy.getManagerStatus(environ,
|
|
start_response=start_response)
|
|
|
|
result = json.loads(result[0], object_hook=objectHookHandler)
|
|
|
|
self.assertEqual(result[0].getStatus(), 'ACTIVE')
|
|
|
|
@mock.patch('synergy.service.LOG', LOG)
|
|
def test_startManager(self):
|
|
start_response = Mock()
|
|
environ = {'QUERY_STRING': 'manager=NONE'}
|
|
|
|
result = self.synergy.startManager(environ, start_response)
|
|
|
|
self.assertEqual(result, "manager NONE not found!")
|
|
|
|
environ = {'QUERY_STRING': 'manager=TimerManager'}
|
|
|
|
result = self.synergy.startManager(environ, start_response)
|
|
result = json.loads(result, object_hook=objectHookHandler)
|
|
|
|
self.assertEqual(result.getStatus(), 'RUNNING')
|
|
self.assertEqual(result.get("message"), 'started successfully')
|
|
|
|
time.sleep(0.5)
|
|
|
|
result = self.synergy.startManager(environ, start_response)
|
|
result = json.loads(result, object_hook=objectHookHandler)
|
|
|
|
self.assertEqual(result.getStatus(), 'RUNNING')
|
|
self.assertEqual(result.get("message"), 'WARN: already started')
|
|
|
|
@mock.patch('synergy.service.LOG', LOG)
|
|
def test_stopManager(self):
|
|
stop_response = Mock()
|
|
environ = {'QUERY_STRING': 'manager=NONE'}
|
|
|
|
result = self.synergy.startManager(environ, stop_response)
|
|
|
|
self.assertEqual(result, "manager NONE not found!")
|
|
|
|
environ = {'QUERY_STRING': 'manager=TimerManager'}
|
|
|
|
result = self.synergy.startManager(environ, stop_response)
|
|
result = json.loads(result, object_hook=objectHookHandler)
|
|
|
|
time.sleep(0.5)
|
|
|
|
result = self.synergy.stopManager(environ, stop_response)
|
|
result = json.loads(result, object_hook=objectHookHandler)
|
|
|
|
self.assertEqual(result.getStatus(), 'ACTIVE')
|
|
|
|
@mock.patch('synergy.service.LOG', LOG)
|
|
def test_executeCommand(self):
|
|
environ = {'QUERY_STRING':
|
|
'manager=TimerManager&args=%7B%22id%22%3A'
|
|
'+null%2C+%22name%22%3A+%22prj_a%22%7D&command=GET_TIME'}
|
|
start_response = Mock()
|
|
|
|
result = self.synergy.executeCommand(environ, start_response)
|
|
result = json.loads(result[0])
|
|
|
|
self.assertEqual(result.keys(), ["localtime"])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|