Click testing with patch

cli_test.py

import click.testing
import os
import snowtool.cli
import unittest
from unittest import mock

from ..mocks import MOCK_ENV, MOCK_SNOWFLAKE_VERSION, MockSnowflakeServer


@mock.patch.dict(os.environ, MOCK_ENV)
@mock.patch("snowflake.connector.connect")
class CliTest(unittest.TestCase):
    def setUp(self):
        self.runner = click.testing.CliRunner()

    def test_version_command(self, connect):
        connect.return_value = MockSnowflakeServer()
        result = self.runner.invoke(snowtool.cli.main, ["version"])
        self.assertEqual(result.exit_code, 0)
        self.assertEqual(
            result.output, "Snowflake Version: {}\n".format(MOCK_SNOWFLAKE_VERSION)
        )