f9af3b6ee4
* initial commit for PUT request authorization * inital auth for put request * auth. request working for all req * deleted library * removed .db files * work in progress for modifying test suite to accomodate credentials * modified tests * gofmt, fixed code based on suggestions; test passed * gofmt again * merged nat's update on pointers, passed tests, cleanup from git rebase * fixed fmt * fixed fmt on tests * work in progress * reduced code * fixed naming conventions * added line in code * fixed server bug, merged new code to server, added env * fixed linter; getting cright issues on piecestore * added comments for what passes on the creds
71 lines
1.2 KiB
Go
71 lines
1.2 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package netstate
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/spf13/viper"
|
|
|
|
"storj.io/storj/storage/boltdb"
|
|
)
|
|
|
|
const (
|
|
API_KEY = "abc123"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
viper.SetEnvPrefix("API")
|
|
os.Setenv("API_KEY", API_KEY)
|
|
viper.AutomaticEnv()
|
|
os.Exit(m.Run())
|
|
}
|
|
|
|
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
|
|
}
|