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)