暫無描述
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

my_config_project_skill_test.go 7.9KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318
  1. package main
  2. import (
  3. "bytes"
  4. "encoding/json"
  5. "io"
  6. "net/http"
  7. "testing"
  8. "git.x2erp.com/qdy/go-base/model/request/configreq"
  9. )
  10. const (
  11. skillTestSkillID = "test-skill-001"
  12. skillTestProjectID = "test-project-skill-001"
  13. skillTestDescription = "测试Skill描述"
  14. skillTestContent = `{"type": "code_generation", "language": "go"}`
  15. skillUpdatedContent = `{"type": "code_review", "language": "python"}`
  16. skillBaseURL = "http://localhost:8080"
  17. )
  18. // TestProjectSkillCRUD 测试项目Skill的增删改查
  19. func TestProjectSkillCRUD(t *testing.T) {
  20. // 1. 创建项目Skill
  21. t.Run("CreateProjectSkill", testCreateProjectSkill)
  22. // 2. 查询单个项目Skill
  23. t.Run("GetProjectSkill", testGetProjectSkill)
  24. // 3. 查询项目Skill列表
  25. t.Run("ListProjectSkills", testListProjectSkills)
  26. // 4. 更新项目Skill
  27. t.Run("UpdateProjectSkill", testUpdateProjectSkill)
  28. // 5. 查询更新后的项目Skill
  29. t.Run("GetUpdatedProjectSkill", testGetUpdatedProjectSkill)
  30. // 6. 删除项目Skill
  31. t.Run("DeleteProjectSkill", testDeleteProjectSkill)
  32. // 7. 验证项目Skill已删除
  33. t.Run("VerifyProjectSkillDeleted", testVerifyProjectSkillDeleted)
  34. }
  35. func testCreateProjectSkill(t *testing.T) {
  36. httpClient := &http.Client{}
  37. url := skillBaseURL + "/api/create/config/project/skill"
  38. reqBody := configreq.ProjectSkillRequest{
  39. SkillID: skillTestSkillID,
  40. ProjectID: skillTestProjectID,
  41. Description: skillTestDescription,
  42. Content: skillTestContent,
  43. }
  44. jsonData, err := json.Marshal(reqBody)
  45. if err != nil {
  46. t.Fatalf("JSON序列化失败: %v", err)
  47. }
  48. req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonData))
  49. if err != nil {
  50. t.Fatalf("创建请求失败: %v", err)
  51. }
  52. // 使用Basic认证
  53. req.SetBasicAuth("admin", "123")
  54. req.Header.Set("Content-Type", "application/json")
  55. resp, err := httpClient.Do(req)
  56. if err != nil {
  57. t.Fatalf("请求失败: %v", err)
  58. }
  59. defer resp.Body.Close()
  60. body, err := io.ReadAll(resp.Body)
  61. if err != nil {
  62. t.Fatalf("读取响应失败: %v", err)
  63. }
  64. var result map[string]interface{}
  65. if err := json.Unmarshal(body, &result); err != nil {
  66. t.Fatalf("JSON解析失败: %v", err)
  67. }
  68. if success, ok := result["success"].(bool); !ok || !success {
  69. t.Errorf("创建项目Skill失败: %v", result)
  70. }
  71. t.Logf("创建项目Skill成功: %s", string(body))
  72. }
  73. func testGetProjectSkill(t *testing.T) {
  74. httpClient := &http.Client{}
  75. url := skillBaseURL + "/api/query/config/project/skill/" + skillTestSkillID
  76. req, err := http.NewRequest("POST", url, nil)
  77. if err != nil {
  78. t.Fatalf("创建请求失败: %v", err)
  79. }
  80. req.SetBasicAuth("admin", "123")
  81. req.Header.Set("Content-Type", "application/json")
  82. resp, err := httpClient.Do(req)
  83. if err != nil {
  84. t.Fatalf("请求失败: %v", err)
  85. }
  86. defer resp.Body.Close()
  87. body, err := io.ReadAll(resp.Body)
  88. if err != nil {
  89. t.Fatalf("读取响应失败: %v", err)
  90. }
  91. var result map[string]interface{}
  92. if err := json.Unmarshal(body, &result); err != nil {
  93. t.Fatalf("JSON解析失败: %v", err)
  94. }
  95. if success, ok := result["success"].(bool); !ok || !success {
  96. t.Errorf("查询项目Skill失败: %v", result)
  97. }
  98. t.Logf("查询项目Skill成功: %s", string(body))
  99. }
  100. func testListProjectSkills(t *testing.T) {
  101. httpClient := &http.Client{}
  102. url := skillBaseURL + "/api/query/config/project/skills"
  103. req, err := http.NewRequest("POST", url, nil)
  104. if err != nil {
  105. t.Fatalf("创建请求失败: %v", err)
  106. }
  107. req.SetBasicAuth("admin", "123")
  108. req.Header.Set("Content-Type", "application/json")
  109. resp, err := httpClient.Do(req)
  110. if err != nil {
  111. t.Fatalf("请求失败: %v", err)
  112. }
  113. defer resp.Body.Close()
  114. body, err := io.ReadAll(resp.Body)
  115. if err != nil {
  116. t.Fatalf("读取响应失败: %v", err)
  117. }
  118. var result map[string]interface{}
  119. if err := json.Unmarshal(body, &result); err != nil {
  120. t.Fatalf("JSON解析失败: %v", err)
  121. }
  122. if success, ok := result["success"].(bool); !ok || !success {
  123. t.Errorf("查询项目Skill列表失败: %v", result)
  124. }
  125. t.Logf("查询项目Skill列表成功: %s", string(body))
  126. }
  127. func testUpdateProjectSkill(t *testing.T) {
  128. httpClient := &http.Client{}
  129. url := skillBaseURL + "/api/update/config/project/skill/" + skillTestSkillID
  130. reqBody := configreq.ProjectSkillRequest{
  131. SkillID: skillTestSkillID,
  132. ProjectID: skillTestProjectID,
  133. Description: "更新后的Skill描述",
  134. Content: skillUpdatedContent,
  135. }
  136. jsonData, err := json.Marshal(reqBody)
  137. if err != nil {
  138. t.Fatalf("JSON序列化失败: %v", err)
  139. }
  140. req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonData))
  141. if err != nil {
  142. t.Fatalf("创建请求失败: %v", err)
  143. }
  144. req.SetBasicAuth("admin", "123")
  145. req.Header.Set("Content-Type", "application/json")
  146. resp, err := httpClient.Do(req)
  147. if err != nil {
  148. t.Fatalf("请求失败: %v", err)
  149. }
  150. defer resp.Body.Close()
  151. body, err := io.ReadAll(resp.Body)
  152. if err != nil {
  153. t.Fatalf("读取响应失败: %v", err)
  154. }
  155. var result map[string]interface{}
  156. if err := json.Unmarshal(body, &result); err != nil {
  157. t.Fatalf("JSON解析失败: %v", err)
  158. }
  159. if success, ok := result["success"].(bool); !ok || !success {
  160. t.Errorf("更新项目Skill失败: %v", result)
  161. }
  162. t.Logf("更新项目Skill成功: %s", string(body))
  163. }
  164. func testGetUpdatedProjectSkill(t *testing.T) {
  165. httpClient := &http.Client{}
  166. url := skillBaseURL + "/api/query/config/project/skill/" + skillTestSkillID
  167. req, err := http.NewRequest("POST", url, nil)
  168. if err != nil {
  169. t.Fatalf("创建请求失败: %v", err)
  170. }
  171. req.SetBasicAuth("admin", "123")
  172. req.Header.Set("Content-Type", "application/json")
  173. resp, err := httpClient.Do(req)
  174. if err != nil {
  175. t.Fatalf("请求失败: %v", err)
  176. }
  177. defer resp.Body.Close()
  178. body, err := io.ReadAll(resp.Body)
  179. if err != nil {
  180. t.Fatalf("读取响应失败: %v", err)
  181. }
  182. var result map[string]interface{}
  183. if err := json.Unmarshal(body, &result); err != nil {
  184. t.Fatalf("JSON解析失败: %v", err)
  185. }
  186. if success, ok := result["success"].(bool); !ok || !success {
  187. t.Errorf("查询更新后项目Skill失败: %v", result)
  188. }
  189. t.Logf("查询更新后项目Skill成功: %s", string(body))
  190. }
  191. func testDeleteProjectSkill(t *testing.T) {
  192. httpClient := &http.Client{}
  193. url := skillBaseURL + "/api/delete/config/project/skill/" + skillTestSkillID
  194. req, err := http.NewRequest("POST", url, nil)
  195. if err != nil {
  196. t.Fatalf("创建请求失败: %v", err)
  197. }
  198. req.SetBasicAuth("admin", "123")
  199. req.Header.Set("Content-Type", "application/json")
  200. resp, err := httpClient.Do(req)
  201. if err != nil {
  202. t.Fatalf("请求失败: %v", err)
  203. }
  204. defer resp.Body.Close()
  205. body, err := io.ReadAll(resp.Body)
  206. if err != nil {
  207. t.Fatalf("读取响应失败: %v", err)
  208. }
  209. var result map[string]interface{}
  210. if err := json.Unmarshal(body, &result); err != nil {
  211. t.Fatalf("JSON解析失败: %v", err)
  212. }
  213. if success, ok := result["success"].(bool); !ok || !success {
  214. t.Errorf("删除项目Skill失败: %v", result)
  215. }
  216. t.Logf("删除项目Skill成功: %s", string(body))
  217. }
  218. func testVerifyProjectSkillDeleted(t *testing.T) {
  219. httpClient := &http.Client{}
  220. url := skillBaseURL + "/api/query/config/project/skill/" + skillTestSkillID
  221. req, err := http.NewRequest("POST", url, nil)
  222. if err != nil {
  223. t.Fatalf("创建请求失败: %v", err)
  224. }
  225. req.SetBasicAuth("admin", "123")
  226. req.Header.Set("Content-Type", "application/json")
  227. resp, err := httpClient.Do(req)
  228. if err != nil {
  229. t.Fatalf("请求失败: %v", err)
  230. }
  231. defer resp.Body.Close()
  232. body, err := io.ReadAll(resp.Body)
  233. if err != nil {
  234. t.Fatalf("读取响应失败: %v", err)
  235. }
  236. var result map[string]interface{}
  237. if err := json.Unmarshal(body, &result); err != nil {
  238. t.Fatalf("JSON解析失败: %v", err)
  239. }
  240. // 删除后查询应该返回失败或空数据
  241. if success, ok := result["success"].(bool); ok && success {
  242. // 如果成功,检查数据是否为空
  243. if data, ok := result["data"].(map[string]interface{}); ok && data != nil {
  244. t.Errorf("项目Skill应该已删除但仍有数据: %v", result)
  245. }
  246. }
  247. t.Logf("验证项目Skill已删除: %s", string(body))
  248. }