Ejemplo n.º 1
0
 async def test_subscribe_messages(self, mock_client):
     mock_client.return_value.__aenter__.return_value.subscribe.return_value = (
         AsyncIterator(
             [{"subscribeJobMessages": f"message {i+1}"} for i in range(10)]
         )
     )
     request = AsyncJobRequest(service="test")
     messages = [m async for m in request.subscribe_messages()]
     assert len(messages) == 10
     assert messages[0] == "message 1"
Ejemplo n.º 2
0
 async def test_subscribe(self, mock_client):
     mock_client.return_value.__aenter__.return_value.subscribe.return_value = (
         AsyncIterator([{"subscribeJobComplete": {"jobId": "test"}}])
     )
     request = AsyncJobRequest(service="test")
     result = await request.subscribe()
     assert result["jobId"] == "test"
Ejemplo n.º 3
0
 def test_method_chaining(self):
     request = AsyncJobRequest(service="test")
     request.add_input(value="s3://bucket/test.txt").add_output(
         value="s3://bucket/test.txt"
     ).add_parameter(value="bar").add_parameter(
         name="-p", value="baz"
     ).add_output_metadata(
         {"foo": "bar"}
     )
     assert len(request.req_args["parameters"]) == 4
     assert request.req_args["parameters"][0]["type"] == "input"
     assert request.req_args["parameters"][1]["type"] == "output"
     assert request.req_args["parameters"][2]["type"] == "parameter"
     assert request.req_args["parameters"][2]["value"] == "bar"
     assert request.req_args["parameters"][2]["type"] == "parameter"
     assert request.req_args["parameters"][3]["name"] == "-p"
     assert request.req_args["parameters"][3]["value"] == "baz"
     assert request.req_args["outputMetadata"]["foo"] == "bar"
Ejemplo n.º 4
0
 async def test_run_job_fail_return_failed(self, mock_query, mock_mutate):
     for i in [-1, -2]:
         self.query_response[i]["job"]["status"] = "failed"
         self.query_response[i]["job"]["error"] = "failed"
     mock_mutate.return_value = self.submit_response
     mock_query.side_effect = self.query_response
     request = AsyncJobRequest(service="test")
     result = await request.run_job(return_failed=True)
     assert result["error"] == "failed"
Ejemplo n.º 5
0
 async def test_run_job_fail(self, mock_query, mock_mutate):
     for i in [-1, -2]:
         self.query_response[i]["job"]["status"] = "failed"
         self.query_response[i]["job"]["error"] = "failed"
     mock_mutate.return_value = self.submit_response
     mock_query.side_effect = self.query_response
     request = AsyncJobRequest(service="test")
     with pytest.raises(RuntimeError):
         result = await request.run_job()
Ejemplo n.º 6
0
 async def test_run_job(self, mock_query, mock_mutate):
     mock_mutate.return_value = self.submit_response
     mock_query.side_effect = self.query_response
     request = AsyncJobRequest(service="test")
     result = await request.run_job()
     assert result
Ejemplo n.º 7
0
 def test_add_output_metadata_warn(self):
     request = AsyncJobRequest(service="test")
     request.job_id = "test"
     with pytest.warns(UserWarning):
         request.add_output_metadata({"foo": "bar"})
Ejemplo n.º 8
0
 def test_add_input_warn(self):
     request = AsyncJobRequest(service="test")
     with pytest.warns(UserWarning):
         request.add_input(value="foo")
Ejemplo n.º 9
0
 def test_add_parameter_warn(self):
     request = AsyncJobRequest(service="test")
     request.job_id = "test"
     with pytest.warns(UserWarning):
         request.add_parameter(value="foo")