From 16c905b5f2a24ee7bdb766f9b60ce68e44b7dab0 Mon Sep 17 00:00:00 2001
From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Sun, 6 Jan 2019 13:06:56 -0500
Subject: [PATCH] Fix expect_warnings_on function

This function is not used in Alembic right now but it contained
undefined symbols, re-copy over from SQLAlchemy.

Change-Id: I3eb6eb76dc1d0d83606fa2a4af88300ee802d3bf
---
 alembic/testing/assertions.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/alembic/testing/assertions.py b/alembic/testing/assertions.py
index 4fb43a4f..2c7382c2 100644
--- a/alembic/testing/assertions.py
+++ b/alembic/testing/assertions.py
@@ -10,6 +10,8 @@ from sqlalchemy.util import decorator
 from sqlalchemy import exc as sa_exc
 import warnings
 from . import mock
+from .exclusions import db_spec
+from . import config
 
 
 if not util.sqla_094:
@@ -127,8 +129,6 @@ def expect_warnings_on(db, *messages, **kw):
 
     if isinstance(db, util.string_types) and not spec(config._current):
         yield
-    elif not _is_excluded(*db):
-        yield
     else:
         with expect_warnings(*messages, **kw):
             yield
-- 
2.47.2