class SearchMemberCollectionTest(unittest.TestCase): def setUp(self): Account.default_adapter = MockAdapter self.members = Search( Account(account_id="100", public_key="xxx", private_key="yyy"), { 'search_id': 1024 }).members def test_can_fetch_all_members(self): del (self.members.search['search_id']) with self.assertRaises(ex.NoSearchIdError): self.members.fetch_all() self.assertEqual(self.members.search.account.adapter.called, 0) def test_can_fetch_all_members2(self): # Setup MockAdapter.expected = [{ 'member_id': 200, 'email': "*****@*****.**" }, { 'member_id': 201, 'email': "*****@*****.**" }, { 'member_id': 202, 'email': "*****@*****.**" }] members = self.members.fetch_all() self.assertEqual(self.members.search.account.adapter.called, 1) self.assertEqual(self.members.search.account.adapter.call, ('GET', '/searches/1024/members', {})) self.assertIsInstance(members, dict) self.assertEqual(3, len(members)) self.assertEqual(3, len(self.members)) self.assertIsInstance(self.members[200], Member) self.assertIsInstance(self.members[201], Member) self.assertIsInstance(self.members[202], Member) self.assertEqual(self.members[200]['email'], "*****@*****.**") self.assertEqual(self.members[201]['email'], "*****@*****.**") self.assertEqual(self.members[202]['email'], "*****@*****.**")
class SearchMemberCollectionTest(unittest.TestCase): def setUp(self): Account.default_adapter = MockAdapter self.members = Search( Account(account_id="100", public_key="xxx", private_key="yyy"), {'search_id': 1024} ).members def test_can_fetch_all_members(self): del(self.members.search['search_id']) with self.assertRaises(ex.NoSearchIdError): self.members.fetch_all() self.assertEquals(self.members.search.account.adapter.called, 0) def test_can_fetch_all_members2(self): # Setup MockAdapter.expected = [ {'member_id': 200, 'email': u"*****@*****.**"}, {'member_id': 201, 'email': u"*****@*****.**"}, {'member_id': 202, 'email': u"*****@*****.**"} ] members = self.members.fetch_all() self.assertEquals(self.members.search.account.adapter.called, 1) self.assertEquals( self.members.search.account.adapter.call, ('GET', '/searches/1024/members', {})) self.assertIsInstance(members, dict) self.assertEquals(3, len(members)) self.assertEquals(3, len(self.members)) self.assertIsInstance(self.members[200], Member) self.assertIsInstance(self.members[201], Member) self.assertIsInstance(self.members[202], Member) self.assertEquals(self.members[200]['email'], u"*****@*****.**") self.assertEquals(self.members[201]['email'], u"*****@*****.**") self.assertEquals(self.members[202]['email'], u"*****@*****.**")