rattail-project/rattail Files · tests/db/test

Files @ 22a589773921

Branch filter:

Location: rattail-project/rattail/tests/db/test_core.py

22a589773921 674 B text/x-python Show Annotation Show as Raw Download as Raw

lance

Removed global `Session` from `rattail.db`.

A Session class may now be had via `get_session_class()`.


from unittest import TestCase

from rattail.db import core
from sqlalchemy import Column


class TestCore(TestCase):

    def test_uuid_column(self):
        column = core.uuid_column()
        self.assertIsInstance(column, Column)
        self.assertEqual(column.name, None)
        self.assertTrue(column.primary_key)
        self.assertFalse(column.nullable)
        self.assertIsNotNone(column.default)

    def test_uuid_column_no_default(self):
        column = core.uuid_column(default=None)
        self.assertIsNone(column.default)

    def test_uuid_column_nullable(self):
        column = core.uuid_column(nullable=True)
        self.assertTrue(column.nullable)