def test_with_blacklist_items(self): expected_pillows = {u'corehq.pillows.case.CasePillow', u'corehq.pillows.xform.XFormPillow', u'corehq.pillows.domain.DomainPillow', u'corehq.pillows.user.UserPillow', u'corehq.pillows.application.AppPillow', u'corehq.pillows.group.GroupPillow', u'corehq.pillows.sms.SMSPillow', u'corehq.pillows.user.GroupToUserPillow', u'corehq.pillows.user.UnknownUsersPillow', u'corehq.pillows.sofabed.FormDataPillow', u'corehq.pillows.sofabed.CaseDataPillow'} self.assertEqual(expected_pillows, apply_pillow_actions_to_pillows( [{'exclude_groups': ['phonelog']}], self.pillowtops))
def test_with_blacklist_items(self): expected_pillows = {'CasePillow', 'XFormPillow', 'DomainPillow', 'UserPillow', 'AppPillow', 'GroupPillow', 'SMSPillow', 'GroupToUserPillow', 'UnknownUsersPillow', 'FormDataPillow', 'CaseDataPillow', 'FakeConstructedPillowName', } configs_back = apply_pillow_actions_to_pillows([{'exclude_groups': ['phonelog']}], self.pillowtops) self.assertEqual(expected_pillows, set([c.name for c in configs_back]))
def test_no_blacklist_items(self): expected_pillows = { u'corehq.pillows.case.CasePillow', u'corehq.pillows.xform.XFormPillow', u'corehq.pillows.domain.DomainPillow', u'corehq.pillows.user.UserPillow', u'corehq.pillows.application.AppPillow', u'corehq.pillows.group.GroupPillow', u'corehq.pillows.sms.SMSPillow', u'corehq.pillows.user.GroupToUserPillow', u'corehq.pillows.user.UnknownUsersPillow', u'corehq.pillows.sofabed.FormDataPillow', u'corehq.pillows.sofabed.CaseDataPillow', u'corehq.pillows.log.PhoneLogPillow' } self.assertEqual(expected_pillows, apply_pillow_actions_to_pillows([], self.pillowtops))