Use response struct for test unmarshalling

suprjinx · Jul 15, 2023 · c890b30 · c890b30
1 parent 70ee18c
commit c890b30
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 11 deletions.
diff --git a/pkg/api/aim/response/run.go b/pkg/api/aim/response/run.go
@@ -1,12 +1,5 @@
 package response
 
-import (
-	"time"
-
-	"github.com/gogo/protobuf/test/tags"
-	"github.com/google/flatbuffers/tests/namespace_test/NamespaceA"
-)
-
 // GetRunInfo represents the response struct for GetRunInfo endpoint
 type GetRunInfo struct {
 	Params GetRunInfoParams `json:"params"`
@@ -30,10 +23,10 @@ type GetRunInfoProps struct {
 	Description string               `json:"description"`
 	Experiment  GetRunInfoExperiment `json:"experiment"`
 	Tags        []string             `json:"tags"`
-	StartTime   time.Time            `json:"creation_time"`
-	EndTime     time.Time            `json:"end_time"`
+	StartTime   float64              `json:"creation_time"`
+	EndTime     float64              `json:"end_time"`
 	Archived    bool                 `json:"archived"`
-	Active      bool                 `json:"archived"`
+	Active      bool                 `json:"active"`
 }
 
 // GetRunInfoExperiment experiment properties

diff --git a/tests/integration/golang/aim/run/get_run_info_test.go b/tests/integration/golang/aim/run/get_run_info_test.go
@@ -12,6 +12,7 @@ import (
 	"github.com/stretchr/testify/suite"
 
 	"github.com/G-Research/fasttrackml/pkg/api/mlflow/dao/models"
+	"github.com/G-Research/fasttrackml/pkg/api/aim/response"
 	"github.com/G-Research/fasttrackml/tests/integration/golang/fixtures"
 	"github.com/G-Research/fasttrackml/tests/integration/golang/helpers"
 )
@@ -67,7 +68,7 @@ func (s *GetRunInfoTestSuite) Test_Ok() {
 	}
 	for _, tt := range tests {
 		s.T().Run(tt.name, func(T *testing.T) {
-			var resp response.GetRunInfoResponse
+			var resp response.GetRunInfo
 			err := s.client.DoGetRequest(
 				fmt.Sprintf("/runs/%s/info", tt.runID),
 				&resp,