feat(slurmdb): add all slurmdb response types and supporting types
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
dailz
309
internal/slurm/types_slurmdb.go
Normal file
309
internal/slurm/types_slurmdb.go
Normal file
@@ -0,0 +1,309 @@
|
|||||||
|
package slurm
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// SlurmDBD Response wrapper types — v0.0.40 OpenAPI SlurmDBD schemas
|
||||||
|
// All fields are optional (pointer types) with json:"name,omitempty".
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
// OpenapiSlurmdbdJobsResp is the response for listing SlurmDBD jobs.
|
||||||
|
// Corresponds to v0.0.40_openapi_slurmdbd_jobs_resp.
|
||||||
|
type OpenapiSlurmdbdJobsResp struct {
|
||||||
|
Jobs JobList `json:"jobs,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiSlurmdbdConfigResp is the response for SlurmDBD configuration.
|
||||||
|
// Corresponds to v0.0.40_openapi_slurmdbd_config_resp.
|
||||||
|
type OpenapiSlurmdbdConfigResp struct {
|
||||||
|
Clusters ClusterRecList `json:"clusters,omitempty"`
|
||||||
|
Tres TresList `json:"tres,omitempty"`
|
||||||
|
Accounts AccountList `json:"accounts,omitempty"`
|
||||||
|
Users UserList `json:"users,omitempty"`
|
||||||
|
Qos QosList `json:"qos,omitempty"`
|
||||||
|
Wckeys WckeyList `json:"wckeys,omitempty"`
|
||||||
|
Associations AssocList `json:"associations,omitempty"`
|
||||||
|
Instances InstanceList `json:"instances,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiSlurmdbdQosResp is the response for listing QOS.
|
||||||
|
// Corresponds to v0.0.40_openapi_slurmdbd_qos_resp.
|
||||||
|
type OpenapiSlurmdbdQosResp struct {
|
||||||
|
Qos QosList `json:"qos,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiSlurmdbdQosRemovedResp is the response for removed QOS.
|
||||||
|
// Corresponds to v0.0.40_openapi_slurmdbd_qos_removed_resp.
|
||||||
|
type OpenapiSlurmdbdQosRemovedResp struct {
|
||||||
|
RemovedQos StringList `json:"removed_qos,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiSlurmdbdStatsResp is the response for SlurmDBD diagnostics.
|
||||||
|
// Corresponds to v0.0.40_openapi_slurmdbd_stats_resp.
|
||||||
|
type OpenapiSlurmdbdStatsResp struct {
|
||||||
|
Statistics *StatsRec `json:"statistics,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiAssocsResp is the response for listing associations.
|
||||||
|
// Corresponds to v0.0.40_openapi_assocs_resp.
|
||||||
|
type OpenapiAssocsResp struct {
|
||||||
|
Associations AssocList `json:"associations,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiAssocsRemovedResp is the response for removed associations.
|
||||||
|
// Corresponds to v0.0.40_openapi_assocs_removed_resp.
|
||||||
|
type OpenapiAssocsRemovedResp struct {
|
||||||
|
RemovedAssociations StringList `json:"removed_associations,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiUsersResp is the response for listing users.
|
||||||
|
// Corresponds to v0.0.40_openapi_users_resp.
|
||||||
|
type OpenapiUsersResp struct {
|
||||||
|
Users UserList `json:"users,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiClustersResp is the response for listing clusters.
|
||||||
|
// Corresponds to v0.0.40_openapi_clusters_resp.
|
||||||
|
type OpenapiClustersResp struct {
|
||||||
|
Clusters ClusterRecList `json:"clusters,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiClustersRemovedResp is the response for deleted clusters.
|
||||||
|
// Corresponds to v0.0.40_openapi_clusters_removed_resp.
|
||||||
|
type OpenapiClustersRemovedResp struct {
|
||||||
|
DeletedClusters StringList `json:"deleted_clusters,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiWckeyResp is the response for listing WCKeys.
|
||||||
|
// Corresponds to v0.0.40_openapi_wckey_resp.
|
||||||
|
type OpenapiWckeyResp struct {
|
||||||
|
Wckeys WckeyList `json:"wckeys,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiWckeyRemovedResp is the response for deleted WCKeys.
|
||||||
|
// Corresponds to v0.0.40_openapi_wckey_removed_resp.
|
||||||
|
type OpenapiWckeyRemovedResp struct {
|
||||||
|
DeletedWckeys StringList `json:"deleted_wckeys,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiAccountsResp is the response for listing accounts.
|
||||||
|
// Corresponds to v0.0.40_openapi_accounts_resp.
|
||||||
|
type OpenapiAccountsResp struct {
|
||||||
|
Accounts AccountList `json:"accounts,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiAccountsRemovedResp is the response for removed accounts.
|
||||||
|
// Corresponds to v0.0.40_openapi_accounts_removed_resp.
|
||||||
|
type OpenapiAccountsRemovedResp struct {
|
||||||
|
RemovedAccounts StringList `json:"removed_accounts,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiInstancesResp is the response for listing instances.
|
||||||
|
// Corresponds to v0.0.40_openapi_instances_resp.
|
||||||
|
type OpenapiInstancesResp struct {
|
||||||
|
Instances InstanceList `json:"instances,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiTresResp is the response for listing TRES.
|
||||||
|
// Corresponds to v0.0.40_openapi_tres_resp.
|
||||||
|
// Note: JSON key for TRES is UPPERCASE per OpenAPI spec.
|
||||||
|
type OpenapiTresResp struct {
|
||||||
|
TRES TresList `json:"TRES,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiUsersAddCondResp is the response for adding users with conditions.
|
||||||
|
// Corresponds to v0.0.40_openapi_users_add_cond_resp.
|
||||||
|
type OpenapiUsersAddCondResp struct {
|
||||||
|
AssociationCondition *UsersAddCond `json:"association_condition,omitempty"`
|
||||||
|
User *UserShort `json:"user,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiUsersAddCondRespStr is the string response for adding users with conditions.
|
||||||
|
// Corresponds to v0.0.40_openapi_users_add_cond_resp_str.
|
||||||
|
type OpenapiUsersAddCondRespStr struct {
|
||||||
|
AddedUsers *string `json:"added_users,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiAccountsAddCondResp is the response for adding accounts with conditions.
|
||||||
|
// Corresponds to v0.0.40_openapi_accounts_add_cond_resp.
|
||||||
|
type OpenapiAccountsAddCondResp struct {
|
||||||
|
AssociationCondition *AccountsAddCond `json:"association_condition,omitempty"`
|
||||||
|
Account *AccountShort `json:"account,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenapiAccountsAddCondRespStr is the string response for adding accounts with conditions.
|
||||||
|
// Corresponds to v0.0.40_openapi_accounts_add_cond_resp_str.
|
||||||
|
type OpenapiAccountsAddCondRespStr struct {
|
||||||
|
AddedAccounts *string `json:"added_accounts,omitempty"`
|
||||||
|
Meta *OpenapiMeta `json:"meta,omitempty"`
|
||||||
|
Errors OpenapiErrors `json:"errors,omitempty"`
|
||||||
|
Warnings OpenapiWarnings `json:"warnings,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// Supporting types for SlurmDBD diagnostics
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
// StatsRec represents SlurmDBD diagnostic statistics.
|
||||||
|
// Corresponds to v0.0.40_stats_rec.
|
||||||
|
type StatsRec struct {
|
||||||
|
TimeStart *int64 `json:"time_start,omitempty"`
|
||||||
|
Rollups RollupStatsList `json:"rollups,omitempty"`
|
||||||
|
RPCs StatsRpcList `json:"RPCs,omitempty"`
|
||||||
|
Users StatsUserList `json:"users,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RollupStatsList is a list of RollupStats.
|
||||||
|
// Corresponds to v0.0.40_rollup_stats.
|
||||||
|
type RollupStatsList []RollupStats
|
||||||
|
|
||||||
|
// RollupStats represents recorded rollup statistics.
|
||||||
|
type RollupStats struct {
|
||||||
|
Type *string `json:"type,omitempty"`
|
||||||
|
LastRun *int32 `json:"last run,omitempty"`
|
||||||
|
MaxCycle *int64 `json:"max_cycle,omitempty"`
|
||||||
|
TotalTime *int64 `json:"total_time,omitempty"`
|
||||||
|
TotalCycles *int64 `json:"total_cycles,omitempty"`
|
||||||
|
MeanCycles *int64 `json:"mean_cycles,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// StatsRpcList is a list of StatsRpc.
|
||||||
|
// Corresponds to v0.0.40_stats_rpc_list.
|
||||||
|
type StatsRpcList []StatsRpc
|
||||||
|
|
||||||
|
// StatsRpc represents RPC statistics.
|
||||||
|
// Corresponds to v0.0.40_stats_rpc.
|
||||||
|
type StatsRpc struct {
|
||||||
|
RPC *string `json:"rpc,omitempty"`
|
||||||
|
Count *int32 `json:"count,omitempty"`
|
||||||
|
Time *StatsRpcTime `json:"time,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// StatsRpcTime represents timing information for RPC statistics.
|
||||||
|
type StatsRpcTime struct {
|
||||||
|
Average *int64 `json:"average,omitempty"`
|
||||||
|
Total *int64 `json:"total,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// StatsUserList is a list of StatsUser.
|
||||||
|
// Corresponds to v0.0.40_stats_user_list.
|
||||||
|
type StatsUserList []StatsUser
|
||||||
|
|
||||||
|
// StatsUser represents per-user statistics.
|
||||||
|
// Corresponds to v0.0.40_stats_user.
|
||||||
|
type StatsUser struct {
|
||||||
|
User *string `json:"user,omitempty"`
|
||||||
|
Count *int32 `json:"count,omitempty"`
|
||||||
|
Time *StatsUserTime `json:"time,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// StatsUserTime represents timing information for user statistics.
|
||||||
|
type StatsUserTime struct {
|
||||||
|
Average *int64 `json:"average,omitempty"`
|
||||||
|
Total *int64 `json:"total,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// UsersAddCond and AccountsAddCond — association condition types
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
// UsersAddCond represents filters to select associations for users.
|
||||||
|
// Corresponds to v0.0.40_users_add_cond.
|
||||||
|
type UsersAddCond struct {
|
||||||
|
Accounts StringList `json:"accounts,omitempty"`
|
||||||
|
Association *AssocRecSet `json:"association,omitempty"`
|
||||||
|
Clusters StringList `json:"clusters,omitempty"`
|
||||||
|
Partitions StringList `json:"partitions,omitempty"`
|
||||||
|
Users StringList `json:"users,omitempty"`
|
||||||
|
Wckeys StringList `json:"wckeys,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// AccountsAddCond represents filters for adding accounts.
|
||||||
|
// Corresponds to v0.0.40_accounts_add_cond.
|
||||||
|
type AccountsAddCond struct {
|
||||||
|
Accounts StringList `json:"accounts,omitempty"`
|
||||||
|
Association *AssocRecSet `json:"association,omitempty"`
|
||||||
|
Clusters StringList `json:"clusters,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// AssocRecSet represents association limits and options for user/account creation.
|
||||||
|
// Corresponds to v0.0.40_assoc_rec_set.
|
||||||
|
type AssocRecSet struct {
|
||||||
|
Comment *string `json:"comment,omitempty"`
|
||||||
|
Defaultqos *string `json:"defaultqos,omitempty"`
|
||||||
|
Grpjobs *Uint32NoVal `json:"grpjobs,omitempty"`
|
||||||
|
Grpjobsaccrue *Uint32NoVal `json:"grpjobsaccrue,omitempty"`
|
||||||
|
Grpsubmitjobs *Uint32NoVal `json:"grpsubmitjobs,omitempty"`
|
||||||
|
Grptres TresList `json:"grptres,omitempty"`
|
||||||
|
Grptresmins TresList `json:"grptresmins,omitempty"`
|
||||||
|
Grptresrunmins TresList `json:"grptresrunmins,omitempty"`
|
||||||
|
Grpwall *Uint32NoVal `json:"grpwall,omitempty"`
|
||||||
|
Maxjobs *Uint32NoVal `json:"maxjobs,omitempty"`
|
||||||
|
Maxjobsaccrue *Uint32NoVal `json:"maxjobsaccrue,omitempty"`
|
||||||
|
Maxsubmitjobs *Uint32NoVal `json:"maxsubmitjobs,omitempty"`
|
||||||
|
Maxtresminsperjob TresList `json:"maxtresminsperjob,omitempty"`
|
||||||
|
Maxtresrunmins TresList `json:"maxtresrunmins,omitempty"`
|
||||||
|
Maxtresperjob TresList `json:"maxtresperjob,omitempty"`
|
||||||
|
Maxtrespernode TresList `json:"maxtrespernode,omitempty"`
|
||||||
|
Maxwalldurationperjob *Uint32NoVal `json:"maxwalldurationperjob,omitempty"`
|
||||||
|
Minpriothresh *Uint32NoVal `json:"minpriothresh,omitempty"`
|
||||||
|
Parent *string `json:"parent,omitempty"`
|
||||||
|
Priority *Uint32NoVal `json:"priority,omitempty"`
|
||||||
|
Qoslevel QosStringIdList `json:"qoslevel,omitempty"`
|
||||||
|
Fairshare *int32 `json:"fairshare,omitempty"`
|
||||||
|
}
|
||||||
816
internal/slurm/types_slurmdb_test.go
Normal file
816
internal/slurm/types_slurmdb_test.go
Normal file
@@ -0,0 +1,816 @@
|
|||||||
|
package slurm
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestSlurmdbJobsRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiSlurmdbdJobsResp{
|
||||||
|
Jobs: JobList{
|
||||||
|
{JobID: Ptr(int32(1)), Name: Ptr("test-job")},
|
||||||
|
{JobID: Ptr(int32(2)), Name: Ptr("other-job")},
|
||||||
|
},
|
||||||
|
Meta: &OpenapiMeta{},
|
||||||
|
Errors: OpenapiErrors{},
|
||||||
|
Warnings: OpenapiWarnings{},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiSlurmdbdJobsResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.Jobs) != 2 {
|
||||||
|
t.Fatalf("expected 2 jobs, got %d", len(decoded.Jobs))
|
||||||
|
}
|
||||||
|
if *decoded.Jobs[0].JobID != 1 {
|
||||||
|
t.Fatal("first job id mismatch")
|
||||||
|
}
|
||||||
|
if *decoded.Jobs[1].Name != "other-job" {
|
||||||
|
t.Fatal("second job name mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbConfigRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiSlurmdbdConfigResp{
|
||||||
|
Clusters: ClusterRecList{
|
||||||
|
{Name: Ptr("cluster1")},
|
||||||
|
},
|
||||||
|
Tres: TresList{
|
||||||
|
{Type: Ptr("cpu"), ID: Ptr(int32(1))},
|
||||||
|
},
|
||||||
|
Accounts: AccountList{
|
||||||
|
{Name: Ptr("acct1")},
|
||||||
|
},
|
||||||
|
Users: UserList{
|
||||||
|
{Name: Ptr("user1")},
|
||||||
|
},
|
||||||
|
Qos: QosList{
|
||||||
|
{Name: Ptr("normal")},
|
||||||
|
},
|
||||||
|
Wckeys: WckeyList{
|
||||||
|
{Name: Ptr("wckey1")},
|
||||||
|
},
|
||||||
|
Associations: AssocList{
|
||||||
|
{Account: Ptr("acct1"), User: Ptr("user1")},
|
||||||
|
},
|
||||||
|
Instances: InstanceList{
|
||||||
|
{InstanceID: Ptr("i-123")},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify lowercase "tres" key in JSON
|
||||||
|
raw := string(data)
|
||||||
|
if strings.Contains(raw, `"TRES"`) {
|
||||||
|
t.Fatal("OpenapiSlurmdbdConfigResp should use lowercase 'tres', not uppercase 'TRES'")
|
||||||
|
}
|
||||||
|
if !strings.Contains(raw, `"tres"`) {
|
||||||
|
t.Fatal("OpenapiSlurmdbdConfigResp should contain lowercase 'tres' key")
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiSlurmdbdConfigResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.Clusters) != 1 || *decoded.Clusters[0].Name != "cluster1" {
|
||||||
|
t.Fatal("clusters mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Tres) != 1 || *decoded.Tres[0].Type != "cpu" {
|
||||||
|
t.Fatal("tres mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Accounts) != 1 || *decoded.Accounts[0].Name != "acct1" {
|
||||||
|
t.Fatal("accounts mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Users) != 1 || *decoded.Users[0].Name != "user1" {
|
||||||
|
t.Fatal("users mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Qos) != 1 || *decoded.Qos[0].Name != "normal" {
|
||||||
|
t.Fatal("qos mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Wckeys) != 1 || *decoded.Wckeys[0].Name != "wckey1" {
|
||||||
|
t.Fatal("wckeys mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Associations) != 1 || *decoded.Associations[0].Account != "acct1" {
|
||||||
|
t.Fatal("associations mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Instances) != 1 || *decoded.Instances[0].InstanceID != "i-123" {
|
||||||
|
t.Fatal("instances mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbQosRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiSlurmdbdQosResp{
|
||||||
|
Qos: QosList{
|
||||||
|
{Name: Ptr("high"), ID: Ptr(int32(1))},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiSlurmdbdQosResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.Qos) != 1 || *decoded.Qos[0].Name != "high" {
|
||||||
|
t.Fatal("qos mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbQosRemovedRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiSlurmdbdQosRemovedResp{
|
||||||
|
RemovedQos: StringList{"normal", "high"},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiSlurmdbdQosRemovedResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.RemovedQos) != 2 || decoded.RemovedQos[0] != "normal" {
|
||||||
|
t.Fatal("removed_qos mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbStatsRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiSlurmdbdStatsResp{
|
||||||
|
Statistics: &StatsRec{
|
||||||
|
TimeStart: Ptr(int64(1700000000)),
|
||||||
|
Rollups: RollupStatsList{
|
||||||
|
{Type: Ptr("internal"), MaxCycle: Ptr(int64(5))},
|
||||||
|
},
|
||||||
|
RPCs: StatsRpcList{
|
||||||
|
{RPC: Ptr("SLURMCTLD_GET_JOB_INFO"), Count: Ptr(int32(100))},
|
||||||
|
},
|
||||||
|
Users: StatsUserList{
|
||||||
|
{User: Ptr("root"), Count: Ptr(int32(50))},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiSlurmdbdStatsResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if decoded.Statistics == nil {
|
||||||
|
t.Fatal("statistics is nil")
|
||||||
|
}
|
||||||
|
if *decoded.Statistics.TimeStart != 1700000000 {
|
||||||
|
t.Fatal("time_start mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Statistics.Rollups) != 1 || *decoded.Statistics.Rollups[0].Type != "internal" {
|
||||||
|
t.Fatal("rollups mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Statistics.RPCs) != 1 || *decoded.Statistics.RPCs[0].RPC != "SLURMCTLD_GET_JOB_INFO" {
|
||||||
|
t.Fatal("RPCs mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Statistics.Users) != 1 || *decoded.Statistics.Users[0].User != "root" {
|
||||||
|
t.Fatal("users mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbAssocsRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiAssocsResp{
|
||||||
|
Associations: AssocList{
|
||||||
|
{Account: Ptr("acct1"), User: Ptr("user1")},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiAssocsResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.Associations) != 1 || *decoded.Associations[0].Account != "acct1" {
|
||||||
|
t.Fatal("associations mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbAssocsRemovedRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiAssocsRemovedResp{
|
||||||
|
RemovedAssociations: StringList{"acct1_user1_cluster1"},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiAssocsRemovedResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.RemovedAssociations) != 1 || decoded.RemovedAssociations[0] != "acct1_user1_cluster1" {
|
||||||
|
t.Fatal("removed_associations mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbUsersRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiUsersResp{
|
||||||
|
Users: UserList{
|
||||||
|
{Name: Ptr("admin"), AdministratorLevel: []string{"Administrator"}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiUsersResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.Users) != 1 || *decoded.Users[0].Name != "admin" {
|
||||||
|
t.Fatal("users mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbClustersRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiClustersResp{
|
||||||
|
Clusters: ClusterRecList{
|
||||||
|
{Name: Ptr("test-cluster"), Nodes: Ptr("node[1-10]")},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiClustersResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.Clusters) != 1 || *decoded.Clusters[0].Name != "test-cluster" {
|
||||||
|
t.Fatal("clusters mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbClustersRemovedRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiClustersRemovedResp{
|
||||||
|
DeletedClusters: StringList{"old-cluster"},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiClustersRemovedResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.DeletedClusters) != 1 || decoded.DeletedClusters[0] != "old-cluster" {
|
||||||
|
t.Fatal("deleted_clusters mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbWckeyRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiWckeyResp{
|
||||||
|
Wckeys: WckeyList{
|
||||||
|
{Name: Ptr("wckey1"), Cluster: Ptr("cluster1")},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiWckeyResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.Wckeys) != 1 || *decoded.Wckeys[0].Name != "wckey1" {
|
||||||
|
t.Fatal("wckeys mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbWckeyRemovedRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiWckeyRemovedResp{
|
||||||
|
DeletedWckeys: StringList{"old-wckey"},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiWckeyRemovedResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.DeletedWckeys) != 1 || decoded.DeletedWckeys[0] != "old-wckey" {
|
||||||
|
t.Fatal("deleted_wckeys mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbAccountsRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiAccountsResp{
|
||||||
|
Accounts: AccountList{
|
||||||
|
{Name: Ptr("science"), Description: Ptr("Science dept"), Organization: Ptr("org1")},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiAccountsResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.Accounts) != 1 || *decoded.Accounts[0].Name != "science" {
|
||||||
|
t.Fatal("accounts mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbAccountsRemovedRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiAccountsRemovedResp{
|
||||||
|
RemovedAccounts: StringList{"old-acct"},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiAccountsRemovedResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.RemovedAccounts) != 1 || decoded.RemovedAccounts[0] != "old-acct" {
|
||||||
|
t.Fatal("removed_accounts mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbInstancesRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiInstancesResp{
|
||||||
|
Instances: InstanceList{
|
||||||
|
{InstanceID: Ptr("i-abc123"), Cluster: Ptr("cluster1")},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiInstancesResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.Instances) != 1 || *decoded.Instances[0].InstanceID != "i-abc123" {
|
||||||
|
t.Fatal("instances mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbTresRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiTresResp{
|
||||||
|
TRES: TresList{
|
||||||
|
{Type: Ptr("cpu"), ID: Ptr(int32(1)), Count: Ptr(int64(100))},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify UPPERCASE "TRES" key in JSON
|
||||||
|
raw := string(data)
|
||||||
|
if !strings.Contains(raw, `"TRES"`) {
|
||||||
|
t.Fatalf("OpenapiTresResp should use UPPERCASE 'TRES' key, got: %s", raw)
|
||||||
|
}
|
||||||
|
if strings.Contains(raw, `"tres"`) {
|
||||||
|
t.Fatal("OpenapiTresResp should NOT contain lowercase 'tres'")
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiTresResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.TRES) != 1 || *decoded.TRES[0].Type != "cpu" {
|
||||||
|
t.Fatal("TRES mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbTresRespUppercaseKey(t *testing.T) {
|
||||||
|
raw := `{"TRES": [{"type": "mem", "id": 2, "count": 1000}]}`
|
||||||
|
var decoded OpenapiTresResp
|
||||||
|
if err := json.Unmarshal([]byte(raw), &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if len(decoded.TRES) != 1 || *decoded.TRES[0].Type != "mem" {
|
||||||
|
t.Fatal("TRES uppercase key parse failed")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbConfigRespLowercaseTresKey(t *testing.T) {
|
||||||
|
raw := `{"tres": [{"type": "cpu", "id": 1, "count": 100}]}`
|
||||||
|
var decoded OpenapiSlurmdbdConfigResp
|
||||||
|
if err := json.Unmarshal([]byte(raw), &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if len(decoded.Tres) != 1 || *decoded.Tres[0].Type != "cpu" {
|
||||||
|
t.Fatal("config resp lowercase tres key parse failed")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbUsersAddCondRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiUsersAddCondResp{
|
||||||
|
AssociationCondition: &UsersAddCond{
|
||||||
|
Accounts: StringList{"acct1", "acct2"},
|
||||||
|
Clusters: StringList{"cluster1"},
|
||||||
|
Users: StringList{"user1", "user2"},
|
||||||
|
Wckeys: StringList{"wckey1"},
|
||||||
|
Partitions: StringList{"partition1"},
|
||||||
|
},
|
||||||
|
User: &UserShort{
|
||||||
|
Adminlevel: []string{"Administrator"},
|
||||||
|
Defaultaccount: Ptr("acct1"),
|
||||||
|
Defaultwckey: Ptr("wckey1"),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiUsersAddCondResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if decoded.AssociationCondition == nil {
|
||||||
|
t.Fatal("association_condition is nil")
|
||||||
|
}
|
||||||
|
if len(decoded.AssociationCondition.Accounts) != 2 {
|
||||||
|
t.Fatal("accounts count mismatch")
|
||||||
|
}
|
||||||
|
if decoded.User == nil || *decoded.User.Defaultaccount != "acct1" {
|
||||||
|
t.Fatal("user mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbUsersAddCondRespStrRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiUsersAddCondRespStr{
|
||||||
|
AddedUsers: Ptr("user1,user2,user3"),
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiUsersAddCondRespStr
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if decoded.AddedUsers == nil || *decoded.AddedUsers != "user1,user2,user3" {
|
||||||
|
t.Fatal("added_users mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbAccountsAddCondRespRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiAccountsAddCondResp{
|
||||||
|
AssociationCondition: &AccountsAddCond{
|
||||||
|
Accounts: StringList{"acct1"},
|
||||||
|
Clusters: StringList{"cluster1"},
|
||||||
|
},
|
||||||
|
Account: &AccountShort{
|
||||||
|
Description: Ptr("Science department"),
|
||||||
|
Organization: Ptr("university"),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiAccountsAddCondResp
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if decoded.AssociationCondition == nil || len(decoded.AssociationCondition.Accounts) != 1 {
|
||||||
|
t.Fatal("association_condition mismatch")
|
||||||
|
}
|
||||||
|
if decoded.Account == nil || *decoded.Account.Description != "Science department" {
|
||||||
|
t.Fatal("account mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbAccountsAddCondRespStrRoundTrip(t *testing.T) {
|
||||||
|
orig := OpenapiAccountsAddCondRespStr{
|
||||||
|
AddedAccounts: Ptr("acct1,acct2"),
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded OpenapiAccountsAddCondRespStr
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if decoded.AddedAccounts == nil || *decoded.AddedAccounts != "acct1,acct2" {
|
||||||
|
t.Fatal("added_accounts mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbStatsRecRoundTrip(t *testing.T) {
|
||||||
|
orig := StatsRec{
|
||||||
|
TimeStart: Ptr(int64(1700000000)),
|
||||||
|
Rollups: RollupStatsList{
|
||||||
|
{
|
||||||
|
Type: Ptr("internal"),
|
||||||
|
LastRun: Ptr(int32(1700001000)),
|
||||||
|
MaxCycle: Ptr(int64(10)),
|
||||||
|
TotalTime: Ptr(int64(100)),
|
||||||
|
TotalCycles: Ptr(int64(50)),
|
||||||
|
MeanCycles: Ptr(int64(2)),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
RPCs: StatsRpcList{
|
||||||
|
{
|
||||||
|
RPC: Ptr("GET_JOBS"),
|
||||||
|
Count: Ptr(int32(500)),
|
||||||
|
Time: &StatsRpcTime{
|
||||||
|
Average: Ptr(int64(100)),
|
||||||
|
Total: Ptr(int64(50000)),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Users: StatsUserList{
|
||||||
|
{
|
||||||
|
User: Ptr("admin"),
|
||||||
|
Count: Ptr(int32(200)),
|
||||||
|
Time: &StatsUserTime{
|
||||||
|
Average: Ptr(int64(50)),
|
||||||
|
Total: Ptr(int64(10000)),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded StatsRec
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if *decoded.TimeStart != 1700000000 {
|
||||||
|
t.Fatal("time_start mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Rollups) != 1 || *decoded.Rollups[0].Type != "internal" {
|
||||||
|
t.Fatal("rollups mismatch")
|
||||||
|
}
|
||||||
|
if *decoded.Rollups[0].MaxCycle != 10 {
|
||||||
|
t.Fatal("max_cycle mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.RPCs) != 1 || *decoded.RPCs[0].RPC != "GET_JOBS" {
|
||||||
|
t.Fatal("RPCs mismatch")
|
||||||
|
}
|
||||||
|
if decoded.RPCs[0].Time == nil || *decoded.RPCs[0].Time.Average != 100 {
|
||||||
|
t.Fatal("RPC time.average mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Users) != 1 || *decoded.Users[0].User != "admin" {
|
||||||
|
t.Fatal("users mismatch")
|
||||||
|
}
|
||||||
|
if decoded.Users[0].Time == nil || *decoded.Users[0].Time.Total != 10000 {
|
||||||
|
t.Fatal("user time.total mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbStatsRecRPCsUppercaseKey(t *testing.T) {
|
||||||
|
raw := `{"time_start": 1700000000, "RPCs": [{"rpc": "TEST", "count": 1}]}`
|
||||||
|
var decoded StatsRec
|
||||||
|
if err := json.Unmarshal([]byte(raw), &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if len(decoded.RPCs) != 1 || *decoded.RPCs[0].RPC != "TEST" {
|
||||||
|
t.Fatal("RPCs uppercase key parse failed")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbRollupStatsRoundTrip(t *testing.T) {
|
||||||
|
orig := RollupStats{
|
||||||
|
Type: Ptr("user"),
|
||||||
|
LastRun: Ptr(int32(1700002000)),
|
||||||
|
MaxCycle: Ptr(int64(20)),
|
||||||
|
TotalTime: Ptr(int64(200)),
|
||||||
|
TotalCycles: Ptr(int64(100)),
|
||||||
|
MeanCycles: Ptr(int64(2)),
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded RollupStats
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if *decoded.Type != "user" {
|
||||||
|
t.Fatal("type mismatch")
|
||||||
|
}
|
||||||
|
if *decoded.LastRun != 1700002000 {
|
||||||
|
t.Fatal("last run mismatch")
|
||||||
|
}
|
||||||
|
if *decoded.TotalCycles != 100 {
|
||||||
|
t.Fatal("total_cycles mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbAssocRecSetRoundTrip(t *testing.T) {
|
||||||
|
orig := AssocRecSet{
|
||||||
|
Comment: Ptr("test comment"),
|
||||||
|
Defaultqos: Ptr("normal"),
|
||||||
|
Fairshare: Ptr(int32(100)),
|
||||||
|
Parent: Ptr("parent-acct"),
|
||||||
|
Grptres: TresList{{Type: Ptr("cpu"), Count: Ptr(int64(1000))}},
|
||||||
|
Maxtresperjob: TresList{{Type: Ptr("mem"), Count: Ptr(int64(4096))}},
|
||||||
|
Qoslevel: QosStringIdList{"high", "normal"},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded AssocRecSet
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if *decoded.Comment != "test comment" {
|
||||||
|
t.Fatal("comment mismatch")
|
||||||
|
}
|
||||||
|
if *decoded.Defaultqos != "normal" {
|
||||||
|
t.Fatal("defaultqos mismatch")
|
||||||
|
}
|
||||||
|
if *decoded.Fairshare != 100 {
|
||||||
|
t.Fatal("fairshare mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Grptres) != 1 || *decoded.Grptres[0].Type != "cpu" {
|
||||||
|
t.Fatal("grptres mismatch")
|
||||||
|
}
|
||||||
|
if len(decoded.Qoslevel) != 2 || decoded.Qoslevel[0] != "high" {
|
||||||
|
t.Fatal("qoslevel mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbUsersAddCondRoundTrip(t *testing.T) {
|
||||||
|
orig := UsersAddCond{
|
||||||
|
Accounts: StringList{"acct1"},
|
||||||
|
Clusters: StringList{"cluster1"},
|
||||||
|
Partitions: StringList{"part1"},
|
||||||
|
Users: StringList{"user1", "user2"},
|
||||||
|
Wckeys: StringList{"wckey1"},
|
||||||
|
Association: &AssocRecSet{
|
||||||
|
Comment: Ptr("assoc comment"),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded UsersAddCond
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.Users) != 2 {
|
||||||
|
t.Fatal("users count mismatch")
|
||||||
|
}
|
||||||
|
if decoded.Association == nil || *decoded.Association.Comment != "assoc comment" {
|
||||||
|
t.Fatal("association mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbAccountsAddCondRoundTrip(t *testing.T) {
|
||||||
|
orig := AccountsAddCond{
|
||||||
|
Accounts: StringList{"acct1", "acct2"},
|
||||||
|
Clusters: StringList{"cluster1"},
|
||||||
|
Association: &AssocRecSet{
|
||||||
|
Fairshare: Ptr(int32(50)),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.Marshal(orig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded AccountsAddCond
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(decoded.Accounts) != 2 {
|
||||||
|
t.Fatal("accounts count mismatch")
|
||||||
|
}
|
||||||
|
if decoded.Association == nil || *decoded.Association.Fairshare != 50 {
|
||||||
|
t.Fatal("association mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSlurmdbEmptyResponses(t *testing.T) {
|
||||||
|
types := []interface{}{
|
||||||
|
OpenapiSlurmdbdJobsResp{},
|
||||||
|
OpenapiSlurmdbdConfigResp{},
|
||||||
|
OpenapiSlurmdbdQosResp{},
|
||||||
|
OpenapiSlurmdbdQosRemovedResp{},
|
||||||
|
OpenapiSlurmdbdStatsResp{},
|
||||||
|
OpenapiAssocsResp{},
|
||||||
|
OpenapiAssocsRemovedResp{},
|
||||||
|
OpenapiUsersResp{},
|
||||||
|
OpenapiClustersResp{},
|
||||||
|
OpenapiClustersRemovedResp{},
|
||||||
|
OpenapiWckeyResp{},
|
||||||
|
OpenapiWckeyRemovedResp{},
|
||||||
|
OpenapiAccountsResp{},
|
||||||
|
OpenapiAccountsRemovedResp{},
|
||||||
|
OpenapiInstancesResp{},
|
||||||
|
OpenapiTresResp{},
|
||||||
|
OpenapiUsersAddCondResp{},
|
||||||
|
OpenapiUsersAddCondRespStr{},
|
||||||
|
OpenapiAccountsAddCondResp{},
|
||||||
|
OpenapiAccountsAddCondRespStr{},
|
||||||
|
StatsRec{},
|
||||||
|
RollupStats{},
|
||||||
|
StatsRpc{},
|
||||||
|
StatsUser{},
|
||||||
|
UsersAddCond{},
|
||||||
|
AccountsAddCond{},
|
||||||
|
AssocRecSet{},
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, v := range types {
|
||||||
|
data, err := json.Marshal(v)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("type %d: marshal error: %v", i, err)
|
||||||
|
}
|
||||||
|
if string(data) != "{}" {
|
||||||
|
t.Fatalf("type %d: empty struct should marshal to {}, got %s", i, data)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user