def test_should_remove_old( self ): global db feeds = list( Feed.view("feed/old") ) self.assertEqual( 1, len(feeds) ) remove_old( db, today = datetime.date(2012,3,16) ) feeds = list( Feed.view("feed/old") ) self.assertEqual( 0, len(feeds) )
def test_should_see_bad_attachment( self ): global db feeds1 = list( Feed.view("feed/new") ) self.assertEqual( 1, len(feeds1) ) feeds2 = list( new_feed_iter() ) self.assertEqual( 1, len(feeds2) ) f, rdr= validate_and_open( feeds2[0] ) self.assertIsNone( rdr ) self.assertEquals( "wrong format attachment", f.status )
def test_should_process_feed( self ): arrivals = [] def track_arrival( mappings, report ): arrivals.append( report ) locations = [] def track_location( mappings, report ): locations.append( report ) mappings = {} counts= transform_new( mappings, new_feed_iter(), track_arrival, track_location, today=datetime.date(2012,3,15) ) self.assertEqual( 1, counts['all'] ) self.assertEqual( 1, counts['arrival'] ) self.assertEqual( 0, counts['location'] ) self.assertEqual( 1, len(arrivals) ) self.assertEqual( 0, len(locations) ) feeds2 = list( new_feed_iter() ) self.assertEqual( 0, len(feeds2) ) old = list( Feed.view('feed/old') ) self.assertEqual( 1, len(old) ) self.assertEqual( "processed", old[0].status )