Function CreateMemStore

api/api_test.go:61–80

CreateMemStore creates a mocked MemStore for testing.

(schema *memCom.TableSchema, shardID int, metaStore metaCom.MetaStore,
	diskStore diskstore.DiskStore)

// CreateMemStore creates a mocked MemStore for testing.
func CreateMemStore(schema *memCom.TableSchema, shardID int, metaStore metaCom.MetaStore,
	diskStore diskstore.DiskStore) *memMocks.MemStore {
	redoManagerFactory, _ := redolog.NewRedoLogManagerMaster("", &common.RedoLogConfig{}, diskStore, metaStore)
	options := memstore.NewOptions(new(memComMocks.BootStrapToken), redoManagerFactory)
	shard := memstore.NewTableShard(schema, metaStore, diskStore, CreateMockHostMemoryManger(), shardID, options)

	memStore := new(memMocks.MemStore)
	memStore.On("GetTableShard", schema.Schema.Name, shardID).Return(shard, nil).
		Run(func(arguments mock.Arguments) {
			shard.Users.Add(1)
		})
	memStore.On("GetSchema", schema.Schema.Name).Return(schema, nil)
	memStore.On("GetSchema", mock.Anything).Return(nil, fmt.Errorf("some error"))
	memStore.On("GetSchemas").Return(map[string]*memCom.TableSchema{schema.Schema.Name: schema})
	memStore.On("RLock").Return()
	memStore.On("RUnlock").Return()
	memStore.On("Lock").Return()
	memStore.On("Unlock").Return()
	return memStore
}

// RequestToBody marshals a request struct into json and provide a reader of it.
func RequestToBody(request interface{}) io.Reader {