0.2.3 #99
2 changed files with 24 additions and 1 deletions
|
|
@ -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:
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue