def delegateResult = Mock(Result)
    def inputFingerprinter = Mock(InputFingerprinter)
    def step = new IdentifyStep<>(buildOperationRunner, delegate)


    def "delegates with assigned workspace"() {
        def inputSnapshot = Mock(ValueSnapshot)
        def inputFilesFingerprint = Mock(CurrentFileCollectionFingerprint)

        when:
        def result = step.execute(work, context)

        then:

source "${KUBE_ROOT}/cluster/common.sh"
source "${KUBE_ROOT}/hack/lib/util.sh"

function detect-project() {
  if [[ -z "${MASTER_IP:-}" ]]; then
    echo "Set 'MASTER_IP' to the instance assigned to be the Kubernetes master" 1>&2
    exit 1
  fi

  if [[ -z "${PROJECT:-}" ]]; then
    echo "Set 'PROJECT' to the name of the container project: $CONTAINER_REGISTRY/$PROJECT/kubemark" >&2
    exit 1
  fi

    active:
      configMap:
        kubeletConfigKey: kubeletConfigKeyValue
        name: nameValue
        namespace: namespaceValue
        resourceVersion: resourceVersionValue
        uid: uidValue
    assigned:
      configMap:
        kubeletConfigKey: kubeletConfigKeyValue
        name: nameValue
        namespace: namespaceValue
        resourceVersion: resourceVersionValue
        uid: uidValue
    error: errorValue

    active:
      configMap:
        kubeletConfigKey: kubeletConfigKeyValue
        name: nameValue
        namespace: namespaceValue
        resourceVersion: resourceVersionValue
        uid: uidValue
    assigned:
      configMap:
        kubeletConfigKey: kubeletConfigKeyValue
        name: nameValue
        namespace: namespaceValue
        resourceVersion: resourceVersionValue
        uid: uidValue
    error: errorValue

    active:
      configMap:
        kubeletConfigKey: kubeletConfigKeyValue
        name: nameValue
        namespace: namespaceValue
        resourceVersion: resourceVersionValue
        uid: uidValue
    assigned:
      configMap:
        kubeletConfigKey: kubeletConfigKeyValue
        name: nameValue
        namespace: namespaceValue
        resourceVersion: resourceVersionValue
        uid: uidValue
    error: errorValue

  conditions:
  - lastTransitionTime: fake
    message: Resource accepted
    reason: Accepted
    status: "True"
    type: Accepted
  - lastTransitionTime: fake
    message: Resource programmed, assigned to service(s) istio-ingressgateway.istio-system.svc.domain.suffix:80
    reason: Programmed
    status: "True"
    type: Programmed
  listeners:
  - attachedRoutes: 1
    conditions:
    - lastTransitionTime: fake

    %10:2 = tf_executor.island(%7#2) wraps "tf.Const"() {device = "", dtype = "tfdtype$DT_INT32", value = dense<1> : tensor<i32>} : () -> tensor<i32> loc("while/Add/y")
    %11:2 = tf_executor.island wraps "tf.AssignAdd"(%7#0, %10#0) {device = "", T = "tfdtype$DT_INT32"} : (tensor<*xi32>, tensor<i32>) -> tensor<*xi32> loc("while/Add")

// RUN: tf-opt %s -split-input-file -tf-device-attribute-to-launch | FileCheck %s

// Tests that single TensorFlow op with device attribute is wrapped in `tf_device.launch` with the correct device assigned.
// CHECK-LABEL: func @single_op_launch
func.func @single_op_launch() {
  // CHECK: "tf_device.launch"
  // CHECK: device = "CPU:0"
  // CHECK: "tf.opA"
  // CHECK-NOT device
  // CHECK: tf_device.return
  "tf.opA"() {device = "CPU:0"} : () -> tensor<i1>

//     xla_device_type - The device type that is being targeted.
// Output:
//     Tensorflow Dialect MLIR with Runtime specific ops. All tf_device.cluster
//     ops are removed. Physical devices are assigned to ops instead of virtual
//     devices.
tensorflow::Status RunLowerClusterToRuntimeOpsPassPipeline(
    mlir::ModuleOp module, tsl::DeviceType xla_device_type,
    llvm::StringRef module_name = llvm::StringRef());

    TF_LOCKS_EXCLUDED(session->graph->mu, session->mu);

std::string getTF_OutputDebugString(TF_Output node);

// Set whether to propagate assigned device information when constructing a new
// Graph from a GraphDef. By default assigned device information is not copied
// and is re-computed by the runtime.
inline void TF_ImportGraphDefOptionsSetPropagateDeviceSpec(