Skip disabled rules

This commit is contained in:
sonny 2020-05-10 23:04:57 +02:00
parent aeb85bd2cf
commit f4adb9635a
2 changed files with 24 additions and 1 deletions

View file

@ -23,7 +23,7 @@ class Client:
stream = Stream
def __init__(self, rules=None):
self.rules = rules if rules else CollectionRule.objects.all()
self.rules = rules if rules else CollectionRule.objects.filter(enabled=True)
def __enter__(self):
for rule in self.rules:

View file

@ -245,3 +245,26 @@ class FeedCollectorTestCase(TestCase):
self.assertEquals(
third_post.title, "Birmingham head teacher threatened over LGBT lessons"
)
@freeze_time("2019-02-22 12:30:00")
def test_disabled_rules(self):
rules = (
CollectionRuleFactory(enabled=False),
CollectionRuleFactory(enabled=True),
)
self.mocked_parse.return_value = multiple_mock
collector = FeedCollector()
collector.collect()
for rule in rules:
rule.refresh_from_db()
self.assertEquals(Post.objects.count(), 3)
self.assertEquals(rules[1].succeeded, True)
self.assertEquals(rules[1].last_suceeded, timezone.now())
self.assertEquals(rules[1].error, None)
self.assertEquals(rules[0].last_suceeded, None)
self.assertEquals(rules[0].succeeded, False)