1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
from pytest import raises
from sqlalchemy_utils import get_bind
from tests import TestCase
class TestGetBind(TestCase):
def test_with_session(self):
assert get_bind(self.session) == self.connection
def test_with_connection(self):
assert get_bind(self.connection) == self.connection
def test_with_model_object(self):
article = self.Article()
self.session.add(article)
assert get_bind(article) == self.connection
def test_with_unknown_type(self):
with raises(TypeError):
get_bind(None)
|
