mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-09-15 16:51:03 +00:00
94 lines
3.1 KiB
Go
94 lines
3.1 KiB
Go
// Copyright 2017 Pilosa Corp.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package pilosa
|
|
|
|
import (
|
|
"bytes"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/pilosa/pilosa/v2/roaring"
|
|
)
|
|
|
|
func TestValidateName(t *testing.T) {
|
|
names := []string{
|
|
"a", "ab", "ab1", "b-c", "d_e", "exists",
|
|
"longbutnottoolongaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa12345689012345689012345678901234567890",
|
|
}
|
|
for _, name := range names {
|
|
if validateName(name) != nil {
|
|
t.Fatalf("Should be valid index name: %s", name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestValidateNameInvalid(t *testing.T) {
|
|
names := []string{
|
|
"", "'", "^", "/", "\\", "A", "*", "a:b", "valid?no", "yüce", "1", "_", "-",
|
|
"long123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa1", "_exists",
|
|
}
|
|
for _, name := range names {
|
|
if validateName(name) == nil {
|
|
t.Fatalf("Should be invalid index name: %s", name)
|
|
}
|
|
}
|
|
}
|
|
|
|
// memAttrStore represents an in-memory implementation of the AttrStore interface.
|
|
type memAttrStore struct {
|
|
store map[uint64]map[string]interface{}
|
|
}
|
|
|
|
func (s *memAttrStore) Path() string { return "" }
|
|
func (s *memAttrStore) Open() error { return nil }
|
|
func (s *memAttrStore) Close() error { return nil }
|
|
func (s *memAttrStore) Attrs(id uint64) (m map[string]interface{}, err error) {
|
|
return s.store[id], nil
|
|
}
|
|
func (s *memAttrStore) SetAttrs(id uint64, m map[string]interface{}) error {
|
|
s.store[id] = m
|
|
return nil
|
|
}
|
|
func (s *memAttrStore) SetBulkAttrs(m map[uint64]map[string]interface{}) error {
|
|
for id, v := range m {
|
|
s.store[id] = v
|
|
}
|
|
return nil
|
|
}
|
|
func (s *memAttrStore) Blocks() ([]AttrBlock, error) { return nil, nil }
|
|
func (s *memAttrStore) BlockData(i uint64) (map[uint64]map[string]interface{}, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func TestAPI_CombineForExistence(t *testing.T) {
|
|
bm := roaring.NewBitmap(pos(1, 1), pos(1, 2), pos(1, 3), pos(1, 65537), pos(1, 65538), pos(2, 1), pos(2, 2), pos(2, 5), pos(2, 65537), pos(2, 65538))
|
|
buf := new(bytes.Buffer)
|
|
_, err := bm.WriteTo(buf)
|
|
panicOn(err)
|
|
raw := buf.Bytes()
|
|
results, err := combineForExistence(raw)
|
|
if err != nil {
|
|
t.Fatalf("failure to combine: %v", err)
|
|
}
|
|
bm2 := roaring.NewBitmap()
|
|
_, _, err = bm2.ImportRoaringBits(results, false, false, 1<<shardVsContainerExponent)
|
|
panicOn(err)
|
|
expected := []uint64{1, 2, 3, 5, 65537, 65538}
|
|
got := bm2.Slice()
|
|
if !reflect.DeepEqual(got, expected) {
|
|
t.Fatalf("expected:%v got:%v", expected, got)
|
|
}
|
|
|
|
}
|