Merge pull request #43 from damklis/develop

Add test for context manager

Author: damklis

.coveragerc

@@ -7,7 +7,5 @@ omit =
 exclude_lines =
  pragma: no cover
  def __repr__
- def __enter__
- def __exit__
  raise NotImplementedError
  if __name__ == .__main__.:

airflow/modules/tests/proxypool/test_redis_proxypool_client.py

@@ -63,3 +63,11 @@ def test_get_proxy(redis, redis_config, redis_mock, proxies):
  result = redis_client.get_proxy()
 
  assert result == expected
+
+
+@patch("proxypool.redis_proxypool_client.redis.StrictRedis")
+def test_redis_client_context_manager(redis, redis_config):
+ key = "test"
+
+ with RedisProxyPoolClient(key, redis_config) as redis_client:
+ pass