This commit is contained in:
@@ -11,6 +11,19 @@ import controller
|
||||
|
||||
|
||||
class ControllerTests(unittest.TestCase):
|
||||
def test_get_requests_encode_parameters_in_the_query_string(self):
|
||||
response = io.BytesIO(json.dumps({"code": "Success", "data": "stopped"}).encode())
|
||||
client = controller.AutoDLClient("test-token")
|
||||
with mock.patch.object(
|
||||
controller.urllib.request, "urlopen", return_value=response
|
||||
) as urlopen:
|
||||
self.assertEqual(client.status("pro-test"), "stopped")
|
||||
|
||||
request = urlopen.call_args.args[0]
|
||||
self.assertEqual(request.method, "GET")
|
||||
self.assertIsNone(request.data)
|
||||
self.assertTrue(request.full_url.endswith("?instance_uuid=pro-test"))
|
||||
|
||||
def test_runner_download_uses_explicit_user_agent(self):
|
||||
payload = b"test-runner"
|
||||
with tempfile.TemporaryDirectory() as directory:
|
||||
|
||||
Reference in New Issue
Block a user