Differential D2514 Diff 8947 swh/core/api/tests/test_gunicorn.py

Changeset View

Standalone View

swh/core/api/tests/test_gunicorn.py

	# Copyright (C) 2019 The Software Heritage developers			# Copyright (C) 2019 The Software Heritage developers
	# See the AUTHORS file at the top-level directory of this distribution			# See the AUTHORS file at the top-level directory of this distribution
	# License: GNU General Public License version 3, or any later version			# License: GNU General Public License version 3, or any later version
	# See top-level LICENSE file for more information			# See top-level LICENSE file for more information

	import os			import os
				import pkg_resources
	from unittest.mock import patch			from unittest.mock import patch

	import swh.core.api.gunicorn_config as gunicorn_config			import swh.core.api.gunicorn_config as gunicorn_config


	def test_post_fork_default():			def test_post_fork_default():
	with patch('sentry_sdk.init') as sentry_sdk_init:			with patch('sentry_sdk.init') as sentry_sdk_init:
	gunicorn_config.post_fork(None, None)			gunicorn_config.post_fork(None, None)

	sentry_sdk_init.assert_not_called()			sentry_sdk_init.assert_not_called()


	def test_post_fork_with_dsn_env():			def test_post_fork_with_dsn_env():
	flask_integration = object() # unique object to check for equality			flask_integration = object() # unique object to check for equality
	with patch('sentry_sdk.integrations.flask.FlaskIntegration',			with patch('sentry_sdk.integrations.flask.FlaskIntegration',
	new=lambda: flask_integration):			new=lambda: flask_integration):
	with patch('sentry_sdk.init') as sentry_sdk_init:			with patch('sentry_sdk.init') as sentry_sdk_init:
	with patch.dict(os.environ, {'SWH_SENTRY_DSN': 'test_dsn'}):			with patch.dict(os.environ, {'SWH_SENTRY_DSN': 'test_dsn'}):
	gunicorn_config.post_fork(None, None)			gunicorn_config.post_fork(None, None)

	sentry_sdk_init.assert_called_once_with(			sentry_sdk_init.assert_called_once_with(
	dsn='test_dsn',			dsn='test_dsn',
	integrations=[flask_integration],			integrations=[flask_integration],
	debug=False,			debug=False,
				release=None,
				)


				def test_post_fork_with_package_env():
				flask_integration = object() # unique object to check for equality
				with patch('sentry_sdk.integrations.flask.FlaskIntegration',
				new=lambda: flask_integration):
				with patch('sentry_sdk.init') as sentry_sdk_init:
				with patch.dict(os.environ, {'SWH_SENTRY_DSN': 'test_dsn',
				'SWH_MAIN_PACKAGE': 'swh.core'}):
				gunicorn_config.post_fork(None, None)

				version = pkg_resources.get_distribution('swh.core').version

				sentry_sdk_init.assert_called_once_with(
				dsn='test_dsn',
				integrations=[flask_integration],
				debug=False,
				release='swh.core@' + version,
	)			)


	def test_post_fork_debug():			def test_post_fork_debug():
	flask_integration = object() # unique object to check for equality			flask_integration = object() # unique object to check for equality
	with patch('sentry_sdk.integrations.flask.FlaskIntegration',			with patch('sentry_sdk.integrations.flask.FlaskIntegration',
	new=lambda: flask_integration):			new=lambda: flask_integration):
	with patch('sentry_sdk.init') as sentry_sdk_init:			with patch('sentry_sdk.init') as sentry_sdk_init:
	with patch.dict(os.environ, {'SWH_SENTRY_DSN': 'test_dsn',			with patch.dict(os.environ, {'SWH_SENTRY_DSN': 'test_dsn',
	'SWH_SENTRY_DEBUG': '1'}):			'SWH_SENTRY_DEBUG': '1'}):
	gunicorn_config.post_fork(None, None)			gunicorn_config.post_fork(None, None)

	sentry_sdk_init.assert_called_once_with(			sentry_sdk_init.assert_called_once_with(
	dsn='test_dsn',			dsn='test_dsn',
	integrations=[flask_integration],			integrations=[flask_integration],
	debug=True,			debug=True,
				release=None,
	)			)


	def test_post_fork_no_flask():			def test_post_fork_no_flask():
	with patch('sentry_sdk.init') as sentry_sdk_init:			with patch('sentry_sdk.init') as sentry_sdk_init:
	with patch.dict(os.environ, {'SWH_SENTRY_DSN': 'test_dsn'}):			with patch.dict(os.environ, {'SWH_SENTRY_DSN': 'test_dsn'}):
	gunicorn_config.post_fork(None, None, flask=False)			gunicorn_config.post_fork(None, None, flask=False)

	sentry_sdk_init.assert_called_once_with(			sentry_sdk_init.assert_called_once_with(
	dsn='test_dsn',			dsn='test_dsn',
	integrations=[],			integrations=[],
	debug=False,			debug=False,
				release=None,
	)			)


	def test_post_fork_extras():			def test_post_fork_extras():
	flask_integration = object() # unique object to check for equality			flask_integration = object() # unique object to check for equality
	with patch('sentry_sdk.integrations.flask.FlaskIntegration',			with patch('sentry_sdk.integrations.flask.FlaskIntegration',
	new=lambda: flask_integration):			new=lambda: flask_integration):
	with patch('sentry_sdk.init') as sentry_sdk_init:			with patch('sentry_sdk.init') as sentry_sdk_init:
	with patch.dict(os.environ, {'SWH_SENTRY_DSN': 'test_dsn'}):			with patch.dict(os.environ, {'SWH_SENTRY_DSN': 'test_dsn'}):
	gunicorn_config.post_fork(			gunicorn_config.post_fork(
	None, None, sentry_integrations=['foo'],			None, None, sentry_integrations=['foo'],
	extra_sentry_kwargs={'bar': 'baz'})			extra_sentry_kwargs={'bar': 'baz'})

	sentry_sdk_init.assert_called_once_with(			sentry_sdk_init.assert_called_once_with(
	dsn='test_dsn',			dsn='test_dsn',
	integrations=['foo', flask_integration],			integrations=['foo', flask_integration],
	debug=False,			debug=False,
	bar='baz',			bar='baz',
				release=None,
	)			)