def test_needs_check(self):
rf1 = RssFeed()
rf1.last_check = datetime.now()
rf1.update_frequency = Commons.load_time_delta("P1TS")
assert not rf1.needs_check()
rf1.last_check = datetime.now() - Commons.load_time_delta("P2TS")
assert rf1.needs_check()
rf1.update_frequency = Commons.load_time_delta("P7TS")
assert not rf1.needs_check()
rf2 = RssFeed()
rf2.last_check = datetime.now()
rf2.update_frequency = Commons.load_time_delta("PT5S")
assert not rf2.needs_check()
time.sleep(10)
assert rf2.needs_check()
def test_run_passive(self):
# Set up test servers and channels
serv1 = ServerMock(None)
serv1.name = "test_serv1"
chan1 = serv1.get_channel_by_name("test_chan1")
chan2 = serv1.get_channel_by_name("test_chan2")
serv2 = ServerMock(None)
serv2.name = "test_serv2"
chan3 = serv2.get_channel_by_name("test_chan1")
try:
self.hallo.add_server(serv1)
self.hallo.add_server(serv2)
# Set up rss feeds
rfl = RssFeedList()
rf1 = RssFeed()
rf1.url = "http://spangle.org.uk/hallo/test_rss.xml?1"
rf1.title = "test_feed1"
rf1.server_name = chan1.server.name
rf1.channel_name = chan1.name
rf1.update_frequency = Commons.load_time_delta("PT3600S")
rfl.add_feed(rf1)
rf2 = RssFeed()
rf2.url = "http://spangle.org.uk/hallo/test_rss.xml?2"
rf2.title = "test_feed2"
rf2.server_name = chan2.server.name
rf2.channel_name = chan2.name
rf2.update_frequency = Commons.load_time_delta("PT3600S")
rfl.add_feed(rf2)
rf3 = RssFeed()
rf3.url = "http://spangle.org.uk/hallo/test_rss.xml?3"
rf3.title = "test_feed1"
rf3.server_name = chan3.server.name
rf3.channel_name = chan3.name
rf3.update_frequency = Commons.load_time_delta("PT3600S")
rfl.add_feed(rf3)
# Splice this rss feed list into the function dispatcher's rss check object
rss_check_class = self.function_dispatcher.get_function_by_name("rss check")
rss_check_obj = self.function_dispatcher.get_function_object(rss_check_class) # type: FeedCheck
rss_check_obj.rss_feed_list = rfl
# Test passive feed updates
self.function_dispatcher.dispatch_passive(Function.EVENT_MINUTE, None, None, None, None)
# Check test server 1 data
serv1_data = serv1.get_send_data(6)
chan1_count = 0
chan2_count = 0
for data_line in serv1_data:
if data_line[1] == chan1:
chan1_count += 1
if data_line[1] == chan2:
chan2_count += 1
assert chan1_count == 3
assert chan2_count == 3
# Check test server 2 data
serv2_data = serv2.get_send_data(3, chan3, Server.MSG_MSG)
# Test that no updates are found the second run
rf1.last_check = None
rf2.last_check = None
rf3.last_check = None
self.function_dispatcher.dispatch_passive(Function.EVENT_MINUTE, None, None, None, None)
serv1.get_send_data(0)
serv2.get_send_data(0)
# Test that no feeds are checked before timeout, set urls to none and see if anything explodes.
self.failed = False
rf1.check_feed = self.do_not_call
rf2.check_feed = self.do_not_call
rf3.check_feed = self.do_not_call
self.function_dispatcher.dispatch_passive(Function.EVENT_MINUTE, None, None, None, None)
serv1.get_send_data(0)
serv2.get_send_data(0)
assert not self.failed, "check_feed() should not have been called on any feed."
finally:
self.hallo.remove_server(serv2)
self.hallo.remove_server(serv1)
