From 98e51079a35b8853ccbf9fbce725e8ad9936cad0 Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Mon, 29 Apr 2019 20:42:42 -0400 Subject: [PATCH] Update TestModelTestMixin to work with Python 3 Signed-off-by: Roberto Rosario --- mayan/apps/common/tests/mixins.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/mayan/apps/common/tests/mixins.py b/mayan/apps/common/tests/mixins.py index af57a41ef2..03aa0260da 100644 --- a/mayan/apps/common/tests/mixins.py +++ b/mayan/apps/common/tests/mixins.py @@ -20,6 +20,7 @@ from django.template import Context, Template from django.test.utils import ContextList from django.urls import clear_url_caches, reverse from django.utils.encoding import force_bytes +from django.utils.six import PY3 from mayan.apps.storage.settings import setting_temporary_directory @@ -323,9 +324,14 @@ class TestModelTestMixin(object): # but from another test module. apps.all_models[app_label].pop(model_name.lower(), None) - TestModel = type( - force_bytes(model_name), (models.Model,), attrs - ) + if PY3: + TestModel = type( + model_name, (models.Model,), attrs + ) + else: + TestModel = type( + force_bytes(model_name), (models.Model,), attrs + ) setattr(self, model_name, TestModel)