def _badge_adjustments(self):
from uber.badge_funcs import needs_badge_num
if self.badge_type == c.PSEUDO_DEALER_BADGE:
self.ribbon = add_opt(self.ribbon_ints, c.DEALER_RIBBON)
self.badge_type = self.badge_type_real
old_type = self.orig_value_of('badge_type')
old_num = self.orig_value_of('badge_num')
if not needs_badge_num(self):
self.badge_num = None
if old_type != self.badge_type or old_num != self.badge_num:
self.session.update_badge(self, old_type, old_num)
elif needs_badge_num(self) and not self.badge_num:
self.badge_num = self.session.get_next_badge_num(self.badge_type)
def test_preassigned_invalid_status(self, session):
session.staff_one.badge_status = c.INVALID_STATUS
assert not needs_badge_num(attendee=session.staff_one)
def test_preassigned_by_type(self, session):
assert needs_badge_num(badge_type=c.STAFF_BADGE)
def test_preassigned_unassigned(self, session):
session.staff_one.first_name = ''
assert not needs_badge_num(attendee=session.staff_one)
def test_preassigned_not_paid(self, session):
session.staff_one.paid = c.NOT_PAID
assert not needs_badge_num(attendee=session.staff_one)
def test_non_preassigned_by_type(self, session):
assert not needs_badge_num(badge_type=c.ATTENDEE_BADGE)
def test_non_preassigned_ready(self, session):
assert needs_badge_num(attendee=session.regular_attendee)
def test_non_preassigned_unassigned(self, session):
session.regular_attendee.first_name = ''
assert session.regular_attendee.badge_type not in c.PREASSIGNED_BADGE_TYPES
assert session.regular_attendee.checked_in
assert needs_badge_num(attendee=session.regular_attendee)
def test_preassigned_not_paid(self, session):
session.staff_one.paid = c.NOT_PAID
assert not needs_badge_num(attendee=session.staff_one)
def test_non_preassigned_not_checked_in(self, session):
session.regular_attendee.checked_in = None
assert session.regular_attendee.badge_type not in c.PREASSIGNED_BADGE_TYPES
assert not needs_badge_num(attendee=session.regular_attendee)
def test_preassigned_unassigned(self, session):
session.staff_one.first_name = ''
assert session.staff_one.badge_type in c.PREASSIGNED_BADGE_TYPES
assert needs_badge_num(attendee=session.staff_one)
def test_non_preassigned_ready(self, session):
assert needs_badge_num(attendee=session.regular_attendee)
def test_preassigned_ready(self, session):
assert needs_badge_num(session.staff_one)
def test_non_preassigned_by_type(self, session):
assert not needs_badge_num(badge_type=c.ATTENDEE_BADGE)
def test_numbered_badges_off(self, session, monkeypatch):
monkeypatch.setattr(c, 'NUMBERED_BADGES', False)
assert not needs_badge_num(badge_type=c.STAFF_BADGE)
assert not needs_badge_num(session.regular_attendee)
def test_non_preassigned_not_paid(self, session):
session.regular_attendee.paid = c.NOT_PAID
assert not needs_badge_num(attendee=session.regular_attendee)
def test_preassigned_by_type(self, session):
assert needs_badge_num(badge_type=c.STAFF_BADGE)
def test_preassigned_invalid_status(self, session):
session.staff_one.badge_status = c.INVALID_STATUS
assert not needs_badge_num(attendee=session.staff_one)
def test_preassigned_ready(self, session):
assert needs_badge_num(session.staff_one)
def test_non_preassigned_invalid_status(self, session):
session.regular_attendee.badge_status = c.INVALID_STATUS
assert not needs_badge_num(attendee=session.regular_attendee)
def test_non_preassigned_not_checked_in(self, session):
session.regular_attendee.checked_in = None
assert not needs_badge_num(attendee=session.regular_attendee)
def test_non_preassigned_not_checked_in(self, session):
session.regular_attendee.checked_in = None
assert session.regular_attendee.badge_type not in c.PREASSIGNED_BADGE_TYPES
assert not needs_badge_num(attendee=session.regular_attendee)
def test_non_preassigned_unassigned(self, session):
session.regular_attendee.first_name = ''
assert not needs_badge_num(attendee=session.regular_attendee)
def test_preassigned_unassigned(self, session):
session.staff_one.first_name = ''
assert session.staff_one.badge_type in c.PREASSIGNED_BADGE_TYPES
assert needs_badge_num(attendee=session.staff_one)
def test_non_preassigned_not_paid(self, session):
session.regular_attendee.paid = c.NOT_PAID
assert not needs_badge_num(attendee=session.regular_attendee)
def test_non_preassigned_unassigned(self, session):
session.regular_attendee.first_name = ''
assert session.regular_attendee.badge_type not in c.PREASSIGNED_BADGE_TYPES
assert session.regular_attendee.checked_in
assert needs_badge_num(attendee=session.regular_attendee)
def test_non_preassigned_invalid_status(self, session):
session.regular_attendee.badge_status = c.INVALID_STATUS
assert not needs_badge_num(attendee=session.regular_attendee)
def test_numbered_badges_off(self, session, monkeypatch):
monkeypatch.setattr(c, 'NUMBERED_BADGES', False)
assert not needs_badge_num(badge_type=c.STAFF_BADGE)
assert not needs_badge_num(session.regular_attendee)
