def test_get_rest(transport: str = "rest", request_type=compute.GetMachineTypeRequest):
    client = MachineTypesClient(
        credentials=ga_credentials.AnonymousCredentials(), transport=transport,
    )

    # Everything is optional in proto3 as far as the runtime is concerned,
    # and we are mocking out the actual API, so just send an empty request.
    request = request_type()

    # Mock the http request call within the method and fake a response.
    with mock.patch.object(Session, "request") as req:
        # Designate an appropriate value for the returned response.
        return_value = compute.MachineType(
            accelerators=[compute.Accelerators(guest_accelerator_count=2452)],
            creation_timestamp="creation_timestamp_value",
            deprecated=compute.DeprecationStatus(deleted="deleted_value"),
            description="description_value",
            guest_cpus=1090,
            id="id_value",
            image_space_gb=1430,
            is_shared_cpu=True,
            kind="kind_value",
            maximum_persistent_disks=2603,
            maximum_persistent_disks_size_gb="maximum_persistent_disks_size_gb_value",
            memory_mb=967,
            name="name_value",
            scratch_disks=[compute.ScratchDisks(disk_gb=723)],
            self_link="self_link_value",
            zone="zone_value",
        )

        # Wrap the value into a proper Response obj
        json_return_value = compute.MachineType.to_json(return_value)
        response_value = Response()
        response_value.status_code = 200
        response_value._content = json_return_value.encode("UTF-8")
        req.return_value = response_value
        response = client.get(request)

    # Establish that the response is the type that we expect.
    assert isinstance(response, compute.MachineType)
    assert response.accelerators == [compute.Accelerators(guest_accelerator_count=2452)]
    assert response.creation_timestamp == "creation_timestamp_value"
    assert response.deprecated == compute.DeprecationStatus(deleted="deleted_value")
    assert response.description == "description_value"
    assert response.guest_cpus == 1090
    assert response.id == "id_value"
    assert response.image_space_gb == 1430
    assert response.is_shared_cpu is True
    assert response.kind == "kind_value"
    assert response.maximum_persistent_disks == 2603
    assert (
        response.maximum_persistent_disks_size_gb
        == "maximum_persistent_disks_size_gb_value"
    )
    assert response.memory_mb == 967
    assert response.name == "name_value"
    assert response.scratch_disks == [compute.ScratchDisks(disk_gb=723)]
    assert response.self_link == "self_link_value"
    assert response.zone == "zone_value"
Exemplo n.º 2
0
def test_get_rest(transport: str = "rest",
                  request_type=compute.GetNodeTypeRequest):
    client = NodeTypesClient(
        credentials=ga_credentials.AnonymousCredentials(),
        transport=transport,
    )

    # Everything is optional in proto3 as far as the runtime is concerned,
    # and we are mocking out the actual API, so just send an empty request.
    request = request_type()

    # Mock the http request call within the method and fake a response.
    with mock.patch.object(Session, "request") as req:
        # Designate an appropriate value for the returned response.
        return_value = compute.NodeType(
            cpu_platform="cpu_platform_value",
            creation_timestamp="creation_timestamp_value",
            deprecated=compute.DeprecationStatus(deleted="deleted_value"),
            description="description_value",
            guest_cpus=1090,
            id="id_value",
            kind="kind_value",
            local_ssd_gb=1244,
            memory_mb=967,
            name="name_value",
            self_link="self_link_value",
            zone="zone_value",
        )

        # Wrap the value into a proper Response obj
        json_return_value = compute.NodeType.to_json(return_value)
        response_value = Response()
        response_value.status_code = 200
        response_value._content = json_return_value.encode("UTF-8")
        req.return_value = response_value
        response = client.get(request)

    # Establish that the response is the type that we expect.
    assert isinstance(response, compute.NodeType)
    assert response.cpu_platform == "cpu_platform_value"
    assert response.creation_timestamp == "creation_timestamp_value"
    assert response.deprecated == compute.DeprecationStatus(
        deleted="deleted_value")
    assert response.description == "description_value"
    assert response.guest_cpus == 1090
    assert response.id == "id_value"
    assert response.kind == "kind_value"
    assert response.local_ssd_gb == 1244
    assert response.memory_mb == 967
    assert response.name == "name_value"
    assert response.self_link == "self_link_value"
    assert response.zone == "zone_value"
Exemplo n.º 3
0
def test_get_rest(transport: str = "rest",
                  request_type=compute.GetRegionRequest):
    client = RegionsClient(
        credentials=ga_credentials.AnonymousCredentials(),
        transport=transport,
    )

    # Everything is optional in proto3 as far as the runtime is concerned,
    # and we are mocking out the actual API, so just send an empty request.
    request = request_type()

    # Mock the http request call within the method and fake a response.
    with mock.patch.object(Session, "request") as req:
        # Designate an appropriate value for the returned response.
        return_value = compute.Region(
            creation_timestamp="creation_timestamp_value",
            deprecated=compute.DeprecationStatus(deleted="deleted_value"),
            description="description_value",
            id="id_value",
            kind="kind_value",
            name="name_value",
            quotas=[compute.Quota(limit=0.543)],
            self_link="self_link_value",
            status=compute.Region.Status.DOWN,
            supports_pzs=True,
            zones=["zones_value"],
        )

        # Wrap the value into a proper Response obj
        json_return_value = compute.Region.to_json(return_value)
        response_value = Response()
        response_value.status_code = 200
        response_value._content = json_return_value.encode("UTF-8")
        req.return_value = response_value
        response = client.get(request)

    # Establish that the response is the type that we expect.
    assert isinstance(response, compute.Region)
    assert response.creation_timestamp == "creation_timestamp_value"
    assert response.deprecated == compute.DeprecationStatus(
        deleted="deleted_value")
    assert response.description == "description_value"
    assert response.id == "id_value"
    assert response.kind == "kind_value"
    assert response.name == "name_value"
    assert response.quotas == [compute.Quota(limit=0.543)]
    assert response.self_link == "self_link_value"
    assert response.status == compute.Region.Status.DOWN
    assert response.supports_pzs is True
    assert response.zones == ["zones_value"]