UPD: More and better tests.

This commit is contained in:
2019-03-29 16:56:44 +01:00
parent 80e9271209
commit 4c1c6b48df
6 changed files with 244 additions and 35 deletions

View File

@@ -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