Skip to content
Snippets Groups Projects
Commit 940766a6 authored by ale's avatar ale
Browse files

Make the db write tests a bit faster

parent c3a78a7e
No related branches found
No related tags found
No related merge requests found
...@@ -15,7 +15,11 @@ import ( ...@@ -15,7 +15,11 @@ import (
ippb "git.autistici.org/ai3/tools/iprep/proto" ippb "git.autistici.org/ai3/tools/iprep/proto"
) )
const manyEvents = 500000 const (
// How many events to fill up the db, test/bench cases.
manyEventsTest = 100000
manyEventsBench = 1000000
)
func randomIP() string { func randomIP() string {
return fmt.Sprintf("%d.%d.%d.%d", 1+rand.Intn(253), 1+rand.Intn(253), 1+rand.Intn(253), 1+rand.Intn(253)) return fmt.Sprintf("%d.%d.%d.%d", 1+rand.Intn(253), 1+rand.Intn(253), 1+rand.Intn(253), 1+rand.Intn(253))
...@@ -68,7 +72,7 @@ func runRWTest(t *testing.T, driver string) { ...@@ -68,7 +72,7 @@ func runRWTest(t *testing.T, driver string) {
defer db.Close() defer db.Close()
// Write n elements. // Write n elements.
n := manyEvents n := manyEventsTest
events := randomEvents(n) events := randomEvents(n)
if err := db.AddAggregate(events); err != nil { if err := db.AddAggregate(events); err != nil {
t.Fatalf("AddEvents: %v", err) t.Fatalf("AddEvents: %v", err)
...@@ -185,7 +189,7 @@ func runReadBenchmark(b *testing.B, driver string, eventsPerIP int, threadCounts ...@@ -185,7 +189,7 @@ func runReadBenchmark(b *testing.B, driver string, eventsPerIP int, threadCounts
// Write a lot of events just to bulk up the database, but // Write a lot of events just to bulk up the database, but
// also write a couple of records for a known IP, that we're // also write a couple of records for a known IP, that we're
// going to read in the benchmark. // going to read in the benchmark.
n := manyEvents n := manyEventsBench
db.AddAggregate(randomEvents(n)) db.AddAggregate(randomEvents(n))
// For a bunch of random IPs, add a known (possibly large) // For a bunch of random IPs, add a known (possibly large)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment