feat(slurmdb): add UsersService

Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
2026-04-08 21:34:18 +08:00
parent f64e0dd8af
commit acd1e175bd
2 changed files with 315 additions and 0 deletions
--- a/internal/slurm/slurmdb_users_test.go
+++ b/internal/slurm/slurmdb_users_test.go
@@ -0,0 +1,160 @@
+package slurm
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+)
+
+func TestSlurmdbUsersService_GetUsers(t *testing.T) {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/slurmdb/v0.0.40/users", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "GET")
+		fmt.Fprint(w, `{"users": [{"name": "testuser"}]}`)
+	})
+	server := httptest.NewServer(mux)
+	defer server.Close()
+
+	client, _ := NewClient(server.URL, nil)
+	resp, _, err := client.SlurmdbUsers.GetUsers(context.Background(), nil)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if resp == nil {
+		t.Fatal("expected non-nil response")
+	}
+}
+
+func TestSlurmdbUsersService_GetUsersWithOptions(t *testing.T) {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/slurmdb/v0.0.40/users", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "GET")
+		if r.URL.Query().Get("with_deleted") != "true" {
+			t.Error("expected with_deleted=true query param")
+		}
+		if r.URL.Query().Get("with_assocs") != "true" {
+			t.Error("expected with_assocs=true query param")
+		}
+		fmt.Fprint(w, `{"users": [{"name": "testuser"}]}`)
+	})
+	server := httptest.NewServer(mux)
+	defer server.Close()
+
+	client, _ := NewClient(server.URL, nil)
+	deleted := "true"
+	assocs := "true"
+	opts := &GetUsersOptions{WithDeleted: &deleted, WithAssocs: &assocs}
+	resp, _, err := client.SlurmdbUsers.GetUsers(context.Background(), opts)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if resp == nil {
+		t.Fatal("expected non-nil response")
+	}
+}
+
+func TestSlurmdbUsersService_GetUser(t *testing.T) {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/slurmdb/v0.0.40/user/testuser", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "GET")
+		fmt.Fprint(w, `{"users": [{"name": "testuser"}]}`)
+	})
+	server := httptest.NewServer(mux)
+	defer server.Close()
+
+	client, _ := NewClient(server.URL, nil)
+	resp, _, err := client.SlurmdbUsers.GetUser(context.Background(), "testuser", nil)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if resp == nil {
+		t.Fatal("expected non-nil response")
+	}
+}
+
+func TestSlurmdbUsersService_PostUsers(t *testing.T) {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/slurmdb/v0.0.40/users", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "POST")
+		fmt.Fprint(w, `{"users": [{"name": "testuser"}]}`)
+	})
+	server := httptest.NewServer(mux)
+	defer server.Close()
+
+	client, _ := NewClient(server.URL, nil)
+	body := &OpenapiUsersResp{}
+	resp, _, err := client.SlurmdbUsers.PostUsers(context.Background(), body)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if resp == nil {
+		t.Fatal("expected non-nil response")
+	}
+}
+
+func TestSlurmdbUsersService_PostUsersAssociation(t *testing.T) {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/slurmdb/v0.0.40/users_association", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "POST")
+		fmt.Fprint(w, `{"added_users": "user1,user2"}`)
+	})
+	server := httptest.NewServer(mux)
+	defer server.Close()
+
+	client, _ := NewClient(server.URL, nil)
+	body := &OpenapiUsersAddCondResp{}
+	resp, _, err := client.SlurmdbUsers.PostUsersAssociation(context.Background(), body)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if resp == nil {
+		t.Fatal("expected non-nil response")
+	}
+	if resp.AddedUsers == nil {
+		t.Fatal("expected non-nil AddedUsers")
+	}
+	if *resp.AddedUsers != "user1,user2" {
+		t.Errorf("expected added_users 'user1,user2', got %s", *resp.AddedUsers)
+	}
+}
+
+func TestSlurmdbUsersService_DeleteUser(t *testing.T) {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/slurmdb/v0.0.40/user/testuser", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "DELETE")
+		fmt.Fprint(w, `{"meta": {"plugin": {"type": "openapi/v0.0.40"}}}`)
+	})
+	server := httptest.NewServer(mux)
+	defer server.Close()
+
+	client, _ := NewClient(server.URL, nil)
+	resp, _, err := client.SlurmdbUsers.DeleteUser(context.Background(), "testuser")
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if resp == nil {
+		t.Fatal("expected non-nil response")
+	}
+}
+
+func TestSlurmdbUsersService_GetUsers_Error(t *testing.T) {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/slurmdb/v0.0.40/users", func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusInternalServerError)
+		fmt.Fprint(w, `{"errors": [{"error": "internal error"}]}`)
+	})
+	server := httptest.NewServer(mux)
+	defer server.Close()
+
+	client, _ := NewClient(server.URL, nil)
+	_, _, err := client.SlurmdbUsers.GetUsers(context.Background(), nil)
+	if err == nil {
+		t.Fatal("expected error for 500 response")
+	}
+	if !strings.Contains(err.Error(), "500") {
+		t.Errorf("expected error to contain 500, got %v", err)
+	}
+}