34

SQLAlchemy でクエリをテストするにはどうすればよいですか? たとえば、これがあるとしますmodels.py

from sqlalchemy import (
        Column,
        Integer,
        String,
)
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class Panel(Base):
    __tablename__ = 'Panels'

    id = Column(Integer, primary_key=True)
    category = Column(Integer, nullable=False)
    platform = Column(String, nullable=False)
    region = Column(String, nullable=False)

    def __init__(self, category, platform, region):
        self.category = category
        self.platform = platform
        self.region = region


    def __repr__(self):
        return (
            "<Panel('{self.category}', '{self.platform}', "
            "'{self.region}')>".format(self=self)
        )

このtests.py

import unittest

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

from models import Base, Panel


class TestQuery(unittest.TestCase):

    engine = create_engine('sqlite:///:memory:')
    Session = sessionmaker(bind=engine)
    session = Session()

    def setUp(self):
        Base.metadata.create_all(self.engine)
        self.session.add(Panel(1, 'ion torrent', 'start'))
        self.session.commit()

    def tearDown(self):
        Base.metadata.drop_all(self.engine)

    def test_query_panel(self):
        expected = [Panel(1, 'ion torrent', 'start')]
        result = self.session.query(Panel).all()
        self.assertEqual(result, expected)

テストを実行しようとすると、2 つのパネルが同じように見えても失敗します。

$ nosetests
F
======================================================================
FAIL: test_query_panel (tests.TestQuery)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/clasher/tmp/tests.py", line 31, in test_query_panel
    self.assertEqual(result, expected)
AssertionError: Lists differ: [<Panel('1', 'ion torrent', 's... != [<Panel('1', 'ion torrent', 's...

First differing element 0:
<Panel('1', 'ion torrent', 'start')>
<Panel('1', 'ion torrent', 'start')>

  [<Panel('1', 'ion torrent', 'start')>, <Panel('2', 'ion torrent', 'end')>]

----------------------------------------------------------------------
Ran 1 test in 0.063s

FAILED (failures=1)

私が見つけた解決策の 1 つは、クエリで見つかると予想されるすべてのインスタンスに対してクエリを作成することです。

class TestQuery(unittest.TestCase):

    ...

    def test_query_panel(self):
        expected = [
            (1, 'ion torrent', 'start'),
            (2, 'ion torrent', 'end')
        ]
        successful = True
        # Check to make sure every expected item is in the query
        try:
            for category, platform, region in expected:
                self.session.query(Panel).filter_by(
                        category=category, platform=platform,
                        region=region).one()
        except (NoResultFound, MultipleResultsFound):
            successful = False
        self.assertTrue(successful)
        # Check to make sure no unexpected items are in the query
        self.assertEqual(self.session.query(Panel).count(),
                         len(expected))

ただし、これはかなり醜いと思います。また、テストしようとしている複雑なフィルター処理されたクエリを作成するまでには至っていません。もっと洗練された解決策はありますか? それとも、常に手動で一連の個々のクエリを作成する必要がありますか?

4

1 に答える 1

34

元のテストは正しい軌道に乗っています。次の 2 つのいずれかを行う必要がPanelあります。同じ主キー ID の 2 つのオブジェクトが次のように比較されることを確認しTrueます。

class Panel(Base):
    # ...

    def __eq__(self, other):
        return isinstance(other, Panel) and other.id == self.id

または、同じインスタンスに対してチェックしていることを確認するようにテストを編成できます (ここではID マップPanelを利用しているため)。

class TestQuery(unittest.TestCase):
    def setUp(self):
        self.engine = create_engine('sqlite:///:memory:')
        self.session = Session(engine)
        Base.metadata.create_all(self.engine)
        self.panel = Panel(1, 'ion torrent', 'start')
        self.session.add(self.panel)
        self.session.commit()

    def tearDown(self):
        Base.metadata.drop_all(self.engine)

    def test_query_panel(self):
        expected = [self.panel]
        result = self.session.query(Panel).all()
        self.assertEqual(result, expected)

エンジン/セッションのセットアップ/ティアダウンに関しては、すべてのテストに単一のエンジンを使用し、スキーマが固定されていると仮定し、すべてのテストに単一のスキーマを使用するパターンを使用します。 with は、ロールバック可能なトランザクション内で実行されます。テスト全体を明示的な 内にラップすることにより、呼び出しが実際に「実際の」トランザクションをコミットしないように、 をこのSessionように動作させることができます。https://docs.sqlalchemy.org/en/latest/orm/session_transaction.html#joining-a-session-into-an-external-transaction-such-as-for-test-suitesの例は、この使用法を示しています。すべてのテスト フィクスチャに ":memory:" エンジンを配置すると、多くのメモリが消費され、実際には SQLite 以外のデータベースにスケールアウトされません。commit()Transaction

于 2013-02-06T00:43:17.990 に答える