Skip to content

Commit

Permalink
fix(memory): Saving on load enabled if configured
Browse files Browse the repository at this point in the history
- Renamed brain module as memory
  • Loading branch information
timkinnane committed Aug 12, 2018
1 parent 026fd9a commit bce1506
Show file tree
Hide file tree
Showing 6 changed files with 98 additions and 118 deletions.
1 change: 0 additions & 1 deletion src/adapters/mongo.ts
Expand Up @@ -78,7 +78,6 @@ export class Mongo extends StorageAdapter {

/** Put memory data in documents by sub-collection */
async saveMemory (data: any) {
this.bot.logger.debug(`[mongo] saving memory data to DB`)
for (let sub in data) {
const query = { sub, type: 'memory' }
const doc = { data: data[sub] }
Expand Down
2 changes: 1 addition & 1 deletion src/index.ts
Expand Up @@ -13,7 +13,7 @@ export * from './lib/envelope'
export * from './lib/path'
export * from './lib/branch'
export * from './lib/bit'
export * from './lib/brain'
export * from './lib/memory'
export * from './lib/core'
export * from './lib/thought'
export * from './lib/json'
23 changes: 8 additions & 15 deletions src/lib/adapter.spec.ts
Expand Up @@ -3,12 +3,9 @@ import sinon from 'sinon'
import { expect } from 'chai'
import * as bot from '..'

class NLUAdapter extends bot.NLUAdapter {
async start () { /* mock start */ }
async shutdown () { /* mock shutdown */ }
async process () { return {} }
}
class StorageAdapter extends bot.StorageAdapter {
/** Mock Adapter class has methods to imitate all different types */
class MockAdapter extends bot.Adapter {
name = 'mock-adapter'
async start () { /* mock start */ }
async shutdown () { /* mock shutdown */ }
async findOne () { /* mock findOne */ }
Expand All @@ -17,11 +14,7 @@ class StorageAdapter extends bot.StorageAdapter {
async lose () { /* mock lose */ }
async loadMemory () { /* mock loadMemory */ }
async saveMemory () { /* mock saveMemory */ }
}
class MockAdapter extends bot.Adapter {
name = 'mock-adapter'
async start () { /* mock start */ }
async shutdown () { /* mock shutdown */ }
async process () { return {} }
}
export const use = sinon.spy(() => new MockAdapter(bot)) // use spec as module

Expand Down Expand Up @@ -58,8 +51,8 @@ describe('[adapter]', () => {
})
describe('.startAdapters', () => {
it('starts all loaded adapters', async () => {
bot.adapters.storage = new StorageAdapter(bot)
bot.adapters.nlu = new NLUAdapter(bot)
bot.adapters.storage = new MockAdapter(bot)
bot.adapters.nlu = new MockAdapter(bot)
const startStorage = sinon.spy(bot.adapters.storage, 'start')
const startNLU = sinon.spy(bot.adapters.nlu, 'start')
await bot.startAdapters()
Expand All @@ -69,8 +62,8 @@ describe('[adapter]', () => {
})
describe('.shutdownAdapters', () => {
it('shuts down all loaded adapters', async () => {
bot.adapters.storage = new StorageAdapter(bot)
bot.adapters.nlu = new NLUAdapter(bot)
bot.adapters.storage = new MockAdapter(bot)
bot.adapters.nlu = new MockAdapter(bot)
const shutdownStorage = sinon.spy(bot.adapters.storage, 'shutdown')
const shutdownNLU = sinon.spy(bot.adapters.nlu, 'shutdown')
await bot.shutdownAdapters()
Expand Down
7 changes: 2 additions & 5 deletions src/lib/core.ts
Expand Up @@ -38,7 +38,6 @@ export async function load () {
try {
bot.loadMiddleware()
bot.loadAdapters()
// loadServer()
await eventDelay()
setStatus('loaded')
bot.events.emit('loaded')
Expand All @@ -59,7 +58,7 @@ export async function start () {
setStatus('starting')
try {
await bot.startAdapters()
// await startSever()
await bot.startMemory()
} catch (err) {
bot.logger.error('[core] failed to start')
await bot.shutdown(1).catch()
Expand All @@ -86,8 +85,7 @@ export async function shutdown (exit = 0) {
await new Promise((resolve) => bot.events.on('started', () => resolve()))
}
await bot.shutdownAdapters()
// shutdown server
// stop thought process
await bot.shutdownMemory()
await eventDelay()
setStatus('shutdown')
bot.events.emit('shutdown')
Expand Down Expand Up @@ -115,7 +113,6 @@ export async function reset () {
try {
bot.unloadAdapters()
bot.unloadMiddleware()
// unloadServer()
bot.global.reset()
bot.settings.resetConfig()
} catch (err) {
Expand Down

0 comments on commit bce1506

Please sign in to comment.