github.com/vu-ngoc-son/XDP-tutor/p2p-router/packet-capture/compute@v0.0.0-20210523102700-152834548476/internal/calculator/calculator_test.go (about) 1 package calculator 2 3 import ( 4 "github.com/vu-ngoc-son/XDP-tutor/p2p-router/packet-capture/compute/internal/storage" 5 "github.com/vu-ngoc-son/XDP-tutor/p2p-router/packet-capture/compute/internal/storage/peers" 6 "gorm.io/driver/sqlite" 7 "gorm.io/gorm" 8 "testing" 9 ) 10 11 func BenchmarkServiceCalculator_UpdatePeersLimit(b *testing.B) { 12 for i := 0; i < 1; i++ { 13 db, err := gorm.Open(sqlite.Open("/home/ted/TheFirstProject/XDP-tutor/p2p-router/packet-capture/p2p-router.db"), &gorm.Config{}) 14 if err != nil { 15 b.Errorf("%v", err) 16 return 17 } 18 err = db.AutoMigrate(&storage.Peers{}, &storage.Limit{}, &storage.Hosts{}) 19 if err != nil { 20 b.Errorf("%v", err) 21 return 22 } 23 myPeers := peers.New(*db) 24 myCalculator := New(*db, myPeers) 25 myCalculator.UpdatePeersLimit() 26 } 27 }