def serviceclient(self, mocker, fixture_ghcs, fixture_sas, request): service_client = mocker.patch(path_service_client) service_client.scenario = request.param if request.param == "v2": service_client.side_effect = [ build_mock_response( mocker, 200, generate_job_show( job_status=JobStatusType.running.value, job_version=request.param, ), ) ] elif request.param == "v1": service_client.side_effect = [ build_mock_response( mocker, 200, generate_job_status( job_status=JobStatusType.unknown.value), ), build_mock_response( mocker, 200, generate_job_show( job_status=JobStatusType.running.value, job_type=JobType.exportDevices.value, job_version=request.param, ), ), ] return service_client
def serviceclient(self, mocker, fixture_ghcs, fixture_sas, request): service_client = mocker.patch(path_service_client) service_client.side_effect = [ build_mock_response(mocker), build_mock_response(mocker, payload=[]), ] return service_client
def serviceclient_error(self, mocker, fixture_ghcs, request): service_client = mocker.patch(path_service_client) output = generate_device_interfaces_payload() test_side_effect = [ build_mock_response(mocker, request.param[0], payload=output), build_mock_response(mocker, request.param[1], {}), ] service_client.side_effect = test_side_effect return service_client
def serviceclient(self, mocker, fixture_ghcs, request): service_client = mocker.patch(path_service_client) output = generate_device_interfaces_payload() payload_list = generate_pnp_interface_list_payload() payload_show = generate_pnp_interface_show_payload() test_side_effect = [ build_mock_response(mocker, request.param, payload=output), build_mock_response(mocker, request.param, payload=payload_list), build_mock_response(mocker, request.param, payload=payload_show), ] service_client.side_effect = test_side_effect return service_client
def serviceclient_test_wait(self, mocker, fixture_ghcs, fixture_sas, request): service_client = mocker.patch(path_service_client) test_side_effect = [ # Job Queued build_mock_response( mocker, request.param[0], { "jobId": "myjob", "status": JobStatusType.queued.value, "type": JobType.scheduleUpdateTwin.value, }, ), build_mock_response( mocker, request.param[0], { "jobId": "myjob", "status": JobStatusType.running.value, "type": JobType.scheduleUpdateTwin.value, }, ), build_mock_response( mocker, request.param[0], { "createdTime": "2020-01-01T01:06:38.2649798Z", "deviceJobStatistics": { "deviceCount": 6, "failedCount": 0, "pendingCount": 0, "runningCount": 0, "succeededCount": 6, }, "endTime": "2020-01-01T01:06:42.4993645Z", "jobId": "myjob", "maxExecutionTimeInSeconds": 900, "queryCondition": "*", "startTime": "2020-01-01T01:06:38.2649798Z", "status": request.param[1], "type": "scheduleUpdateTwin", "updateTwin": { "deviceId": None, "etag": "*", "tags": { "deviceClass": "Class1, Class2" }, }, }, ), ] service_client.side_effect = test_side_effect
def serviceclient(self, mocker, fixture_ghcs, fixture_sas, request): service_client = mocker.patch(path_service_client) side_effect = [ build_mock_response(mocker, request.param), ] service_client.side_effect = side_effect return service_client
def handle_calls(*args, **kwargs): parsed_url = urlparse(args[0].url) parsed_qs = parse_qs(parsed_url.query) result_set = [] if "/jobs?" in args[0].url: payload = request.param["v1"] for job in payload: result_set.append( generate_job_show( job_status=job["status"], job_type=job["type"], job_version="v1", )) return build_mock_response(mocker, 200, result_set) if "/jobs/v2/query?" in args[0].url: payload = request.param["v2"] status_request = None type_request = None if "jobType" in parsed_qs: type_request = parsed_qs["jobType"][0] if "jobStatus" in parsed_qs: status_request = parsed_qs["jobStatus"][0] for job in payload: if type_request: if job["type"] != type_request: continue if status_request: if job["status"] != status_request: continue result_set.append( generate_job_show(job_status=job["status"], job_type=job["type"])) return build_mock_response(mocker, 200, result_set, {"x-ms-continuation": None})
def serviceclient(self, mocker, fixture_ghcs, request): service_client = mocker.patch(path_service_client) service_client.return_value = build_mock_response( mocker, request.param, {}) return service_client
def serviceclient(self, mocker, request): service_client = mocker.patch(path_service_client) output = generate_device_interfaces_payload() service_client.return_value = build_mock_response( mocker, request.param, output) return service_client
def serviceclient(self, mocker, fixture_ghcs, fixture_sas, request, sample_config_show): service_client = mocker.patch(path_service_client) service_client.return_value = build_mock_response( mocker, request.param, sample_config_show) return service_client