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_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_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 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()
