Пример #1
0
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'], "*****@*****.**")
Пример #2
0
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"*****@*****.**")