change archive state to 3 value enum; add experiment integration test

backend/src/apiserver/storage/experiment_store.go

@@ -252,6 +252,18 @@ func (s *ExperimentStore) ArchiveExperiment(expId string) error {
 			"Failed to create query to archive experiment %s. error: '%v'", expId, err.Error())
 	}
 
+	updateRunsWithExperimentUUIDSql, updateRunsWithExperimentUUIDArgs, err := sq.
+		Update("run_details").
+		SetMap(sq.Eq{
+			"StorageState": api.Run_STORAGESTATE_ARCHIVED.String(),
+		}).
+		Where(sq.Eq{"ExperimentUUID": expId}).
+		ToSql()
+	if err != nil {
+		return util.NewInternalServerError(err,
+			"Failed to create query to archive the runs in an experiment %s. error: '%v'", expId, err.Error())
+	}
+
 	// TODO(jingzhang36): use inner join to replace nested query for better performance.
 	filteredRunsSql, filteredRunsArgs, err := sq.Select("ResourceUUID").
 		From("resource_references as rf").
@@ -263,7 +275,6 @@ func (s *ExperimentStore) ArchiveExperiment(expId string) error {
 		return util.NewInternalServerError(err,
 			"Failed to create query to filter the runs in an experiment %s. error: '%v'", expId, err.Error())
 	}
-
 	updateRunsSql, updateRunsArgs, err := sq.
 		Update("run_details").
 		SetMap(sq.Eq{
@@ -314,11 +325,18 @@ func (s *ExperimentStore) ArchiveExperiment(expId string) error {
 			"Failed to archive experiment %s. error: '%v'", expId, err.Error())
 	}
 
+	_, err = tx.Exec(updateRunsWithExperimentUUIDSql, updateRunsWithExperimentUUIDArgs)
+	if err != nil {
+		tx.Rollback()
+		return util.NewInternalServerError(err,
+			"Failed to archive runs with ExperimentUUID being %s. error: '%v'", expId, err.Error())
+	}
+
 	_, err = tx.Exec(updateRunsSql, updateRunsArgs...)
 	if err != nil {
 		tx.Rollback()
 		return util.NewInternalServerError(err,
-			"Failed to archive all runs in an experiment %s. error: '%v'", expId, err.Error())
+			"Failed to archive runs with experiment reference being %s. error: '%v'", expId, err.Error())
 	}
 
 	_, err = tx.Exec(updateJobsSql, updateJobsArgs...)