def doesnt_return_the_same_feed_from_different_libraries(self): library1 = ContentFeedLibrary(shared=False) library1.register(ListItemFeed, 'li') library2 = ContentFeedLibrary(shared=False) library2.register(ListItemFeed, 'li') feed = ListItemFeed('[1, 2, 3]') record = library1.get_record(feed) assert_is_not(feed, library2.get_feed(record))
def has_isSubscribed_set_to_true_when_current_user_is_subscribed(self): from django.contrib.auth.models import User from subscriptions.models import Subscriber, Subscription from subscriptions.feeds import ContentFeedLibrary from main.feeds import NewLegislationFeed subscriber = Subscriber(username="******") subscriber.set_password("world") subscriber.save() library = ContentFeedLibrary() library.register(NewLegislationFeed, 'blah blah') feed = NewLegislationFeed() subscription = subscriber.subscribe(feed) subscriber.save() # Make the request. client = Client() assert client.login(username="******", password="******") response = client.get('/legislation/') # Check the context assert response.context['is_subscribed'] assert_equal(response.context['subscription'], subscription)
def causes_the_same_feed_record_to_be_returned_on_different_calls_to_getRecord(self): library = ContentFeedLibrary(shared=False) library.register(ListItemFeed, 'li') feed = ListItemFeed('[1, 2, 3]') record = library.get_record(feed) assert_is(record, library.get_record(feed))
def setUp(self): Subscriber.objects.all().delete() ContentFeedRecord.objects.all().delete() ContentFeedParameter.objects.all().delete() Subscription.objects.all().delete() ContentFeedLibrary().register(ContentFeed, 'generic content feed') record = self.record = ContentFeedRecord.objects.create(feed_name='generic content feed') subscriber = self.subscriber = Subscriber.objects.create()
def setup(self): Subscription.objects.all().delete() Subscriber.objects.all().delete() ContentFeedRecord.objects.all().delete() self.library = ContentFeedLibrary(shared=False) self.library.register(ListItemFeed, 'my list items') self.subscriber = Subscriber.objects.create() self.feed = ListItemFeed('[1, 2, 3]') self.feed_record = self.library.get_record(self.feed)
def setUp(self): library = self.library = ContentFeedLibrary(shared=False) library.register(ListItemFeed, 'list feed') self.feeds = [ ListItemFeed("['hello']"), ListItemFeed("['world']") ] for feed in self.feeds: feed.get_last_updated_time = Mock(return_value=datetime.date.today()) self.feed_records = [library.get_record(feed) for feed in self.feeds]
def setUp(self): Subscriber.objects.all().delete() ContentFeedRecord.objects.all().delete() ContentFeedParameter.objects.all().delete() Subscription.objects.all().delete() library = self.library = ContentFeedLibrary(shared=False) library.register(ListItemFeed, 'list feed') feed = self.feed = ListItemFeed('[1,2,3]') record = self.record = library.get_record(feed) subscriber = self.subscriber = Subscriber.objects.create() subscriber.subscribe(feed, library) subscriber.save()
def setUp(self): ContentFeedRecord.objects.all().delete() Subscriber.objects.all().delete() library = self.library = ContentFeedLibrary(shared=False) library.register(ListItemFeed, 'list feed') feeds = [ ListItemFeed("['hello']"), ListItemFeed("['world']") ] feed_records = [library.get_record(feed) for feed in feeds] subscriber = Subscriber.objects.create() subscriber.subscribe(feeds[0], library) self.keeper = feed_records[0] self.tosser = feed_records[1]
def setup(self): library = self.library = ContentFeedLibrary() # Create a subscriber subscriber = self.subscriber = Mock() # Create a subscription subscription = self.subscription = Mock() subscription.last_sent = datetime.datetime(2011, 1, 1, 0, 0) subscription.feed_record.last_updated = datetime.datetime(2011, 8, 4, 6, 50) subscription.feed_record.feed_name = 'MockFeed' param1 = Mock(); param1.name = 'p1'; param1.value = '1' param2 = Mock(); param2.name = 'p2'; param2.value = '2' subscription.feed.feed_params.all = lambda: [param1, param2] # The subscriber is subscribed to the subscription we created subscriber.subscriptions.all = lambda: [subscription]
def setUp(self): class DoNothinView (object): def get_context_data(self, **kwargs): # This is just so that SingleSubscriptionMixin.get_context_data # super has something to call. return {} library = ContentFeedLibrary(shared=False) library.register(ListItemFeed, 'my list item feed') class SubscriptionView (SingleSubscriptionMixin, DoNothinView): def get_content_feed_library(self): return library def get_content_feed(self): return ListItemFeed('[1,2,3]') self.view = SubscriptionView() self.view.request = Mock() self.view.feed_data = ContentFeed
def setUp(self): LegFile.objects.all().delete() key = 0 for intro, final in [ (datetime.date(2011, 1, 28), datetime.date(2011, 1, 29)), (datetime.date(2010, 7, 28), datetime.date(2010, 7, 29)), (datetime.date(2011, 8, 17), datetime.date(2011, 8, 18)), (datetime.date(2006, 12, 11), datetime.date(2006, 12, 12)), (datetime.date(2006, 12, 12), datetime.date(2006, 12, 13)) ]: LegFile.objects.create(intro_date=intro, final_date=final, key=key, date_scraped=datetime.date.today()) key += 1 self.library = ContentFeedLibrary(shared=False) self.library.register(NewLegFilesFeed, 'feed_class_123456') self.record = self.library.get_record(NewLegFilesFeed())
def test_subscription_flow(): Subscriber.objects.all().delete() ContentFeedRecord.objects.all().delete() del mail.outbox[:] library = ContentFeedLibrary(shared=True) # First, let's make some content content = [(datetime.date(2011, 12, 13), 1), (datetime.date(2011, 12, 13), 2), (datetime.date(2011, 12, 13), 3), (datetime.date(2011, 12, 13), 4), (datetime.date(2011, 12, 13), 5)] # Then, let's make a ContentFeed that watches it. class EasyContentFeed(ContentFeed): def get_content(self): return content def get_params(self): return {} def get_updates_since(self, previous): return [t for t in content if t[0] > previous.date()] def get_last_updated_time(self): return max([t[0] for t in content]) def get_changes_to(self, item, since): if since.date() < item[0]: return { 'value': item[1] }, datetime.datetime(2011, 12, 13, 14, 15) def get_label(self): return 'Easy...' library.register(EasyContentFeed, 'easy') feed = EasyContentFeed() # Now we want a subscriber to subscribe to the feed. subscriber = Subscriber.objects.create() subscription = subscriber.subscribe(feed, library=library) # Assume that we last sent the subscription before the current items. subscription.last_sent = datetime.date(2011, 11, 11) subscription.save() # Now, update all the feeds dates/times and send out the updated content. update = updatefeeds.Command() update.handle() send = sendfeedupdates.Command() send.handle() # Check that we have a message to send. assert_equal(len(mail.outbox), 1) assert_equal(mail.outbox[0].subject[:19], 'Philly Councilmatic') assert_equal(mail.outbox[0].body, ( '\n\nPhiladelphia Councilmatic!\n==========================\n\nYou are subscribed to the following feeds:\n\n\n* Easy...\n\n\n--------------------------------------------------------------------------------\n\n \n\nvalue: 5\n\nMore at http://example.com\n\n\n--------------------------------------------------------------------------------\n\n \n\nvalue: 2\n\nMore at http://example.com\n\n\n--------------------------------------------------------------------------------\n\n \n\nvalue: 4\n\nMore at http://example.com\n\n\n--------------------------------------------------------------------------------\n\n \n\nvalue: 1\n\nMore at http://example.com\n\n\n--------------------------------------------------------------------------------\n\n \n\nvalue: 3\n\nMore at http://example.com\n\n\n\nTo manage your subscriptions, visit http://example.com/subscriptions/\n' )) # Cool. Clear the mailbox. del mail.outbox[:] # Now, if we update the feed times and send the updated content, there # should be nothing to send. update = updatefeeds.Command() update.handle() send = sendfeedupdates.Command() send.handle() assert_equal(len(mail.outbox), 0) # Put something else into the feed. content.append((datetime.date(2012, 2, 4), 6)) # Now, update all the feeds dates/times and send out the updated content. update = updatefeeds.Command() update.handle() send = sendfeedupdates.Command() send.handle() assert_equal(len(mail.outbox), 1)
def subscription_title(subscription, library=None): if library is None: library = ContentFeedLibrary() feed = library.get_feed(subscription.feed_record) return feed.get_label()
import json import logging from datetime import date, time, datetime from collections import defaultdict from itertools import chain from itertools import product from subscriptions.feeds import ContentFeed from subscriptions.feeds import ContentFeedLibrary from phillyleg.models import LegFile from phillyleg.models import LegMinutes from haystack.query import SearchQuerySet log = logging.getLogger(__name__) library = ContentFeedLibrary() class NewLegislationFeed (ContentFeed): def get_content(self): return LegFile.objects.all().order_by('-intro_date') def get_updates_since(self, datetime): return self.get_content().filter(intro_date__gt=datetime) def get_changes_to(self, legfile, since_datetime): if since_datetime.date() <= legfile.intro_date: return {'Title': legfile.title}, datetime.combine(legfile.intro_date, time()) else: return {}, datetime.min def get_last_updated_time(self):