Milestone API implement in sqlite

sarchlab · Nov 15, 2024 · 327b5c8 · 327b5c8
1 parent d0e88b9
commit 327b5c8
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/timing/rob/rob_test.go b/timing/rob/rob_test.go
@@ -11,6 +11,8 @@ import (
 	"github.com/sarchlab/akita/v4/datarecording"
 )
 
+var dataRecorder *datarecording.SQLiteWriter
+
 type myHook struct {
     f func(ctx sim.HookCtx)
 }
@@ -35,6 +37,19 @@ func (b *sqliteTracerBackend) Flush() {
     b.backend.Flush()
 }
 
+var _ = BeforeSuite(func() {
+    dataRecorder = datarecording.NewSQLiteWriter("test_database")
+    dataRecorder.Init()
+    dataRecorder.CreateTable("milestones", tracing.Milestone{})
+    tracing.SetDataRecorder(dataRecorder)
+})
+
+var _ = AfterSuite(func() {
+    dataRecorder.Flush()
+    dataRecorder.DB.Close()
+    // os.Remove("test_database.sqlite3")
+})
+
 var _ = Describe("Reorder Buffer", func() {
 	var (
 		mockCtrl   *gomock.Controller