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()