assertEquals(ChatMessage.ROLE_USER, messages.get(0).getRole());
        assertEquals("User 2", messages.get(0).getContent());
        assertEquals(ChatMessage.ROLE_ASSISTANT, messages.get(1).getRole());
        assertEquals("Assistant 2", messages.get(1).getContent());
        assertEquals(ChatMessage.ROLE_USER, messages.get(2).getRole());
        assertEquals("User 3", messages.get(2).getContent());

    }

    @Override
    public byte[] getName() {
        return CONTEXT_NAME_BYTES;
    }

    /**
     * Gets the lease key from the server response
     * @return the lease key
     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

    /**
     * Gets the lease state granted by the server
     * @return the granted lease state
     */
    public int getLeaseState() {

        }

        String logId = logs.get(0).get("id").toString();

        // Get the specific log
        response = checkMethodBase(searchBody).get(API_PATH + "/log/" + logId).asString();
        Map<String, Object> log = JsonPath.from(response).getMap("response.log");

        logger.info("Job log detail: {}", log);
        assertEquals(logId, log.get("id").toString());

    }

    // Test put and get operations
    @Test
    public void test_putAndGet() {
        dataStoreParams.put("key1", "value1");
        assertEquals("value1", dataStoreParams.get("key1"));

        dataStoreParams.put("key2", 123);
        assertEquals(123, dataStoreParams.get("key2"));

        dataStoreParams.put("key3", null);
        assertNull(dataStoreParams.get("key3"));
    }

        assertEquals(7, pnList.size());
        assertEquals("1", pnList.get(0));
        assertEquals("2", pnList.get(1));
        assertEquals("3", pnList.get(2));
        assertEquals("4", pnList.get(3));
        assertEquals("5", pnList.get(4));
        assertEquals("6", pnList.get(5));
        assertEquals("7", pnList.get(6));

    }

    @Test
    public void test_calculatePageInfo_collapse() {

        assertEquals("4.1.0", options.modelVersion().get(), "--model-version should be 4.1.0");

        assertTrue(options.infer().isPresent(), "--infer option should be present");
        assertTrue(options.infer().get(), "--infer option should be true");

        assertTrue(options.model().isPresent(), "--model option should be present");
        assertTrue(options.model().get(), "--model option should be true");

                final String scheme = paramMap.get(CRAWLER_FILE_AUTH + "." + fileAuthName + ".scheme");
                if (Constants.SAMBA.equals(scheme)) {
                    final String domain = paramMap.get(CRAWLER_FILE_AUTH + "." + fileAuthName + ".domain");
                    final String hostname = paramMap.get(CRAWLER_FILE_AUTH + "." + fileAuthName + ".host");

        this.maxLength = maxLength;
        this.pointer = pointer;
    }

    /**
     * Gets the actual length of the string in bytes.
     *
     * @return the string length
     */
    public short getLength() {
        return this.length;
    }

    /**
     * Gets the maximum allocated length for the string in bytes.
     *
     * @return the maximum string length
     */

    assertEquals(1L, map.get("1"));
    assertEquals(2L, map.get("2"));
    assertEquals(3L, map.get("3"));
  }

  public void testIncrementAndGet() {
    AtomicLongMap<String> map = AtomicLongMap.create();
    String key = "key";
    for (int i = 0; i < ITERATIONS; i++) {
      long before = map.get(key);
      long result = map.incrementAndGet(key);
      long after = map.get(key);
      assertEquals(before + 1, after);

  auto tape = std::make_unique<Tape>(/*persistent=*/false);
  s = Execute(check_numerics_op.get(), ctx.get(), absl::MakeSpan(outputs),
              &num_retvals, &forward_op, tape.get(), registry);
  ASSERT_EQ(errors::OK, s.code()) << s.message();

  string read_message;
  s = forward_op.attrs.Get("message", &read_message);
  ASSERT_EQ(errors::OK, s.code()) << s.message();
  ASSERT_EQ(read_message, message);
}