def test_strategy_remove_old_fields_when_new_changed(self):
self.strategy.impression_pacing_interval = 'derp'
self.strategy.impression_pacing_type = 'derp'
self.strategy.impression_pacing_amount = 1
fields_to_remove = self.strategy._migration_asst()
expected = ['impression_cap']
six.assertCountEqual(self, expected, fields_to_remove)
def test_strategy_remove_all_fields_when_no_changes(self):
fields_to_remove = self.strategy._migration_asst()
expected = ['impression_pacing_interval', 'impression_pacing_type', 'impression_pacing_amount',
'impression_cap']
six.assertCountEqual(self, expected, fields_to_remove)
