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

fix: correct test assertions for addMany return value and use valid VerbType

Browse source 
- Update addMany() expectations to use result.successful array
- Change invalid 'worksOn' VerbType to VerbType.WorksWith
- Import VerbType in test file for type safety
This commit is contained in:
David Snelling 2025-10-10 14:23:24 -07:00
parent 46c6af3f21
commit d9ac9bc10e
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/integration/hnsw-rebuild.test.ts
View file

@ -14,7 +14,7 @@
*/ */
import { describe, it, expect, beforeAll, afterAll, beforeEach, afterEach } from 'vitest' import { describe, it, expect, beforeAll, afterAll, beforeEach, afterEach } from 'vitest'
import { Brainy } from '../../src/brainy' import { Brainy, VerbType } from '../../src/brainy'
import * as fs from 'fs/promises' import * as fs from 'fs/promises'
import * as path from 'path' import * as path from 'path'
import * as os from 'os' import * as os from 'os'
@ -182,9 +182,9 @@ describe('HNSW Index Rebuild (Integration Tests)', () => {
await brain1.init() await brain1.init()
const testData = createTestData(MEDIUM_DATASET_SIZE) const testData = createTestData(MEDIUM_DATASET_SIZE)
const ids = await brain1.addMany({ items: testData }) const result = await brain1.addMany({ items: testData })
expect(ids).toHaveLength(MEDIUM_DATASET_SIZE) expect(result.successful).toHaveLength(MEDIUM_DATASET_SIZE)
console.log(`✅ Phase 1: Added ${MEDIUM_DATASET_SIZE} entities`) console.log(`✅ Phase 1: Added ${MEDIUM_DATASET_SIZE} entities`)
// Perform multiple searches to establish baseline // Perform multiple searches to establish baseline
@ -257,12 +257,12 @@ describe('HNSW Index Rebuild (Integration Tests)', () => {
const addStart = Date.now() const addStart = Date.now()
const testData = createTestData(LARGE_DATASET_SIZE) const testData = createTestData(LARGE_DATASET_SIZE)
const ids = await brain1.addMany({ items: testData }) const result = await brain1.addMany({ items: testData })
const addDuration = Date.now() - addStart const addDuration = Date.now() - addStart
console.log(`✅ Added ${LARGE_DATASET_SIZE} entities in ${addDuration}ms`) console.log(`✅ Added ${LARGE_DATASET_SIZE} entities in ${addDuration}ms`)
expect(ids).toHaveLength(LARGE_DATASET_SIZE) expect(result.successful).toHaveLength(LARGE_DATASET_SIZE)
// Test search before rebuild // Test search before rebuild
const queryBefore = await brain1.find({ const queryBefore = await brain1.find({
@ -326,9 +326,9 @@ describe('HNSW Index Rebuild (Integration Tests)', () => {
await brain1.init() await brain1.init()
const testData = createTestData(50) const testData = createTestData(50)
const ids = await brain1.addMany({ items: testData }) const result = await brain1.addMany({ items: testData })
expect(ids).toHaveLength(50) expect(result.successful).toHaveLength(50)
// Perform search // Perform search
const queryResults = await brain1.find({ const queryResults = await brain1.find({
@ -400,13 +400,13 @@ describe('HNSW Index Rebuild (Integration Tests)', () => {
await brain1.relate({ await brain1.relate({
from: alice, from: alice,
to: project, to: project,
type: 'worksOn' type: VerbType.WorksWith
}) })
await brain1.relate({ await brain1.relate({
from: bob, from: bob,
to: project, to: project,
type: 'worksOn' type: VerbType.WorksWith
}) })
console.log('✅ Phase 1: Created entities and relationships') console.log('✅ Phase 1: Created entities and relationships')