chore(poll): return NotModifiedTimeoutErr on 304

poll_batch_test.go

@@ -80,3 +80,38 @@ func TestClient_PollRange(t *testing.T) {
 		require.Equal(t, "hello3", string(result.Items[0].Values[0]))
 	}
 }
+
+func TestClient_PollRange_Timeout(t *testing.T) {
+	if testing.Short() {
+		t.Skip("Skipping in short mode: 1 sec minimum to trigger timeout")
+		return
+	}
+	t.Parallel()
+
+	f, ctx := newFixture(t)
+
+	pk := randomPk()
+	sk := randomSk()
+
+	for i := range 5 {
+		err := f.cli.InsertItem(ctx, f.bucket, pk, sk+"-"+strconv.Itoa(i), "", []byte("hello1"))
+		require.NoError(t, err)
+	}
+
+	// first read should complete immediately
+	q := k2v.PollRangeQuery{
+		Start: sk,
+	}
+	result, err := f.cli.PollRange(ctx, f.bucket, pk, q, 5*time.Second)
+	require.NoError(t, err)
+	require.NotEmpty(t, result.SeenMarker)
+	require.Len(t, result.Items, 5)
+	for i := range result.Items {
+		require.Len(t, result.Items[i].Values, 1)
+		require.Equal(t, "hello1", string(result.Items[i].Values[0]))
+	}
+
+	q.SeenMarker = result.SeenMarker
+	result, err = f.cli.PollRange(ctx, f.bucket, pk, q, 1*time.Second)
+	require.ErrorIs(t, err, k2v.NotModifiedTimeoutErr)
+}