예제 #1
0
    def test_success(self, taskqueue, users_len=3):
        pswd_users_emails = [mommy.save_one(PswdUserEmail, email="*****@*****.**" % i) for i in range(users_len)]
        pswd_users = [mommy.save_one(PswdUser) for i in range(users_len)]
        keys = [us.key for us in pswd_users]
        for e, u in izip(pswd_users_emails, pswd_users):
            CreateEmailUser(e, u)()

        def task_add_mock(url):
            cursor = router.to_handler(url)[1][0]
            tasks(cursor)

        taskqueue.add = task_add_mock
        tasks()

        self.assertIsNone(PswdUser.query().get())

        main_users = ndb.get_multi(keys)

        for user, pswd_user_email, pswd_user in izip(main_users, pswd_users_emails, pswd_users):
            self.assertIsInstance(user, MainUser)
            self.assertEqual(pswd_user_email.email, user.email)
            self.assertEqual(pswd_user_email.email, user.name)
            self.assertEqual(pswd_user.creation, user.creation)
            self.assertListEqual([""], user.groups)
예제 #2
0
 def __init__(self, start_cursor=None):
     super(UserSearch, self).__init__(PswdUser.query_by_creation(), 1, start_cursor, 0, False, False)
예제 #3
0
 def __init__(self, start_cursor=None):
     super(UserSearch, self).__init__(PswdUser.query_by_creation(), 1,
                                      start_cursor, 0, False, False)