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)
