mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
Add some tests for client/__init__.py
This commit is contained in:
twangboy
Pedro Algarvio
parent
9f8fc27cba
commit
3b5dc02e70
1 changed files with 271 additions and 0 deletions
271
tests/pytests/unit/client/test_init.py
Normal file
271
tests/pytests/unit/client/test_init.py
Normal file
|
|
@ -0,0 +1,271 @@
|
|||
import pytest
|
||||
|
||||
import salt.client
|
||||
from salt.exceptions import SaltInvocationError
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def local_client():
|
||||
return salt.client.get_local_client()
|
||||
|
||||
|
||||
def test_get_local_client(local_client):
|
||||
assert isinstance(local_client, salt.client.LocalClient)
|
||||
|
||||
|
||||
def test_get_local_client_mopts(master_opts):
|
||||
master_opts["rest_cherrypy"] = {"port": 8000}
|
||||
local_client = salt.client.get_local_client(mopts=master_opts)
|
||||
assert isinstance(local_client, salt.client.LocalClient)
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"val, expected",
|
||||
((None, 5), (7, 7), ("9", 9), ("eleven", 5), (["13"], 5)),
|
||||
)
|
||||
def test_local_client_get_timeout(local_client, val, expected):
|
||||
assert local_client._get_timeout(timeout=val) == expected
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"val, expected",
|
||||
(
|
||||
("group1", ["L@spongebob,patrick"]),
|
||||
("group2", ["G@os:squidward"]),
|
||||
("group3", ["(", "G@os:plankton", "and", "(", "L@spongebob,patrick", ")", ")"]),
|
||||
),
|
||||
)
|
||||
def test_resolve_nodegroup(master_opts, val, expected):
|
||||
master_opts["nodegroups"] = {
|
||||
"group1": "L@spongebob,patrick",
|
||||
"group2": "G@os:squidward",
|
||||
"group3": "G@os:plankton and N@group1",
|
||||
}
|
||||
local_client = salt.client.get_local_client(mopts=master_opts)
|
||||
assert local_client._resolve_nodegroup(val) == expected
|
||||
|
||||
|
||||
def test_resolve_nodegroup_error(master_opts):
|
||||
master_opts["nodegroups"] = {
|
||||
"group1": "L@spongebob,patrick",
|
||||
"group2": "G@os:squidward",
|
||||
"group3": "G@os:plankton and N@group1",
|
||||
}
|
||||
local_client = salt.client.get_local_client(mopts=master_opts)
|
||||
with pytest.raises(SaltInvocationError):
|
||||
local_client._resolve_nodegroup("missing")
|
||||
|
||||
|
||||
def test_prep_pub(local_client):
|
||||
result = local_client._prep_pub(
|
||||
tgt="*",
|
||||
fun="test.ping",
|
||||
arg="",
|
||||
tgt_type="glob",
|
||||
ret="",
|
||||
jid="123",
|
||||
timeout=7,
|
||||
)
|
||||
expected = {
|
||||
"arg": "",
|
||||
"cmd": "publish",
|
||||
"fun": "test.ping",
|
||||
"jid": "123",
|
||||
"key": "",
|
||||
"ret": "",
|
||||
"tgt": "*",
|
||||
"tgt_type": "glob",
|
||||
"user": local_client.salt_user,
|
||||
}
|
||||
assert result == expected
|
||||
|
||||
|
||||
def test_prep_pub_kwargs(local_client):
|
||||
result = local_client._prep_pub(
|
||||
tgt="*",
|
||||
fun="test.ping",
|
||||
arg="",
|
||||
tgt_type="glob",
|
||||
ret="",
|
||||
jid="123",
|
||||
timeout=7,
|
||||
some_kwarg="spongebob",
|
||||
)
|
||||
expected = {
|
||||
"arg": "",
|
||||
"cmd": "publish",
|
||||
"fun": "test.ping",
|
||||
"jid": "123",
|
||||
"key": "",
|
||||
"ret": "",
|
||||
"tgt": "*",
|
||||
"tgt_type": "glob",
|
||||
"user": local_client.salt_user,
|
||||
"kwargs": {
|
||||
"some_kwarg": "spongebob",
|
||||
},
|
||||
}
|
||||
assert result == expected
|
||||
|
||||
|
||||
def test_prep_pub_order_masters(master_opts):
|
||||
master_opts["order_masters"] = True
|
||||
local_client = salt.client.get_local_client(mopts=master_opts)
|
||||
result = local_client._prep_pub(
|
||||
tgt="*",
|
||||
fun="test.ping",
|
||||
arg="",
|
||||
tgt_type="glob",
|
||||
ret="",
|
||||
jid="123",
|
||||
timeout=7,
|
||||
)
|
||||
expected = {
|
||||
"arg": "",
|
||||
"cmd": "publish",
|
||||
"fun": "test.ping",
|
||||
"jid": "123",
|
||||
"key": "",
|
||||
"ret": "",
|
||||
"tgt": "*",
|
||||
"tgt_type": "glob",
|
||||
"to": 7,
|
||||
"user": local_client.salt_user,
|
||||
}
|
||||
assert result == expected
|
||||
|
||||
|
||||
def test_prep_pub_nodegroup(master_opts):
|
||||
master_opts["nodegroups"] = {
|
||||
"group1": "L@spongebob,patrick",
|
||||
"group2": "G@os:squidward",
|
||||
"group3": "G@os:plankton and N@group1",
|
||||
}
|
||||
local_client = salt.client.get_local_client(mopts=master_opts)
|
||||
result = local_client._prep_pub(
|
||||
tgt="group1",
|
||||
fun="test.ping",
|
||||
arg="",
|
||||
tgt_type="nodegroup",
|
||||
ret="",
|
||||
jid="123",
|
||||
timeout=7,
|
||||
)
|
||||
expected = {
|
||||
"arg": "",
|
||||
"cmd": "publish",
|
||||
"fun": "test.ping",
|
||||
"jid": "123",
|
||||
"key": "",
|
||||
"ret": "",
|
||||
"tgt": "L@spongebob,patrick",
|
||||
"tgt_type": "compound",
|
||||
"user": local_client.salt_user,
|
||||
}
|
||||
assert result == expected
|
||||
|
||||
|
||||
def test_prep_pub_compound(local_client):
|
||||
result = local_client._prep_pub(
|
||||
tgt="spongebob,patrick",
|
||||
fun="test.ping",
|
||||
arg="",
|
||||
tgt_type="compound",
|
||||
ret="",
|
||||
jid="123",
|
||||
timeout=7,
|
||||
)
|
||||
expected = {
|
||||
"arg": "",
|
||||
"cmd": "publish",
|
||||
"fun": "test.ping",
|
||||
"jid": "123",
|
||||
"key": "",
|
||||
"ret": "",
|
||||
"tgt": "spongebob,patrick",
|
||||
"tgt_type": "compound",
|
||||
"user": local_client.salt_user,
|
||||
}
|
||||
assert result == expected
|
||||
|
||||
|
||||
def test_prep_pub_compound_nodegroup(master_opts):
|
||||
master_opts["nodegroups"] = {
|
||||
"group1": "L@spongebob,patrick",
|
||||
"group2": "G@os:squidward",
|
||||
"group3": "G@os:plankton and N@group1",
|
||||
}
|
||||
local_client = salt.client.get_local_client(mopts=master_opts)
|
||||
result = local_client._prep_pub(
|
||||
tgt="N@group1",
|
||||
fun="test.ping",
|
||||
arg="",
|
||||
tgt_type="compound",
|
||||
ret="",
|
||||
jid="123",
|
||||
timeout=7,
|
||||
)
|
||||
expected = {
|
||||
"arg": "",
|
||||
"cmd": "publish",
|
||||
"fun": "test.ping",
|
||||
"jid": "123",
|
||||
"key": "",
|
||||
"ret": "",
|
||||
"tgt": "L@spongebob,patrick",
|
||||
"tgt_type": "compound",
|
||||
"user": local_client.salt_user,
|
||||
}
|
||||
assert result == expected
|
||||
|
||||
|
||||
def test_prep_pub_ext_job_cache(master_opts):
|
||||
master_opts["ext_job_cache"] = "mysql"
|
||||
local_client = salt.client.get_local_client(mopts=master_opts)
|
||||
result = local_client._prep_pub(
|
||||
tgt="spongebob,patrick",
|
||||
fun="test.ping",
|
||||
arg="",
|
||||
tgt_type="glob",
|
||||
ret="",
|
||||
jid="123",
|
||||
timeout=7,
|
||||
)
|
||||
expected = {
|
||||
"arg": "",
|
||||
"cmd": "publish",
|
||||
"fun": "test.ping",
|
||||
"jid": "123",
|
||||
"key": "",
|
||||
"ret": "mysql",
|
||||
"tgt": "spongebob,patrick",
|
||||
"tgt_type": "glob",
|
||||
"user": local_client.salt_user,
|
||||
}
|
||||
assert result == expected
|
||||
|
||||
|
||||
def test_prep_pub_ext_job_cache_existing(master_opts):
|
||||
master_opts["ext_job_cache"] = "mysql"
|
||||
local_client = salt.client.get_local_client(mopts=master_opts)
|
||||
result = local_client._prep_pub(
|
||||
tgt="spongebob,patrick",
|
||||
fun="test.ping",
|
||||
arg="",
|
||||
tgt_type="glob",
|
||||
ret="postgres",
|
||||
jid="123",
|
||||
timeout=7,
|
||||
)
|
||||
expected = {
|
||||
"arg": "",
|
||||
"cmd": "publish",
|
||||
"fun": "test.ping",
|
||||
"jid": "123",
|
||||
"key": "",
|
||||
"ret": "postgres,mysql",
|
||||
"tgt": "spongebob,patrick",
|
||||
"tgt_type": "glob",
|
||||
"user": local_client.salt_user,
|
||||
}
|
||||
assert result == expected
|
||||
Loading…
Add table
Reference in a new issue