class user_dao_test:
def setup(self):
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.user_dao = UserDAO()
self.created_users = 0
def teardown(self):
self.testbed.deactivate()
def generate_custom_email(self):
custom_email = 'myemail'
custom_email += str(self.created_users)
self.created_users += 1
custom_email += '@gmail.com'
return custom_email
def insert_user_with_status(self, status):
google_user = users.User(email=self.generate_custom_email())
self.user_dao.insert(google_user)
my_user = self.user_dao.load(google_user)
my_user.status = status
my_user.put()
return my_user
def should_insert_an_user_as_admin(self):
google_user = users.User(email='*****@*****.**')
self.user_dao.insert_admin(google_user)
loaded_user = self.user_dao.load(google_user)
assert loaded_user.status == 'ADMIN'
def should_verify_if_there_is_an_user(self):
assert self.user_dao.has_no_user() == True
new_user = users.User(email='*****@*****.**')
self.user_dao.insert_admin(new_user)
assert self.user_dao.has_no_user() == False
def should_retrieve_users_pending_approval(self):
self.insert_user_with_status('ADMIN')
self.insert_user_with_status('APPROVED')
self.insert_user_with_status('DECLINED')
pending_approval = self.insert_user_with_status('PENDING')
users_pending_approval = [u.user.email() for u in self.user_dao.list_pending_approval()]
assert pending_approval.user.email() in users_pending_approval
def should_remove_user(self):
new_user = users.User(email='*****@*****.**')
self.user_dao.insert(google_user=new_user)
loaded_user = self.user_dao.load(new_user)
assert loaded_user is not None
self.user_dao.remove(new_user)
loaded_user = self.user_dao.load(new_user)
assert loaded_user is None
def should_get_admin(self):
admin = users.User(email='*****@*****.**')
self.user_dao.insert_admin(admin)
assert self.user_dao.get_admin().user == admin
def should_get_or_create_a_user(self):
new_user = users.User(email='*****@*****.**')
assert self.user_dao.load(new_user) is None
assert self.user_dao.get_or_create(new_user) is not None
assert self.user_dao.load(new_user) is not None
def should_approve_an_user(self):
google_to_be_approved = users.User(email='*****@*****.**')
self.user_dao.insert(google_to_be_approved)
to_be_approved = self.user_dao.load(google_to_be_approved)
assert to_be_approved.status == 'PENDING'
self.user_dao.approve(str(to_be_approved.key()))
after_approval = self.user_dao.load(google_to_be_approved)
assert after_approval.status == 'APPROVED'
def should_load_an_user_by_key(self):
google_user = users.User(email='*****@*****.**')
self.user_dao.insert(google_user)
user = self.user_dao.load(google_user)
assert self.user_dao.get_by_key(str(user.key())).user == google_user
def setup(self):
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.user_dao = UserDAO()
self.created_users = 0
def _test(self):
pass
user = UserDAO()
users = user.get_all_users()
print(users)