def test_india_server_exclusions(self):
self.pillowtops['fluff'] = [
'custom.bihar.models.CareBiharFluffPillow',
'custom.opm.models.OpmCaseFluffPillow',
'custom.opm.models.OpmUserFluffPillow',
]
reject = Command.get_rejected_pillow_types(os.path.join(self.here, '..', '..', '..'), 'india')
pillows = Command.get_pillows_from_settings(self.pillowtops, reject)
has_bihar_pillow = False
for pillow in pillows:
assert pillow != 'custom.opm.models.OpmCaseFluffPillow'
if pillow == 'custom.bihar.models.CareBiharFluffPillow':
has_bihar_pillow = True
assert has_bihar_pillow
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, Command.get_pillows_from_settings(self.pillowtops))
def test_loading_no_existing_conf_file(self):
expected_reject = {}
reject = Command.get_rejected_pillow_types(os.path.join(self.here, '..', '..', '..'), 'production')
self.assertEqual(reject, expected_reject)
def test_loading_existing_conf_file(self):
expected_reject = {'pillowtop_blacklist': ['fluff']}
reject = Command.get_rejected_pillow_types(os.path.join(self.here, '..', '..', '..'), 'staging')
self.assertEqual(reject, expected_reject)
