Skip to content

Commit 7e3b4c6

Browse files
committed
Add unit test for env_management_callback func
1 parent 04ab998 commit 7e3b4c6

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

tests/test_main.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
import httpretty
99
import pytest
10+
import click
1011
from click.testing import CliRunner
1112

1213
from rsconnect.json_web_token import SECRET_KEY_ENV
@@ -22,7 +23,7 @@
2223
require_connect,
2324
has_jwt_structure,
2425
)
25-
from rsconnect.main import cli
26+
from rsconnect.main import cli, env_management_callback
2627
from rsconnect import VERSION
2728

2829

@@ -746,6 +747,22 @@ def test_add_shinyapps_missing_options(self):
746747
os.environ["CONNECT_SERVER"] = original_server_value
747748

748749

750+
def test_env_management_callback(self):
751+
ctx = click.Context(cli)
752+
753+
# env_management is always False when --disable-env-management is True
754+
ctx.params = {'disable_env_management': True}
755+
assert env_management_callback(ctx, None, None) is False
756+
assert env_management_callback(ctx, None, True) is False
757+
assert env_management_callback(ctx, None, False) is False
758+
759+
# (env_management == not value) when --disable-env-management is None
760+
ctx.params = {'disable_env_management': None}
761+
assert env_management_callback(ctx, None, None) is None
762+
assert env_management_callback(ctx, None, True) is False
763+
assert env_management_callback(ctx, None, False) is True
764+
765+
749766
class TestBootstrap(TestCase):
750767
def setUp(self):
751768
self.mock_server = "http://localhost:8080"

0 commit comments

Comments
 (0)