def test_get_repo_units(self): # Setup self.populate() # Test conduit = ProfilerConduit() units1 = conduit.get_repo_units(self.REPO_ID, content_type_id=self.TYPE_1_DEF.id, additional_unit_fields=[]) units2 = conduit.get_repo_units(self.REPO_ID, content_type_id=self.TYPE_2_DEF.id, additional_unit_fields=[]) # Verify that all the units in the repo with given type are returned along with unit_key self.assertEquals(len(units1), 9) for u in units1: self.assertTrue('key-1' in u.unit_key) self.assertFalse('key-2' in u.unit_key) self.assertEquals(len(units2), 9) for u in units2: self.assertTrue('key-2' in u.unit_key) self.assertFalse('key-1' in u.unit_key)
def test_get_repo_units(self): # Setup self.populate() # Test conduit = ProfilerConduit() units1 = conduit.get_repo_units(self.REPO_ID, content_type_id=self.TYPE_1_DEF.id, additional_unit_fields=[]) units2 = conduit.get_repo_units(self.REPO_ID, content_type_id=self.TYPE_2_DEF.id, additional_unit_fields=[]) # Verify that all the units in the repo with given type are returned along with unit_key self.assertEquals(len(units1), 9) for u in units1: self.assertTrue('key-1' in u.unit_key) self.assertFalse('key-2' in u.unit_key) self.assertEquals(len(units2), 9) for u in units2: self.assertTrue('key-2' in u.unit_key) self.assertFalse('key-1' in u.unit_key)
def test_get_repo_units_additional_field(self): # Setup self.populate(additional_key='extra_field') # Test conduit = ProfilerConduit() units = conduit.get_repo_units(self.REPO_ID, content_type_id=self.TYPE_1_DEF.id, additional_unit_fields=['extra_field']) # Verify that all the units in the repo with given type are returned along with unit_key and extra field self.assertEquals(len(units), 9) for u in units: self.assertTrue('key-1' in u.unit_key) self.assertTrue('extra_field' in u.metadata)
def test_get_repo_units_additional_field(self): # Setup self.populate(additional_key='extra_field') # Test conduit = ProfilerConduit() units = conduit.get_repo_units(self.REPO_ID, content_type_id=self.TYPE_1_DEF.id, additional_unit_fields=['extra_field']) # Verify that all the units in the repo with given type are returned along with unit_key and extra field self.assertEquals(len(units), 9) for u in units: self.assertTrue('key-1' in u.unit_key) self.assertTrue('extra_field' in u.metadata)