mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
migrate test_tomcat to pytest
This commit is contained in:
parent
e8d5f8394e
commit
4522bb2aba
2 changed files with 58 additions and 58 deletions
58
tests/pytests/unit/modules/test_tomcat.py
Normal file
58
tests/pytests/unit/modules/test_tomcat.py
Normal file
|
@ -0,0 +1,58 @@
|
|||
"""
|
||||
Tests cases for salt.modules.tomcat
|
||||
"""
|
||||
|
||||
|
||||
import io
|
||||
import urllib.request
|
||||
|
||||
import pytest
|
||||
|
||||
import salt.modules.tomcat as tomcat
|
||||
from tests.support.mock import MagicMock, patch
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def configure_loader_modules():
|
||||
return {tomcat: {}}
|
||||
|
||||
|
||||
def test_tomcat_wget_no_bytestring():
|
||||
responses = {
|
||||
"string": io.StringIO("Best response ever\r\nAnd you know it!"),
|
||||
"bytes": io.BytesIO(b"Best response ever\r\nAnd you know it!"),
|
||||
}
|
||||
|
||||
string_mock = MagicMock(return_value=responses["string"])
|
||||
bytes_mock = MagicMock(return_value=responses["bytes"])
|
||||
with patch(
|
||||
"salt.modules.tomcat._auth",
|
||||
MagicMock(
|
||||
return_value=urllib.request.build_opener(
|
||||
urllib.request.HTTPBasicAuthHandler(),
|
||||
urllib.request.HTTPDigestAuthHandler(),
|
||||
)
|
||||
),
|
||||
):
|
||||
with patch("urllib.request.urlopen", string_mock):
|
||||
response = tomcat._wget("tomcat.wait", url="http://localhost:8080/nofail")
|
||||
for line in response["msg"]:
|
||||
assert isinstance(line, str)
|
||||
|
||||
with patch("urllib.request.urlopen", bytes_mock):
|
||||
try:
|
||||
response = tomcat._wget(
|
||||
"tomcat.wait", url="http://localhost:8080/nofail"
|
||||
)
|
||||
except TypeError as type_error:
|
||||
if (
|
||||
type_error.args[0]
|
||||
== "startswith first arg must be bytes or a tuple of bytes,"
|
||||
" not str"
|
||||
):
|
||||
print("Got back a byte string, should've been a string")
|
||||
else:
|
||||
raise type_error
|
||||
|
||||
for line in response["msg"]:
|
||||
assert isinstance(line, str)
|
|
@ -1,58 +0,0 @@
|
|||
import io
|
||||
import urllib.request
|
||||
|
||||
import salt.modules.tomcat as tomcat
|
||||
from tests.support.mixins import LoaderModuleMockMixin
|
||||
from tests.support.mock import MagicMock, patch
|
||||
from tests.support.unit import TestCase
|
||||
|
||||
|
||||
class TomcatTestCasse(TestCase, LoaderModuleMockMixin):
|
||||
"""
|
||||
Tests cases for salt.modules.tomcat
|
||||
"""
|
||||
|
||||
def setup_loader_modules(self):
|
||||
return {tomcat: {}}
|
||||
|
||||
def test_tomcat_wget_no_bytestring(self):
|
||||
responses = {
|
||||
"string": io.StringIO("Best response ever\r\nAnd you know it!"),
|
||||
"bytes": io.BytesIO(b"Best response ever\r\nAnd you know it!"),
|
||||
}
|
||||
|
||||
string_mock = MagicMock(return_value=responses["string"])
|
||||
bytes_mock = MagicMock(return_value=responses["bytes"])
|
||||
with patch(
|
||||
"salt.modules.tomcat._auth",
|
||||
MagicMock(
|
||||
return_value=urllib.request.build_opener(
|
||||
urllib.request.HTTPBasicAuthHandler(),
|
||||
urllib.request.HTTPDigestAuthHandler(),
|
||||
)
|
||||
),
|
||||
):
|
||||
with patch("urllib.request.urlopen", string_mock):
|
||||
response = tomcat._wget(
|
||||
"tomcat.wait", url="http://localhost:8080/nofail"
|
||||
)
|
||||
for line in response["msg"]:
|
||||
self.assertIsInstance(line, str)
|
||||
|
||||
with patch("urllib.request.urlopen", bytes_mock):
|
||||
try:
|
||||
response = tomcat._wget(
|
||||
"tomcat.wait", url="http://localhost:8080/nofail"
|
||||
)
|
||||
except TypeError as type_error:
|
||||
if (
|
||||
type_error.args[0]
|
||||
== "startswith first arg must be bytes or a tuple of bytes,"
|
||||
" not str"
|
||||
):
|
||||
self.fail("Got back a byte string, should've been a string")
|
||||
else:
|
||||
raise type_error
|
||||
|
||||
for line in response["msg"]:
|
||||
self.assertIsInstance(line, str)
|
Loading…
Add table
Reference in a new issue