FilesExpand file tree

 python-remove-thread-macro

/

utils.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

18 lines (17 loc) · 618 Bytes

 python-remove-thread-macro

/

utils.py

Copy path

Top

File metadata and controls

• Code
• Blame

18 lines (17 loc) · 618 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

from concurrent.futures import Future

from functools import wraps

from cassandra.cluster import Session

from mock import patch

def mock_session_pools(f):

"""

Helper decorator that allows tests to initialize :class:.`Session` objects

without actually connecting to a Cassandra cluster.

"""

@wraps(f)

def wrapper(*args, **kwargs):

with patch.object(Session, "add_or_renew_pool") as mocked_add_or_renew_pool:

future = Future()

future.set_result(object())

mocked_add_or_renew_pool.return_value = future

f(*args, **kwargs)

return wrapper