tests

ydb-platform · Sep 4, 2024 · 6c44d61 · 6c44d61
1 parent 4697ecb
commit 6c44d61
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 1,035 deletions.
diff --git a/internal/pool/pool_test.go b/internal/pool/pool_test.go
@@ -842,5 +842,41 @@ func TestPool(t *testing.T) {
 				wg.Wait()
 			})
 		})
+		t.Run("PutInFull", func(t *testing.T) {
+			p := New(rootCtx,
+				WithLimit[*testItem, testItem](1),
+				WithCreateItemTimeout[*testItem, testItem](50*time.Millisecond),
+				WithCloseItemTimeout[*testItem, testItem](50*time.Millisecond),
+				// replace default async closer for sync testing
+				withCloseItemFunc(func(ctx context.Context, item *testItem) {
+					_ = item.Close(ctx)
+				}),
+			)
+			item := mustGetItem(t, p)
+			if err := p.putItem(context.Background(), item); err != nil {
+				t.Fatalf("unexpected error on put session into non-full client: %v, wand: %v", err, nil)
+			}
+
+			if err := p.putItem(context.Background(), &testItem{}); !xerrors.Is(err, errPoolIsOverflow) {
+				t.Fatalf("unexpected error on put item into full pool: %v, wand: %v", err, errPoolIsOverflow)
+			}
+		})
+		t.Run("PutTwice", func(t *testing.T) {
+			p := New(rootCtx,
+				WithLimit[*testItem, testItem](2),
+				WithCreateItemTimeout[*testItem, testItem](50*time.Millisecond),
+				WithCloseItemTimeout[*testItem, testItem](50*time.Millisecond),
+				// replace default async closer for sync testing
+				withCloseItemFunc(func(ctx context.Context, item *testItem) {
+					_ = item.Close(ctx)
+				}),
+			)
+			item := mustGetItem(t, p)
+			mustPutItem(t, p, item)
+
+			require.Panics(t, func() {
+				_ = p.putItem(context.Background(), item)
+			})
+		})
 	})
 }