pymssql/tests/test_memory.py

# -*- coding: utf-8 -*-

import gc
import sys
import psutil
import pymssql
import pytest


@pytest.mark.slow
@pytest.mark.xfail(reason="Memory test is not stable")
def test_memory_leak_on_unsuccessful_connect():
    """
    This test checks python process memory usage and not just unsuccessful
    connect path. Many other factors (i.e. garbage collection) affect memory
    usage pattern. So this is an indirect test which is somewhat flaky and
    because of that is marked 'xfail'.
    """
    p = psutil.Process()
    m0 = p.memory_full_info()
    for i in range(10):
        gc.collect()
        try:
            pymssql.connect(server="www.google.com", port=81, user='username',
                            password='password', login_timeout=1)
        except:
            pass
        gc.collect()
        m1 = p.memory_full_info()
        duss = m1.uss - m0.uss
        print(i, "uss=", m1.uss, "duss:", duss)
        if i > 5:
            assert duss <= 0
        m0 = m1


if __name__ == '__main__':

    test_memory_leak_on_unsuccessful_connect()