6723064bfb
* adds pointer to netstate proto file * generated updated netstate proto * changes boltdb netstate to save pointers as values * updates netstate Put to save Pointers, updates client example to put a pointer, adds grpc status errors, updates tests, changes boltdb 'File' struct to 'PointerEntry' * updates netstate client example and client test to save pointers, updates netstate List and Delete * begins adding netstate-http tests * removes netstate http service * re-adds netstate auth * updates boltdb netstate test * changes encrypted_unencrypted_size from int64 to bytes in netstate proto * updates READMEs
57 lines
1.0 KiB
Go
57 lines
1.0 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package netstate
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"storj.io/storj/storage/boltdb"
|
|
)
|
|
|
|
// MockDB mocks db functionality for testing
|
|
type MockDB struct {
|
|
timesCalled int
|
|
puts []boltdb.PointerEntry
|
|
pathKeys [][]byte
|
|
}
|
|
|
|
func (m *MockDB) Put(f boltdb.PointerEntry) error {
|
|
m.timesCalled++
|
|
m.puts = append(m.puts, f)
|
|
return nil
|
|
}
|
|
|
|
func (m *MockDB) Get(path []byte) ([]byte, error) {
|
|
m.timesCalled++
|
|
|
|
for _, pointerEntry := range m.puts {
|
|
if bytes.Equal(path, pointerEntry.Path) {
|
|
return pointerEntry.Pointer, nil
|
|
}
|
|
}
|
|
panic("failed to get the given file")
|
|
}
|
|
|
|
func (m *MockDB) List() ([][]byte, error) {
|
|
m.timesCalled++
|
|
|
|
for _, putReq := range m.puts {
|
|
m.pathKeys = append(m.pathKeys, putReq.Path)
|
|
}
|
|
|
|
return m.pathKeys, nil
|
|
}
|
|
|
|
func (m *MockDB) Delete(path []byte) error {
|
|
m.timesCalled++
|
|
|
|
for i, pointerEntry := range m.puts {
|
|
if bytes.Equal(path, pointerEntry.Path) {
|
|
m.puts = append(m.puts[:i], m.puts[i+1:]...)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|