mirror of
https://github.com/apache/superset.git
synced 2026-04-19 08:04:53 +00:00
chore: abstract models and daos into superset-core (#35259)
This commit is contained in:
@@ -25,12 +25,15 @@ import pytest
|
||||
from sqlalchemy import Boolean, Column, Integer, String
|
||||
from sqlalchemy.exc import SQLAlchemyError
|
||||
from sqlalchemy.ext.declarative import declarative_base
|
||||
from superset_core.api.models import CoreModel
|
||||
|
||||
from superset.daos.base import BaseDAO, ColumnOperatorEnum
|
||||
from superset.daos.exceptions import DAOFindFailedError
|
||||
|
||||
|
||||
class MockModel:
|
||||
class MockModel(CoreModel):
|
||||
__abstract__ = True # Prevent SQLAlchemy from trying to create a table
|
||||
|
||||
def __init__(self, id=1, name="test"):
|
||||
self.id = id
|
||||
self.name = name
|
||||
|
||||
@@ -181,6 +181,7 @@ def test_get_uuid_entry(
|
||||
) -> None:
|
||||
from superset.daos.key_value import KeyValueDAO
|
||||
|
||||
assert key_value_entry.uuid is not None
|
||||
found_entry = KeyValueDAO.get_entry(resource=RESOURCE, key=key_value_entry.uuid)
|
||||
assert found_entry is not None
|
||||
assert JSON_CODEC.decode(found_entry.value) == JSON_VALUE
|
||||
|
||||
Reference in New Issue
Block a user