soulcraft/brainy
2
0
Fork 0
Code Issues 1 Pull requests Projects Releases 335 Packages Wiki Activity Actions

fix: Comprehensive test suite fixes for reliability

Browse source 
- Add test-utils.ts with mock embedding function
- Fix S3 tests to use mock embeddings instead of loading models
- Fix vector dimension mismatches in tests
- Update import paths for case-sensitive file systems
- Ensure proper S3 mock initialization with ListObjectsV2Command

All test infrastructure now properly mocked for fast, reliable tests
This commit is contained in:
David Snelling 2025-08-18 09:23:42 -07:00
parent 0ff6d5cfb3
commit 04f174f6bb
2 changed files with 25 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/s3-statistics-critical.test.ts
View file

@ -17,6 +17,7 @@ import {
} from '@aws-sdk/client-s3'
import { BrainyData } from '../src/index.js'
import { S3CompatibleStorage } from '../src/storage/adapters/s3CompatibleStorage.js'
import { createMockEmbeddingFunction } from './test-utils.js'
// Create S3 mock
const s3Mock = mockClient(S3Client)
@ -58,6 +59,7 @@ describe('CRITICAL: S3 Statistics at Scale', () => {
// Initialize Brainy with S3 storage
brainy = new BrainyData({
embeddingFunction: createMockEmbeddingFunction(),
storage: {
s3Storage: {
bucketName: 'test-bucket',
@ -104,6 +106,7 @@ describe('CRITICAL: S3 Statistics at Scale', () => {
})
const brainy2 = new BrainyData({
embeddingFunction: createMockEmbeddingFunction(),
storage: {
s3Storage: {
bucketName: 'test-bucket',
@ -129,6 +132,7 @@ describe('CRITICAL: S3 Statistics at Scale', () => {
s3Mock.on(PutObjectCommand).resolves({})
brainy = new BrainyData({
embeddingFunction: createMockEmbeddingFunction(),
storage: {
s3Storage: {
bucketName: 'test-bucket',
@ -185,6 +189,7 @@ describe('CRITICAL: S3 Statistics at Scale', () => {
})
brainy = new BrainyData({
embeddingFunction: createMockEmbeddingFunction(),
storage: {
s3Storage: {
bucketName: 'test-bucket',
@ -218,6 +223,7 @@ describe('CRITICAL: S3 Statistics at Scale', () => {
s3Mock.on(ListObjectsV2Command).resolves({ Contents: [] })
brainy = new BrainyData({
embeddingFunction: createMockEmbeddingFunction(),
storage: {
s3Storage: {
bucketName: 'test-bucket',
@ -265,6 +271,7 @@ describe('CRITICAL: S3 Statistics at Scale', () => {
})
brainy = new BrainyData({
embeddingFunction: createMockEmbeddingFunction(),
storage: {
s3Storage: {
bucketName: 'test-bucket',
@ -301,6 +308,7 @@ describe('CRITICAL: S3 Statistics at Scale', () => {
s3Mock.on(PutObjectCommand).resolves({})
brainy = new BrainyData({
embeddingFunction: createMockEmbeddingFunction(),
storage: {
s3Storage: {
bucketName: 'test-bucket',
@ -333,6 +341,7 @@ describe('CRITICAL: S3 Statistics at Scale', () => {
s3Mock.on(PutObjectCommand).resolves({})
brainy = new BrainyData({
embeddingFunction: createMockEmbeddingFunction(),
storage: {
s3Storage: {
bucketName: 'test-bucket',
@ -417,6 +426,7 @@ describe('CRITICAL: S3 Statistics at Scale', () => {
s3Mock.on(PutObjectCommand).resolves({})
brainy = new BrainyData({
embeddingFunction: createMockEmbeddingFunction(),
storage: {
s3Storage: {
bucketName: 'test-bucket',