@ParameterizedTest
        @DisplayName("Should calculate correct size for various path lengths")
        @CsvSource({ "'', 4", "'a', 6", "'\\\\', 8", "'\\\\server', 20", "'\\\\server\\share', 32", "'\\\\server\\share\\path', 42",
                "'\\\\server\\share\\very\\long\\path\\with\\many\\segments', 100" })
        void testSizeCalculation(String path, int expectedSize) {
            buffer = new DfsReferralRequestBuffer(path, 3);

        CompletableFuture<WitnessRegistration> future = client.registerForNotifications("\\\\server\\share", serverAddress, listener);

        WitnessRegistration registration = future.get(5, TimeUnit.SECONDS);

        assertNotNull(registration);
        assertEquals(WitnessRegistrationState.REGISTERED, registration.getState());
        assertEquals("\\\\server\\share", registration.getShareName());

        // Setup minimal mocks
        when(request.getRequestURI()).thenReturn("/explorer");
        when(request.getPathInfo()).thenReturn("/share/");
        when(locator.getCanonicalURL()).thenReturn("smb://server/share/");

        // Mock directory listing
        SmbFile file1 = mock(SmbFile.class);
        when(file1.getName()).thenReturn("document.pdf");

    }

    @Test
    @DisplayName("Test getPath returns correct formatted path")
    void testGetPath() {
        request = new Smb2CreateRequest(mockConfig, "share\\file.txt");
        assertEquals("\\share\\file.txt", request.getPath());

        request.setPath("newpath\\newfile.txt");
        assertEquals("\\newpath\\newfile.txt", request.getPath());
    }

    @Test

                            SmbTreeImpl uct = smbSession.getSmbTree(share, null).unwrap(SmbTreeImpl.class);
                            SmbTreeImpl ct = connectTree(loc, host, share, trans, uct, dr)) {

                        if (dr != null) {
                            ct.setTreeReferral(dr);
                            if (dr != start) {

     */
    String getURLPath();

    /**
     * Retrieves the share associated with this SMB resource. In
     * the case of <code>smb://</code>, <code>smb://workgroup/</code>,
     * and <code>smb://server/</code> URLs which do not specify a share,
     * <code>null</code> will be returned.
     *
     * @return The share component or <code>null</code> if there is no share
     */
    String getShare();

    /**

// TODO(b/193656009): Defining these in a cc file causes linker errors with
// fastbuild.
inline absl::Status AbstractOperation::SetAttrShape(
    const char* attr_name, const PartialTensorShape shape) {
  return SetAttrShape(attr_name, shape.dim_sizes().data(), shape.dims());
}

inline absl::Status AbstractOperation::SetAttrStringList(
    const char* attr_name, absl::Span<string const> values) {
  std::vector<const char*> raw_strs;

  if (reader_ != nullptr) {
    status = reader_->GetTensor(name, out_tensor);
  } else {
    tensorflow::DataType dtype;
    tensorflow::TensorShape shape;
    status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape);
    if (status.ok()) {
      out_tensor->reset(new Tensor(dtype, shape));
      status = v2_reader_->Lookup(name, out_tensor->get());
      if (!status.ok()) out_tensor->reset();
    }
  }
  if (!status.ok()) {

    void testCIFSContextInterface() {
        // Verify interface methods exist
        assertDoesNotThrow(() -> {
            mockContext.get("smb://server/share");
            mockContext.getPipe("smb://server/share/pipe", 0);
            mockContext.close();
            mockContext.getConfig();
            mockContext.getNameServiceClient();
            mockContext.getBufferCache();

}

// Create an empty tensor of type 'dtype'. 'shape' can be arbitrary, but has to
// result in a zero-sized tensor.
static TF_Tensor* EmptyTensor(TF_DataType dtype,
                              const tensorflow::TensorShape& shape) {
  static char empty;
  int64_t nelems = 1;
  std::vector<int64_t> dims;
  dims.reserve(shape.dims());
  for (int i = 0; i < shape.dims(); ++i) {
    dims.push_back(shape.dim_size(i));