0.2.3 #99
2 changed files with 24 additions and 1 deletions
|
|
@ -23,7 +23,7 @@ class Client:
|
||||||
stream = Stream
|
stream = Stream
|
||||||
|
|
||||||
def __init__(self, rules=None):
|
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):
|
def __enter__(self):
|
||||||
for rule in self.rules:
|
for rule in self.rules:
|
||||||
|
|
|
||||||
|
|
@ -245,3 +245,26 @@ class FeedCollectorTestCase(TestCase):
|
||||||
self.assertEquals(
|
self.assertEquals(
|
||||||
third_post.title, "Birmingham head teacher threatened over LGBT lessons"
|
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