class TestContentUnitsSearchNonWeb(base.PulpServerTests):
def setUp(self):
super(TestContentUnitsSearchNonWeb, self).setUp()
self.controller = ContentUnitsSearch()
@mock.patch(
'pulp.server.managers.repo.unit_association_query.RepoUnitAssociationQueryManager.find_by_criteria'
)
def test_add_repo_memberships_empty(self, mock_find):
# make sure it doesn't do a search for associations if there are no
# units found
self.controller._add_repo_memberships([], 'rpm')
self.assertEqual(mock_find.call_count, 0)
@mock.patch(
'pulp.server.managers.repo.unit_association_query.RepoUnitAssociationQueryManager.find_by_criteria'
)
def test_add_repo_memberships_(self, mock_find):
mock_find.return_value = [{'repo_id': 'repo1', 'unit_id': 'unit1'}]
units = [{'_id': 'unit1'}]
ret = self.controller._add_repo_memberships(units, 'rpm')
self.assertEqual(mock_find.call_count, 1)
self.assertEqual(len(ret), 1)
self.assertEqual(ret[0].get('repository_memberships'), ['repo1'])
class TestContentUnitsSearchNonWeb(base.PulpServerTests):
def setUp(self):
super(TestContentUnitsSearchNonWeb, self).setUp()
self.controller = ContentUnitsSearch()
@mock.patch('pulp.server.managers.repo.unit_association_query.RepoUnitAssociationQueryManager.find_by_criteria')
def test_add_repo_memberships_empty(self, mock_find):
# make sure it doesn't do a search for associations if there are no
# units found
self.controller._add_repo_memberships([], 'rpm')
self.assertEqual(mock_find.call_count, 0)
@mock.patch('pulp.server.managers.repo.unit_association_query.RepoUnitAssociationQueryManager.find_by_criteria')
def test_add_repo_memberships_(self, mock_find):
mock_find.return_value = [{'repo_id':'repo1', 'unit_id':'unit1'}]
units = [{'_id': 'unit1'}]
ret = self.controller._add_repo_memberships(units, 'rpm')
self.assertEqual(mock_find.call_count, 1)
self.assertEqual(len(ret), 1)
self.assertEqual(ret[0].get('repository_memberships'), ['repo1'])
def setUp(self):
super(TestContentUnitsSearchNonWeb, self).setUp()
self.controller = ContentUnitsSearch()
def setUp(self):
super(TestContentUnitsSearchNonWeb, self).setUp()
self.controller = ContentUnitsSearch()
