UPD: More and better tests.
This commit is contained in:
@@ -1,7 +1,12 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from __future__ import unicode_literals
|
||||
from django.apps import apps
|
||||
from django.contrib.auth import get_user_model
|
||||
from django.contrib.auth.models import Group
|
||||
|
||||
from ..models.event import Event
|
||||
from ..models.eventstatus import EventStatus
|
||||
|
||||
|
||||
class RoleMixin(object):
|
||||
def create_user_for_role(self, role_name, password, first_name, last_name):
|
||||
@@ -20,3 +25,21 @@ class RoleMixin(object):
|
||||
setattr(app_config.settings, setting_name, [role_name])
|
||||
|
||||
return user
|
||||
|
||||
|
||||
class EventMixin(object):
|
||||
def get_status_label(self, status_code):
|
||||
return EventStatus.objects.get(code=status_code).label
|
||||
|
||||
def create_event(self, data):
|
||||
event = Event(**data)
|
||||
event.save()
|
||||
return event
|
||||
|
||||
def submit_event(self, event):
|
||||
event.workflow.update_status('submitted', event.owner)
|
||||
return event
|
||||
|
||||
def accept_event(self, event, user):
|
||||
event.workflow.update_status('accepted', user)
|
||||
return event
|
||||
|
||||
Reference in New Issue
Block a user