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)