diff --git a/tests/testapp/test_dynamicfield.py b/tests/testapp/test_dynamicfield.py index 0b671ebde..f9657bce9 100644 --- a/tests/testapp/test_dynamicfield.py +++ b/tests/testapp/test_dynamicfield.py @@ -2,6 +2,7 @@ import datetime as dt import json +from typing import cast from unittest import SkipTest, mock import mariadb_dyncol @@ -9,6 +10,7 @@ from django.core import serializers from django.core.exceptions import FieldError from django.db import connection, connections, models +from django.db.backends.mysql.base import DatabaseWrapper from django.db.migrations.writer import MigrationWriter from django.db.models import CharField, Transform from django.test import TestCase @@ -22,7 +24,7 @@ class DynColTestCase(TestCase): @classmethod def setUpClass(cls): - if not connection.mysql_is_mariadb: + if not cast(DatabaseWrapper, connection).mysql_is_mariadb: raise SkipTest("Dynamic Columns require MariaDB") super().setUpClass()