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  }