def test_email_users(self, send_mass_mail): data = {'title': 'yeah', 'content': 'message'} form = UserQueryForm(data) user_1 = User(email='*****@*****.**') user_2 = User(email='*****@*****.**') user_3 = User(email='*****@*****.**') view = EmailInterface() view.users_result = [user_1, user_2, user_3] view.email_users(form) admin_emails = [] for admin in settings.ADMINS: admin_emails.append(admin[1]) messages = [(data['title'], data['content'], '*****@*****.**', [user_1.email]), (data['title'], data['content'], '*****@*****.**', [user_2.email]), (data['title'], data['content'], '*****@*****.**', [user_3.email]), (data['title'] + ' (sent from admin panel)', data['content'], '*****@*****.**', admin_emails)] send_mass_mail.assert_called_with(messages)
def test_admin_staff_status(self): user = User() self.assertFalse(user.is_staff) user.is_admin = True self.assertTrue(user.is_staff)
def test_properties(self): user = User() user.email = "*****@*****.**" user.name = "dude" self.assertEquals(user.get_full_name(), "dude") self.assertEquals(user.get_short_name(), "dude") self.assertEquals(unicode(user), user.email)
def add_user(): form = NewUserForm() if form.validate_on_submit(): user = User(form.name.data, form.email.data, int(form.key_id.data)) db.session.add(user) db.session.commit() flash('%s (%s) added' % (user.name, user.email), 'success') return redirect(url_for('users')) return render_template('new_user.html', form=form)
def test_email_error(self, mass_mail): data = {'title': 'yeah', 'content': 'message'} form = UserQueryForm(data) user_1 = User(email='*****@*****.**') view = EmailInterface() view.users_result = [user_1] mass_mail.side_effect = SMTPException('fail') view.email_users(form) self.assertEquals(form.errors['__all__'], 'error sending email: fail')
#!/usr/bin/env python import argparse from access import db from access.models import User parser = argparse.ArgumentParser() parser.add_argument("name", help="Name of member") parser.add_argument("email", help="Email address of member") parser.add_argument("key_id", help="RFID key code") args = parser.parse_args() print "Adding user %s <%s> = %d" % (args.name, args.email, int(args.key_id)) user = User(args.name, args.email, int(args.key_id)) db.session.add(user) db.session.commit()