File: test_connect.py

package info (click to toggle)
pypuppetdb 3.2.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 508 kB
  • sloc: python: 3,810; makefile: 127; sh: 9
file content (20 lines) | stat: -rw-r--r-- 497 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
 
from unittest import mock

import pypuppetdb


@mock.patch("pypuppetdb.api.base.requests.Session.close")
def test_connect_api(session_close):
    puppetdb = pypuppetdb.connect()
    assert puppetdb.version == "v4"

    puppetdb.disconnect()
    assert session_close.called


@mock.patch("pypuppetdb.api.base.requests.Session.close")
def test_connect_with_statement(session_close):
    with pypuppetdb.connect() as puppetdb:
        assert puppetdb.version == "v4"

    assert session_close.called