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