70 lines
1.2 KiB
Go
70 lines
1.2 KiB
Go
|
package database
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/FreifunkBremen/yanic/runtime"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
type testConn struct {
|
||
|
|
Connection
|
||
|
|
countClose int
|
||
|
|
countPrune int
|
||
|
|
sync.Mutex
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *testConn) Close() {
|
||
|
|
c.Lock()
|
||
|
|
c.countClose++
|
||
|
|
c.Unlock()
|
||
|
|
}
|
||
|
|
func (c *testConn) GetClose() int {
|
||
|
|
c.Lock()
|
||
|
|
defer c.Unlock()
|
||
|
|
return c.countClose
|
||
|
|
}
|
||
|
|
func (c *testConn) PruneNodes(time.Duration) {
|
||
|
|
c.Lock()
|
||
|
|
c.countPrune++
|
||
|
|
c.Unlock()
|
||
|
|
}
|
||
|
|
func (c *testConn) GetPruneNodes() int {
|
||
|
|
c.Lock()
|
||
|
|
defer c.Unlock()
|
||
|
|
return c.countPrune
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStart(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
|
||
|
|
conn := &testConn{}
|
||
|
|
config := &runtime.Config{
|
||
|
|
Database: struct {
|
||
|
|
DeleteInterval runtime.Duration `toml:"delete_interval"`
|
||
|
|
DeleteAfter runtime.Duration `toml:"delete_after"`
|
||
|
|
Connection map[string]interface{}
|
||
|
|
}{
|
||
|
|
DeleteInterval: runtime.Duration{Duration: time.Millisecond * 10},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
assert.Nil(quit)
|
||
|
|
|
||
|
|
Start(conn, config)
|
||
|
|
assert.NotNil(quit)
|
||
|
|
|
||
|
|
assert.Equal(0, conn.GetPruneNodes())
|
||
|
|
time.Sleep(time.Millisecond * 12)
|
||
|
|
assert.Equal(1, conn.GetPruneNodes())
|
||
|
|
|
||
|
|
assert.Equal(0, conn.GetClose())
|
||
|
|
Close(conn)
|
||
|
|
assert.NotNil(quit)
|
||
|
|
assert.Equal(1, conn.GetClose())
|
||
|
|
|
||
|
|
time.Sleep(time.Millisecond * 12) // to reach timer.Stop() line
|
||
|
|
|
||
|
|
}
|