class SocialCenterPageTestCase(TestCase): """Test Case 5: Social Center Must Retrieve Pages """ TEST_SITE = "test" TEST_PAGED_SITE = "test_paged" @classmethod def setUpClass(self): self.test_bot = TestBot() self.test_paged_bot = TestBotWithPages() self.social_center = SocialCenter() self.social_center.add_site(self.TEST_SITE, self.test_bot) self.social_center.add_site(self.TEST_PAGED_SITE, self.test_paged_bot) self.social_center.authenticate(self.TEST_SITE, "test_main_token", "test_sub_token") self.social_center.authenticate(self.TEST_PAGED_SITE, "test_main_token", "test_sub_token") @classmethod def tearDownClass(self): self.social_center.logout(self.TEST_SITE) self.social_center.logout(self.TEST_PAGED_SITE) self.social_center.remove_site(self.TEST_SITE) self.social_center.remove_site(self.TEST_PAGED_SITE) del self.social_center del self.test_bot del self.test_paged_bot def test_case(self): self.only_paged_site_has_pages() self.can_get_pages() def only_paged_site_has_pages(self): self.assertFalse(self.social_center.must_select_page(self.TEST_SITE)) self.assertTrue(self.social_center.must_select_page(self.TEST_PAGED_SITE)) self.assertRaises(NotImplementedError, self.social_center.get_pages, self.TEST_SITE, "any_token") # Should not raise any errors: pages = self.social_center.get_pages(self.TEST_PAGED_SITE, self.test_paged_bot._valid_main_token) def can_get_pages(self): self.social_center.authenticate(self.TEST_PAGED_SITE, "test_main_token", "test_sub_token") invalid_pages = self.social_center.get_pages(self.TEST_PAGED_SITE, "invalid_request_token") self.assertFalse(invalid_pages) valid_pages = self.social_center.get_pages(self.TEST_PAGED_SITE, self.test_paged_bot._valid_main_token) self.assertTrue(valid_pages) self.assertEqual(len(valid_pages), 10) for i in range(10): page = valid_pages[i] self.assertEqual(page["id"], str(i+1)) self.assertEqual(page["name"], "Page " + str(i+1))
class SocialCenterExposeTestCase(TestCase): """Test Case 3: Social Center Must Expose Social Network Properties """ TEST_SITE = "test" @classmethod def setUpClass(self): self.test_bot = TestBot() self.social_center = SocialCenter() self.social_center.add_site(self.TEST_SITE, self.test_bot) @classmethod def tearDownClass(self): self.social_center.remove_site(self.TEST_SITE) del self.social_center del self.test_bot def test_returns_bot_properties(self): #import pdb; pdb.set_trace() self.assertEqual(self.social_center.must_select_page(self.TEST_SITE), self.test_bot.must_select_page()) self.assertEqual(self.social_center.get_site_name(self.TEST_SITE), self.test_bot.get_site_name()) self.assertEqual(self.social_center.get_client_token_name(self.TEST_SITE), self.test_bot.get_client_token_name()) self.assertEqual(self.social_center.get_account_name(self.TEST_SITE), self.test_bot.get_account_name()) self.assertEqual(self.social_center.get_account_url(self.TEST_SITE), self.test_bot.get_account_url())
class SocialCenterExposeTestCase(TestCase): """Test Case 3: Social Center Must Expose Social Network Properties """ TEST_SITE = "test" @classmethod def setUpClass(self): self.test_bot = TestBot() self.social_center = SocialCenter() self.social_center.add_site(self.TEST_SITE, self.test_bot) @classmethod def tearDownClass(self): self.social_center.remove_site(self.TEST_SITE) del self.social_center del self.test_bot def test_returns_bot_properties(self): #import pdb; pdb.set_trace() self.assertEqual(self.social_center.must_select_page(self.TEST_SITE), self.test_bot.must_select_page()) self.assertEqual(self.social_center.get_site_name(self.TEST_SITE), self.test_bot.get_site_name()) self.assertEqual( self.social_center.get_client_token_name(self.TEST_SITE), self.test_bot.get_client_token_name()) self.assertEqual(self.social_center.get_account_name(self.TEST_SITE), self.test_bot.get_account_name()) self.assertEqual(self.social_center.get_account_url(self.TEST_SITE), self.test_bot.get_account_url())
class SocialCenterPageTestCase(TestCase): """Test Case 5: Social Center Must Retrieve Pages """ TEST_SITE = "test" TEST_PAGED_SITE = "test_paged" @classmethod def setUpClass(self): self.test_bot = TestBot() self.test_paged_bot = TestBotWithPages() self.social_center = SocialCenter() self.social_center.add_site(self.TEST_SITE, self.test_bot) self.social_center.add_site(self.TEST_PAGED_SITE, self.test_paged_bot) self.social_center.authenticate(self.TEST_SITE, "test_main_token", "test_sub_token") self.social_center.authenticate(self.TEST_PAGED_SITE, "test_main_token", "test_sub_token") @classmethod def tearDownClass(self): self.social_center.logout(self.TEST_SITE) self.social_center.logout(self.TEST_PAGED_SITE) self.social_center.remove_site(self.TEST_SITE) self.social_center.remove_site(self.TEST_PAGED_SITE) del self.social_center del self.test_bot del self.test_paged_bot def test_case(self): self.only_paged_site_has_pages() self.can_get_pages() def only_paged_site_has_pages(self): self.assertFalse(self.social_center.must_select_page(self.TEST_SITE)) self.assertTrue( self.social_center.must_select_page(self.TEST_PAGED_SITE)) self.assertRaises(NotImplementedError, self.social_center.get_pages, self.TEST_SITE, "any_token") # Should not raise any errors: pages = self.social_center.get_pages( self.TEST_PAGED_SITE, self.test_paged_bot._valid_main_token) def can_get_pages(self): self.social_center.authenticate(self.TEST_PAGED_SITE, "test_main_token", "test_sub_token") invalid_pages = self.social_center.get_pages(self.TEST_PAGED_SITE, "invalid_request_token") self.assertFalse(invalid_pages) valid_pages = self.social_center.get_pages( self.TEST_PAGED_SITE, self.test_paged_bot._valid_main_token) self.assertTrue(valid_pages) self.assertEqual(len(valid_pages), 10) for i in range(10): page = valid_pages[i] self.assertEqual(page["id"], str(i + 1)) self.assertEqual(page["name"], "Page " + str(i + 1))