def setUp(self): self.test_board = TEST_BOARD base_url = self.test_board.url board_info = [(self.test_board.url, self.test_board.name, 5)] gfaqs_client = GFAQSClient() self.archiver = Archiver( board_info=board_info, base=base_url, pidfile="", gfaqs_client=gfaqs_client)
class ArchiverTest(TestCase): """ Unit test """ def setUp(self): self.test_board = TEST_BOARD base_url = self.test_board.url board_info = [(self.test_board.url, self.test_board.name, 5)] gfaqs_client = GFAQSClient() self.archiver = Archiver( board_info=board_info, base=base_url, pidfile="", gfaqs_client=gfaqs_client) @patch("gfaqs.archiver.BoardScraper", MockBoardScraper) def test_archive_board(self): self.archiver.archive_board(self.test_board, recursive=False) self.assertEquals(len(Topic.objects.all()), 1) @patch("gfaqs.archiver.TopicScraper", MockTopicScraper) def test_archive_topic(self): test_topic = TEST_TOPIC self.archiver.archive_topic(test_topic) self.assertEquals(len(Post.objects.all()), 1)
def handle(self, *args, **options): if len(args) != 1: print help() sys.exit(2) my_gfaqs_client = None if settings.GFAQS_LOGIN_AS_USER: my_gfaqs_client = AuthenticatedGFAQSClient( settings.GFAQS_LOGIN_EMAIL, settings.GFAQS_LOGIN_PASSWORD) else: my_gfaqs_client = GFAQSClient() daemon = Archiver(pidfile=PIDFILE, gfaqs_client=my_gfaqs_client) #create pid file if args[0] == "start": try: log_start() daemon.start() except IOError: print "unable to write pid file %s" % PIDFILE sys.exit(2) elif args[0] == "stop": daemon.stop() log_end() elif args[0] == "restart": daemon.restart() elif args[0] == "status": show_status() else: print "unknown command" print help() sys.exit(2)