2015-12-07 01:22:13 +01:00
|
|
|
"""
|
|
|
|
This module provides tests for the functions in
|
|
|
|
:py:mod:`userdbs.templatetags.userdb`.
|
|
|
|
|
|
|
|
"""
|
|
|
|
from unittest import TestCase
|
|
|
|
|
|
|
|
from django.utils.translation import gettext as _
|
|
|
|
|
|
|
|
from userdbs.models import DB_TYPES
|
|
|
|
from userdbs.templatetags.userdb import db_type_icon_class, db_type_name
|
|
|
|
|
|
|
|
|
|
|
|
class UserdbTemplateTagTests(TestCase):
|
|
|
|
"""
|
|
|
|
Test suite for :py:mod:`userdbs.templatetags.userdb` functions.
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
def test_db_type_icon_class_unknown(self):
|
2023-02-18 22:46:48 +01:00
|
|
|
self.assertEqual(db_type_icon_class({"db_type": "unknown"}), "icon-database")
|
2015-12-07 01:22:13 +01:00
|
|
|
|
|
|
|
def test_db_type_icon_class_mysql(self):
|
2023-02-18 22:46:48 +01:00
|
|
|
self.assertEqual(db_type_icon_class({"db_type": DB_TYPES.mysql}), "icon-mysql")
|
2015-12-07 01:22:13 +01:00
|
|
|
|
|
|
|
def test_db_type_icon_class_pgsql(self):
|
|
|
|
self.assertEqual(
|
2023-02-18 22:46:48 +01:00
|
|
|
db_type_icon_class({"db_type": DB_TYPES.pgsql}), "icon-postgres"
|
|
|
|
)
|
2015-12-07 01:22:13 +01:00
|
|
|
|
|
|
|
def test_db_type_name_mysql(self):
|
|
|
|
self.assertEqual(
|
2023-02-18 22:46:48 +01:00
|
|
|
db_type_name({"db_type": DB_TYPES.mysql}), _(DB_TYPES[DB_TYPES.mysql])
|
|
|
|
)
|
2015-12-07 01:22:13 +01:00
|
|
|
|
|
|
|
def test_db_type_name_pgsql(self):
|
|
|
|
self.assertEqual(
|
2023-02-18 22:46:48 +01:00
|
|
|
db_type_name({"db_type": DB_TYPES.pgsql}), _(DB_TYPES[DB_TYPES.pgsql])
|
|
|
|
)
|