def test_refresh_not_called_after_save_if_disabled(self):
app_settings.SEGMENTS_REFRESH_ON_SAVE = False
s = AllUserSegmentFactory()
s.refresh = Mock()
s.save()
self.assertEqual(s.refresh.call_count, 0)
app_settings.SEGMENTS_REFRESH_ON_SAVE = True
def test_segment_refresh(self):
s = AllUserSegmentFactory()
UserFactory()
s.refresh()
self.assertEqual(len(s), 2)
# Change up the segment to only match one user
s.definition = 'select * from %s where id = %s limit 1' % (
user_table(), self.u.id)
s.save()
self.assertEqual(len(s), 1)
s.refresh()
self.assertEqual(len(s), 1)
# Add a 3rd user, should still only store one user
u3 = UserFactory()
s.refresh()
self.assertEqual(len(s), 1)
# Expand the definition to include 3 users again
s.definition = 'select * from %s' % (user_table())
s.save()
s.refresh()
self.assertEqual(len(s), 3)
# Remove one user
u3.delete()
s.refresh()
self.assertEqual(len(s), 2)
def test_refresh_not_called_after_save_if_disabled(self):
app_settings.SEGMENTS_REFRESH_ON_SAVE = False
s = AllUserSegmentFactory()
s.refresh = Mock()
s.save()
self.assertEqual(s.refresh.call_count, 0)
app_settings.SEGMENTS_REFRESH_ON_SAVE = True
def test_segment_refresh(self):
s = AllUserSegmentFactory()
UserFactory()
s.refresh()
self.assertEqual(len(s), 2)
# Change up the segment to only match one user
s.definition = 'select * from %s where id = %s limit 1' % (user_table(), self.u.id)
s.save()
self.assertEqual(len(s), 1)
s.refresh()
self.assertEqual(len(s), 1)
# Add a 3rd user, should still only store one user
u3 = UserFactory()
s.refresh()
self.assertEqual(len(s), 1)
# Expand the definition to include 3 users again
s.definition = 'select * from %s' % (user_table())
s.save()
s.refresh()
self.assertEqual(len(s), 3)
# Remove one user
u3.delete()
s.refresh()
self.assertEqual(len(s), 2)
def test_refresh_after_save(self):
s = AllUserSegmentFactory()
s.refresh = Mock()
s.save()
self.assertEqual(s.refresh.call_count, 1)
def test_refresh_after_save(self):
s = AllUserSegmentFactory()
s.refresh = Mock()
s.save()
self.assertEqual(s.refresh.call_count, 1)
