def test_06_get(self):
"""Test06 GetURLpyCurl().get() generic get."""
geturl = GetURLpyCurl()
geturl._geturl = type('test', (object, ), {})()
geturl.get = self._get
status = geturl.get("http://host")
self.assertEqual(status, "http://host")
def test_02_is_available(self, mock_pycurl):
"""Test02 GetURLpyCurl().is_available()."""
mock_pycurl.return_value = True
geturl = GetURLpyCurl()
geturl.is_available()
self.assertTrue(geturl.is_available())
self.assertTrue(mock_pycurl.called)
def test_02_is_available(self, mock_gupycurl, mock_msg):
"""Test02 GetURLpyCurl()._is_available()."""
mock_msg.level = 0
mock_gupycurl.return_value = True
geturl = GetURLpyCurl()
geturl.is_available()
self.assertTrue(geturl.is_available())
mock_gupycurl.return_value = False
geturl = GetURLpyCurl()
self.assertFalse(geturl.is_available())
def test_05__mkpycurl(self, mock_jdump):
"""Test05 GetURLpyCurl()._mkpycurl()."""
curl_patch = patch("udocker.utils.curl.CurlHeader")
curlhdr = curl_patch.start()
mock_curlhdr = Mock()
curlhdr.return_value = mock_curlhdr
pyc_patch = patch("udocker.utils.curl.pycurl.Curl")
pycurl = pyc_patch.start()
mock_pycurl = Mock()
pycurl.return_value = mock_pycurl
buff = strio()
argl = ["http://host"]
geturl = GetURLpyCurl()
status = geturl._mkpycurl(pycurl, curlhdr, buff, argl)
self.assertTrue(pycurl.setopt.called)
self.assertEqual(status, ("", None))
kargl = {
"post": "pst1",
"sizeonly": True,
"proxy": "http://proxy",
"ctimeout": 1000,
"header": "Authorization: Bearer",
"v": True,
"nobody": True,
"timeout": 50,
}
mock_jdump.return_value = {"post": "pst1"}
geturl = GetURLpyCurl()
status = geturl._mkpycurl(pycurl, curlhdr, buff, argl, kargl)
self.assertTrue(pycurl.setopt.called)
self.assertTrue(curlhdr.sizeonly)
self.assertEqual(status, ("", None))
curlhdr = curl_patch.stop()
pycurl = pyc_patch.stop()
def test_04__set_defaults(self, mock_hdr, mock_pyc, mock_msg,
mock_selinsec):
"""Test04 GetURLpyCurl()._set_defaults()."""
mock_selinsec.return_value = True
mock_msg.level = 0
mock_msg.VER = 4
geturl = GetURLpyCurl()
geturl._set_defaults(mock_pyc, mock_hdr)
self.assertTrue(mock_pyc.setopt.called)
# when Msg.level >= Msg.VER = 4: AND insecure
mock_msg.level = 5
mock_msg.VER = 4
geturl = GetURLpyCurl()
geturl._set_defaults(mock_pyc, mock_hdr)
self.assertEqual(mock_pyc.setopt.call_count, 18)
mock_selinsec.return_value = True
# when Msg.level < Msg.VER = 4: AND secure
mock_msg.level = 2
mock_msg.VER = 4
geturl = GetURLpyCurl()
geturl._set_defaults(mock_pyc, mock_hdr)
self.assertEqual(mock_pyc.setopt.call_count, 27)
def test_01_init(self):
"""Test01 GetURLpyCurl() constructor."""
geturl = GetURLpyCurl()
self.assertEqual(geturl._url, None)