def test_add_sources(self): url = 'sup' self.mock_find_feed.return_value = url feed.add_sources({'the something times': ['some_url']}) self.assertEquals(Source.query.count(), 1) # Ensure duplicates aren't added. feed.add_sources({'the something times': ['some_url']}) self.assertEquals(Source.query.count(), 1)
def create_sources(filepath): """ Load feeds from a JSON file. It should consist of an dict of source name => list of feeds like so:: { 'The New York Times': [ 'http//www.nytimes.com/services/xml/rss/nyt/World.xml', 'http//www.nytimes.com/services/xml/rss/nyt/politics.xml' ] } """ logger.info('Loading sources from file. This may take awhile...') sources = open(filepath, 'r') raw_sources = json.load(sources) feed.add_sources(raw_sources)
def create_sources(): sources = open('manage/core/data/seed_sources.json', 'r') raw_sources = json.load(sources) feed.add_sources(raw_sources)