summaryrefslogtreecommitdiff
path: root/contrib/python/podman/test/test_client.py
blob: 3fc6d39dc27e0e6cf941ccc1651c07f937d99317 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
from __future__ import absolute_import

import unittest
from unittest.mock import patch

import podman
from podman.client import BaseClient, Client, LocalClient, RemoteClient


class TestClient(unittest.TestCase):
    def setUp(self):
        pass

    @patch('podman.libs.system.System.ping', return_value=True)
    def test_local(self, mock_ping):
        p = Client(
            uri='unix:/run/podman',
            interface='io.podman',
        )

        self.assertIsInstance(p._client, LocalClient)
        self.assertIsInstance(p._client, BaseClient)

        mock_ping.assert_called_once_with()

    @patch('podman.libs.system.System.ping', return_value=True)
    def test_remote(self, mock_ping):
        p = Client(
            uri='unix:/run/podman',
            interface='io.podman',
            remote_uri='ssh://user@hostname/run/podman/podman',
            identity_file='~/.ssh/id_rsa')

        self.assertIsInstance(p._client, BaseClient)
        mock_ping.assert_called_once_with()