forked from platypush/platypush
Improvements on HTTP integration test. Plus, removed ApplicationStoppedEvent - it's unreliable and it could be delivered when the bus has already been stopped
This commit is contained in:
parent
2800bac3fb
commit
571a8ca9d1
5 changed files with 28 additions and 50 deletions
|
@ -17,7 +17,7 @@ from .cron.scheduler import CronScheduler
|
||||||
from .event.processor import EventProcessor
|
from .event.processor import EventProcessor
|
||||||
from .logger import Logger
|
from .logger import Logger
|
||||||
from .message.event import Event
|
from .message.event import Event
|
||||||
from .message.event.application import ApplicationStartedEvent, ApplicationStoppedEvent
|
from .message.event.application import ApplicationStartedEvent
|
||||||
from .message.request import Request
|
from .message.request import Request
|
||||||
from .message.response import Response
|
from .message.response import Response
|
||||||
from .utils import set_thread_name
|
from .utils import set_thread_name
|
||||||
|
@ -75,6 +75,8 @@ class Daemon:
|
||||||
Config.init(self.config_file)
|
Config.init(self.config_file)
|
||||||
logging.basicConfig(**Config.get('logging'))
|
logging.basicConfig(**Config.get('logging'))
|
||||||
|
|
||||||
|
redis_conf = Config.get('backend.redis') or {}
|
||||||
|
self.bus = RedisBus(on_message=self.on_message(), **redis_conf.get('redis_args', {}))
|
||||||
self.no_capture_stdout = no_capture_stdout
|
self.no_capture_stdout = no_capture_stdout
|
||||||
self.no_capture_stderr = no_capture_stderr
|
self.no_capture_stderr = no_capture_stderr
|
||||||
self.event_processor = EventProcessor()
|
self.event_processor = EventProcessor()
|
||||||
|
@ -145,8 +147,6 @@ class Daemon:
|
||||||
|
|
||||||
def stop_app(self):
|
def stop_app(self):
|
||||||
""" Stops the backends and the bus """
|
""" Stops the backends and the bus """
|
||||||
self.bus.post(ApplicationStoppedEvent())
|
|
||||||
|
|
||||||
for backend in self.backends.values():
|
for backend in self.backends.values():
|
||||||
backend.stop()
|
backend.stop()
|
||||||
|
|
||||||
|
@ -154,7 +154,7 @@ class Daemon:
|
||||||
if self.cron_scheduler:
|
if self.cron_scheduler:
|
||||||
self.cron_scheduler.stop()
|
self.cron_scheduler.stop()
|
||||||
|
|
||||||
def start(self):
|
def run(self):
|
||||||
""" Start the daemon """
|
""" Start the daemon """
|
||||||
if not self.no_capture_stdout:
|
if not self.no_capture_stdout:
|
||||||
sys.stdout = Logger(logger.info)
|
sys.stdout = Logger(logger.info)
|
||||||
|
@ -162,12 +162,7 @@ class Daemon:
|
||||||
sys.stderr = Logger(logger.warning)
|
sys.stderr = Logger(logger.warning)
|
||||||
|
|
||||||
set_thread_name('platypush')
|
set_thread_name('platypush')
|
||||||
|
logger.info('---- Starting platypush v.{}'.format(__version__))
|
||||||
print('---- Starting platypush v.{}'.format(__version__))
|
|
||||||
|
|
||||||
redis_conf = Config.get('backend.redis') or {}
|
|
||||||
self.bus = RedisBus(on_message=self.on_message(),
|
|
||||||
**redis_conf.get('redis_args', {}))
|
|
||||||
|
|
||||||
# Initialize the backends and link them to the bus
|
# Initialize the backends and link them to the bus
|
||||||
self.backends = register_backends(bus=self.bus, global_scope=True)
|
self.backends = register_backends(bus=self.bus, global_scope=True)
|
||||||
|
@ -198,9 +193,8 @@ def main():
|
||||||
"""
|
"""
|
||||||
Platypush daemon main
|
Platypush daemon main
|
||||||
"""
|
"""
|
||||||
|
|
||||||
app = Daemon.build_from_cmdline(sys.argv[1:])
|
app = Daemon.build_from_cmdline(sys.argv[1:])
|
||||||
app.start()
|
app.run()
|
||||||
|
|
||||||
|
|
||||||
# vim:sw=4:ts=4:et:
|
# vim:sw=4:ts=4:et:
|
||||||
|
|
|
@ -10,14 +10,5 @@ class ApplicationStartedEvent(Event):
|
||||||
super().__init__(*args, **kwargs)
|
super().__init__(*args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
class ApplicationStoppedEvent(Event):
|
|
||||||
"""
|
|
||||||
Event triggered when the application stops
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, *args, **kwargs):
|
|
||||||
super().__init__(*args, **kwargs)
|
|
||||||
|
|
||||||
|
|
||||||
# vim:sw=4:ts=4:et:
|
# vim:sw=4:ts=4:et:
|
||||||
|
|
||||||
|
|
|
@ -1,20 +1,18 @@
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
|
from platypush.config import Config
|
||||||
|
|
||||||
testdir = os.path.dirname(__file__)
|
testdir = os.path.dirname(__file__)
|
||||||
sys.path.insert(0, os.path.abspath(os.path.join(testdir, '..')))
|
sys.path.insert(0, os.path.abspath(os.path.join(testdir, '..')))
|
||||||
config_file = os.path.join(testdir, 'etc', 'config.yaml')
|
config_file = os.path.join(testdir, 'etc', 'config.yaml')
|
||||||
|
|
||||||
from platypush.config import Config
|
|
||||||
Config.init(config_file)
|
Config.init(config_file)
|
||||||
|
|
||||||
import platypush
|
|
||||||
|
|
||||||
|
class TimeoutException(RuntimeError):
|
||||||
class TestTimeoutException(RuntimeError):
|
|
||||||
def __init__(self, msg):
|
def __init__(self, msg):
|
||||||
self.msg = msg
|
self.msg = msg
|
||||||
|
|
||||||
|
|
||||||
# vim:sw=4:ts=4:et:
|
# vim:sw=4:ts=4:et:
|
||||||
|
|
||||||
|
|
|
@ -1,10 +1,9 @@
|
||||||
from .context import platypush
|
|
||||||
|
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
from platypush.event.hook import EventCondition
|
from platypush.event.hook import EventCondition
|
||||||
from platypush.message.event.ping import PingEvent
|
from platypush.message.event.ping import PingEvent
|
||||||
|
|
||||||
|
|
||||||
class TestEventParse(unittest.TestCase):
|
class TestEventParse(unittest.TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.condition = EventCondition.build({
|
self.condition = EventCondition.build({
|
||||||
|
@ -30,4 +29,3 @@ if __name__ == '__main__':
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|
||||||
# vim:sw=4:ts=4:et:
|
# vim:sw=4:ts=4:et:
|
||||||
|
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
import os
|
import os
|
||||||
|
|
||||||
from .context import config_file, TestTimeoutException
|
from .context import config_file, TimeoutException
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
import requests
|
import requests
|
||||||
|
@ -8,7 +8,7 @@ import sys
|
||||||
import time
|
import time
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
from threading import Thread
|
from threading import Thread, Event
|
||||||
|
|
||||||
from platypush import Daemon
|
from platypush import Daemon
|
||||||
from platypush.config import Config
|
from platypush.config import Config
|
||||||
|
@ -22,7 +22,7 @@ class TestHttp(unittest.TestCase):
|
||||||
Runs a remote command over HTTP via shell.exec plugin and gets the output """
|
Runs a remote command over HTTP via shell.exec plugin and gets the output """
|
||||||
|
|
||||||
timeout = 10
|
timeout = 10
|
||||||
sleep_secs = 10
|
sleep_secs = 5
|
||||||
db_file = '/tmp/platypush-tests.db'
|
db_file = '/tmp/platypush-tests.db'
|
||||||
test_user = 'platypush'
|
test_user = 'platypush'
|
||||||
test_pass = 'test'
|
test_pass = 'test'
|
||||||
|
@ -33,6 +33,7 @@ class TestHttp(unittest.TestCase):
|
||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, *args, **kwargs):
|
||||||
super().__init__(*args, **kwargs)
|
super().__init__(*args, **kwargs)
|
||||||
self.app = None
|
self.app = None
|
||||||
|
self._app_started = Event()
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
logging.basicConfig(level=logging.INFO, stream=sys.stdout)
|
logging.basicConfig(level=logging.INFO, stream=sys.stdout)
|
||||||
|
@ -68,26 +69,24 @@ class TestHttp(unittest.TestCase):
|
||||||
|
|
||||||
# A request with the wrong user/pass should fail.
|
# A request with the wrong user/pass should fail.
|
||||||
response = self.send_request(auth=('wrong', 'wrong'))
|
response = self.send_request(auth=('wrong', 'wrong'))
|
||||||
self.assertEqual(self.expected_login_redirect, response.url,
|
self.assertEqual(self.expected_login_redirect, response.url, 'A request with wrong credentials should fail')
|
||||||
'A request with wrong credentials should fail')
|
|
||||||
|
|
||||||
def start_daemon(self):
|
def start_daemon(self):
|
||||||
def _f():
|
self.app = Daemon(config_file=config_file)
|
||||||
self.app = Daemon(config_file=config_file)
|
Thread(target=lambda: self.app.run()).start()
|
||||||
self.app.start()
|
|
||||||
|
|
||||||
Thread(target=_f).start()
|
def stop_daemon(self):
|
||||||
|
if self.app:
|
||||||
|
self.app.stop_app()
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def on_timeout(msg):
|
def on_timeout(msg):
|
||||||
def _f(): raise TestTimeoutException(msg)
|
def _f(): raise TimeoutException(msg)
|
||||||
|
|
||||||
return _f
|
return _f
|
||||||
|
|
||||||
def send_request(self, **kwargs):
|
def send_request(self, **kwargs):
|
||||||
set_timeout(seconds=self.timeout,
|
set_timeout(seconds=self.timeout, on_timeout=self.on_timeout('Receiver response timed out'))
|
||||||
on_timeout=self.on_timeout('Receiver response timed out'))
|
|
||||||
|
|
||||||
response = requests.post(
|
response = requests.post(
|
||||||
'{}/execute'.format(self.base_url),
|
'{}/execute'.format(self.base_url),
|
||||||
json={
|
json={
|
||||||
|
@ -102,9 +101,7 @@ class TestHttp(unittest.TestCase):
|
||||||
return response
|
return response
|
||||||
|
|
||||||
def register_user(self):
|
def register_user(self):
|
||||||
set_timeout(seconds=self.timeout,
|
set_timeout(seconds=self.timeout, on_timeout=self.on_timeout('User registration response timed out'))
|
||||||
on_timeout=self.on_timeout('User registration response timed out'))
|
|
||||||
|
|
||||||
response = requests.post('{base_url}/register?redirect={base_url}/'.format(base_url=self.base_url), data={
|
response = requests.post('{base_url}/register?redirect={base_url}/'.format(base_url=self.base_url), data={
|
||||||
'username': self.test_user,
|
'username': self.test_user,
|
||||||
'password': self.test_pass,
|
'password': self.test_pass,
|
||||||
|
@ -119,11 +116,11 @@ class TestHttp(unittest.TestCase):
|
||||||
return Message.build(response.json())
|
return Message.build(response.json())
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
if self.app:
|
try:
|
||||||
self.app.stop_app()
|
self.stop_daemon()
|
||||||
|
finally:
|
||||||
if os.path.isfile(self.db_file):
|
if os.path.isfile(self.db_file):
|
||||||
os.unlink(self.db_file)
|
os.unlink(self.db_file)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|
Loading…
Reference in a new issue