Method test_session_close_resets_only

test/orm/test_session.py:1682–1719

(self, close_resets_only, method)

    @testing.combinations(True, False, "default", argnames="close_resets_only")
    @testing.variation("method", ["close", "reset"])
    def test_session_close_resets_only(self, close_resets_only, method):
        users, User = self.tables.users, self.classes.User
        self.mapper_registry.map_imperatively(User, users)

        if close_resets_only is True:
            kw = {"close_resets_only": True}
        elif close_resets_only is False:
            kw = {"close_resets_only": False}
        else:
            eq_(close_resets_only, "default")
            kw = {}

        s = fixture_session(**kw)
        u1 = User()
        s.add(u1)
        assertions.in_(u1, s)

        if method.reset:
            s.reset()
        elif method.close:
            s.close()
        else:
            method.fail()

        assertions.not_in(u1, s)

        u2 = User()
        if method.close and close_resets_only is False:
            with expect_raises_message(
                exc.InvalidRequestError,
                "This Session has been permanently closed and is unable "
                "to handle any more transaction requests.",
            ):
                s.add(u2)
            assertions.not_in(u2, s)
        else:
            s.add(u2)
            assertions.in_(u2, s)


class SessionSchemaTranslateTest(