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)
