def setUp(self):
self.job_state = mock.create_autospec(JobState)
self.job_state.events_for_task.return_value = []
self.job = mock.create_autospec(Job)
self.job.job_state = self.job_state
self.generate_news_task = GenerateNewsFromRepositoryUpdates()
self.generate_news_task.job = self.job
def setUp(self):
self.job_state = mock.create_autospec(JobState)
self.job_state.events_for_task.return_value = []
self.job_state.processed_tasks = []
self.job = mock.create_autospec(Job)
self.job.job_state = self.job_state
self.task = ExtractSourcePackageFiles()
self.task.job = self.job
def create_mock_task(self, task_name, events=()):
mock_task = mock.create_autospec(BaseTask)
mock_task.task_name.return_value = task_name
mock_task.raised_events = [
self.create_mock_event(event['name'], event.get('arguments', None))
for event in events
]
return mock_task
def setUp(self):
# Since the tests module is used to provide the vendor rules,
# we dynamically add the needed function
self.packages = ['package1', 'package2']
self.mock_get_pseudo_package_list = mock.create_autospec(
lambda: None, return_value=self.packages)
sys.modules[__name__].get_pseudo_package_list = (
self.mock_get_pseudo_package_list
)
def setUp(self):
self.package_name = SourcePackageName.objects.create(name='dummy-package')
self.package = SourcePackage.objects.create(
source_package_name=self.package_name, version='1.0.0')
self.default_policy_version = '3.9.4.0'
self.debian_policy_name = SourcePackageName.objects.create(
name='debian-policy')
self.debian_policy = SourcePackage.objects.create(
source_package_name=self.debian_policy_name,
version=self.default_policy_version)
self.job_state = mock.create_autospec(JobState)
self.job_state.events_for_task.return_value = []
self.job = mock.create_autospec(Job)
self.job.job_state = self.job_state
self.task = UpdateStandardsVersionWarnings()
self.task.job = self.job
def setUp(self):
self.maintainer_email = '*****@*****.**'
self.uploaders = [
'*****@*****.**',
'*****@*****.**',
]
self.package = create_source_package({
'name': 'dummy-package',
'version': '1.0.0',
'maintainer': {
'name': 'Maintainer',
'email': self.maintainer_email,
},
'uploaders': self.uploaders,
})
self.repository = Repository.objects.create(
name='repo', shorthand='repo', default=True)
self.non_default_repository = Repository.objects.create(
name='nondef', shorthand='nondef')
self.password = '******'
self.user = User.objects.create_user(
main_email='*****@*****.**', password=self.password,
first_name='', last_name='')
self.team = Team.objects.create_with_slug(
owner=self.user,
name="Team",
maintainer_email=self.maintainer_email)
# Create a team for each of the uploaders and maintainers
self.teams = [
Team.objects.create_with_slug(owner=self.user, name="Team" + str(i))
for i in range(5)
]
self.job_state = mock.create_autospec(JobState)
self.job_state.events_for_task.return_value = []
self.job = mock.create_autospec(Job)
self.job.job_state = self.job_state
self.task = UpdateTeamPackagesTask()
self.task.job = self.job
def test_get_resource_content_utility_function_not_cached(self):
"""
Tests the :func:`pts.core.utils.http.get_resource_content` utility
function when the resource is not cached in the given cache instance.
"""
mock_cache = mock.create_autospec(HttpCache)
mock_cache.is_expired.return_value = True
expected_content = b"Some content"
mock_cache.get_content.return_value = expected_content
url = 'http://some.url.com'
content = get_resource_content(url, mock_cache)
self.assertEqual(content, expected_content)
# The function updated the cache
mock_cache.update.assert_called_once_with(url)
def test_get_resource_content_utlity_function_cached(self):
"""
Tests the :func:`pts.core.utils.http.get_resource_content` utility
function when the resource is cached in the given cache instance.
"""
mock_cache = mock.create_autospec(HttpCache)
mock_cache.is_expired.return_value = False
expected_content = b"Some content"
mock_cache.get_content.return_value = expected_content
url = 'http://some.url.com'
content = get_resource_content(url, mock_cache)
# The expected content is retrieved
self.assertEqual(content, expected_content)
# The function did not update the cache
self.assertFalse(mock_cache.update.called)
def get_mock_connection(self):
"""
Helper method returning a mock SMTP connection object.
"""
import smtplib
return mock.create_autospec(smtplib.SMTP, return_value={})
def create_mock_event(self, event_name, event_arguments=None):
mock_event = mock.create_autospec(Event)
mock_event.name = event_name
mock_event.arguments = event_arguments
return mock_event
def test_calls_process(self, mock_process):
cmd = MailNewsCommand()
cmd.input_file = mock.create_autospec(six.BytesIO)
mock_process.assert_called()
def test_calls_process(self, mock_process):
cmd = MailNewsCommand()
cmd.input_file = mock.create_autospec(six.BytesIO)
cmd.handle()
self.assertTrue(mock_process.called)
