diff --git a/management/server/activity/sqlite_test.go b/management/server/activity/sqlite_test.go index 5d49aa3e9..4fb7d76ef 100644 --- a/management/server/activity/sqlite_test.go +++ b/management/server/activity/sqlite_test.go @@ -1,27 +1,29 @@ package activity import ( + "fmt" + "github.com/stretchr/testify/assert" "testing" "time" ) func TestNewSQLiteStore(t *testing.T) { - //dataDir := t.TempDir() - store, err := NewSQLiteStore("/home/braginini/wiretrustee/test/") + dataDir := t.TempDir() + store, err := NewSQLiteStore(dataDir) if err != nil { t.Fatal(err) return } - //accountID := "account_1" + accountID := "account_1" - for i := 0; i < 10000; i++ { + for i := 0; i < 10; i++ { _, err = store.Save(&Event{ - Timestamp: time.Now().Add(-1 * time.Minute), + Timestamp: time.Now(), Activity: PeerAddedByUser, - InitiatorID: "google-oauth2|110866222733584764488", - TargetID: "100.101.249.29", - AccountID: "cebi9h3lo1hkhn1qc7cg", + InitiatorID: "user_" + fmt.Sprint(i), + TargetID: "peer_" + fmt.Sprint(i), + AccountID: accountID, }) if err != nil { t.Fatal(err) @@ -29,7 +31,7 @@ func TestNewSQLiteStore(t *testing.T) { } } - /*result, err := store.Get(accountID, 0, 10, false) + result, err := store.Get(accountID, 0, 10, false) if err != nil { t.Fatal(err) return @@ -45,5 +47,5 @@ func TestNewSQLiteStore(t *testing.T) { } assert.Len(t, result, 5) - assert.True(t, result[0].Timestamp.After(result[len(result)-1].Timestamp))*/ + assert.True(t, result[0].Timestamp.After(result[len(result)-1].Timestamp)) }