state logger tests

Author: msanlli

backend/pkg/logger/logger_test.go

@@ -11,6 +11,7 @@ import (
 	"github.com/HyperloopUPV-H8/h9-backend/pkg/logger/state"
 	dataPacketer "github.com/HyperloopUPV-H8/h9-backend/pkg/transport/packet/data"
 	protectionPacketer "github.com/HyperloopUPV-H8/h9-backend/pkg/transport/packet/protection"
+	statePacketer "github.com/HyperloopUPV-H8/h9-backend/pkg/transport/packet/state"
 	"github.com/rs/zerolog"
 	"os"
 	"path"
@@ -167,6 +168,57 @@ func TestLogger(t *testing.T) {
 		t.Errorf("orderErr: expected [test test 0], got %v", output)
 	}
 
+	// State
+	matrix := [8][15]float32{
+		{1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0},
+		{1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0},
+		{1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0},
+		{1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0},
+		{1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0},
+		{1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0},
+		{1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0},
+		{1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0},
+	}
+	statePacket := statePacketer.NewSpace(0, matrix)
+	stateRecord := &state.Record{
+		Packet:    statePacket,
+		From:      "test",
+		To:        "test",
+		Timestamp: timestamp,
+	}
+
+	err = loggerHandler.PushRecord(stateRecord)
+	if err != nil {
+		t.Error(err)
+	}
+
+	filename = path.Join(
+		"logger", "state",
+		logger.Timestamp.Format(logger.TimestampFormat),
+		fmt.Sprintf("%s.csv", stateRecord.Timestamp.Format(logger.TimestampFormat)),
+	)
+	file, err = os.Open(filename)
+	if err != nil {
+		t.Error(err)
+	}
+	defer file.Close()
+
+	time.Sleep(1 * time.Second)
+
+	reader := csv.NewReader(file)
+	for i := 0; i < 8; i++ {
+		output, err = reader.Read()
+		if err != nil {
+			t.Error(err)
+		}
+
+		for j := 0; j < 15; j++ {
+			if output[j] != fmt.Sprint(float32(1.0)) {
+				t.Errorf("stateErr: expected 1.0, got %v", output[j])
+			}
+		}
+	}
+
 	if closeErr := loggerHandler.Stop(); closeErr != nil {
 		t.Error(err)
 	}

backend/pkg/logger/state/logger.go

@@ -70,7 +70,7 @@ func (sublogger *Logger) PushRecord(record abstraction.LoggerRecord) error {
 	filename := path.Join(
 		"logger", "state",
 		logger.Timestamp.Format(logger.TimestampFormat),
-		fmt.Sprintf("%s.csv", time.Now().Format(logger.TimestampFormat)),
+		fmt.Sprintf("%s.csv", stateRecord.Timestamp.Format(logger.TimestampFormat)),
 	)
 	err := os.MkdirAll(path.Dir(filename), os.ModePerm)
 	if err != nil {
@@ -96,14 +96,19 @@ func (sublogger *Logger) PushRecord(record abstraction.LoggerRecord) error {
 	}(file)
 
 	for _, item := range stateRecord.Packet.State() {
-		err = writer.Write([]string{fmt.Sprint(item)})
+		values := make([]string, 0, len(item))
+		for _, value := range item {
+			values = append(values, fmt.Sprint(value))
+		}
+		err = writer.Write(values)
 		if err != nil {
 			return logger.ErrWritingFile{
 				Name:      Name,
 				Timestamp: time.Now(),
 				Inner:     err,
 			}
 		}
+		writer.Flush()
 	}
 
 	return nil