userLib_test.go 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146
  1. package test
  2. import (
  3. "app.yhyue.com/moapp/jy_docs/rpc/userlib/userlibclient"
  4. "context"
  5. "github.com/tal-tech/go-zero/core/discov"
  6. "github.com/tal-tech/go-zero/zrpc"
  7. "log"
  8. "testing"
  9. )
  10. //用户收藏单元测试
  11. func Test_UserDocCollect(t *testing.T) {
  12. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  13. Etcd: discov.EtcdConf{
  14. Hosts: []string{"127.0.0.1:2379"},
  15. Key: "jydocs.userlib.rpc",
  16. },
  17. })
  18. userLib := userlibclient.NewUserLib(client)
  19. resp, err := userLib.DocCollect(context.Background(), &userlibclient.UserCollectRequest{
  20. DocId: "1",
  21. UserId: "1",
  22. Cost: 0,
  23. })
  24. if err != nil {
  25. log.Println("UserDoc Collect call error:", err)
  26. }
  27. log.Print("UserDoc Collect call response:", resp)
  28. }
  29. //用户取消收藏单元测试
  30. func Test_UserCancelDocCollect(t *testing.T) {
  31. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  32. Etcd: discov.EtcdConf{
  33. Hosts: []string{"127.0.0.1:2379"},
  34. Key: "jydocs.userlib.rpc",
  35. },
  36. })
  37. userLib := userlibclient.NewUserLib(client)
  38. resp, err := userLib.DocCancelCollect(context.Background(), &userlibclient.UserDeleteRequest{
  39. DocId: "7",
  40. })
  41. if err != nil {
  42. log.Println("UserDoc Collect call error:", err)
  43. }
  44. log.Print("UserDoc Collect call response:", resp)
  45. }
  46. //下载删除
  47. func Test_UserDocDelete(t *testing.T) {
  48. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  49. Etcd: discov.EtcdConf{
  50. Hosts: []string{"127.0.0.1:2380"},
  51. Key: "jydocs.userlib.rpc",
  52. },
  53. })
  54. userLib := userlibclient.NewUserLib(client)
  55. resp, err := userLib.DocDelete(context.Background(), &userlibclient.UserRecoveryRequest{
  56. UserDocId: int32(13),
  57. AppId:int64(10000),
  58. UserId: "wh11",
  59. })
  60. if err != nil {
  61. log.Println("UserDoc Collect call error:", err)
  62. }
  63. log.Print("UserDoc Collect call response:", resp)
  64. }
  65. //删除找回
  66. func Test_UserDocRestore(t *testing.T) {
  67. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  68. Etcd: discov.EtcdConf{
  69. Hosts: []string{"127.0.0.1:2380"},
  70. Key: "jydocs.userlib.rpc",
  71. },
  72. })
  73. userLib := userlibclient.NewUserLib(client)
  74. resp, err := userLib.DocRestore(context.Background(), &userlibclient.UserRecoveryRequest{
  75. UserDocId: int32(10),
  76. AppId:int64(10000),
  77. UserId: "wh11",
  78. })
  79. if err != nil {
  80. log.Println("UserDoc Collect call error:", err)
  81. }
  82. log.Print("UserDoc Collect call response:", resp)
  83. }
  84. //永久删除
  85. func Test_UserDocPermanentDelete(t *testing.T) {
  86. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  87. Etcd: discov.EtcdConf{
  88. Hosts: []string{"127.0.0.1:2380"},
  89. Key: "jydocs.userlib.rpc",
  90. },
  91. })
  92. userLib := userlibclient.NewUserLib(client)
  93. resp, err := userLib.DocPermanentDelete(context.Background(), &userlibclient.UserRecoveryRequest{
  94. UserDocId: int32(10),
  95. AppId:int64(10000),
  96. UserId: "wh11",
  97. })
  98. if err != nil {
  99. log.Println("UserDoc Collect call error:", err)
  100. }
  101. log.Print("UserDoc Collect call response:", resp)
  102. }
  103. //我的文库0兑换的 1收藏的 2回收站的
  104. func Test_UserDocList(t *testing.T) {
  105. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  106. Etcd: discov.EtcdConf{
  107. Hosts: []string{"127.0.0.1:2380"},
  108. Key: "jydocs.userlib.rpc",
  109. },
  110. })
  111. userLib := userlibclient.NewUserLib(client)
  112. resp, err := userLib.UserDocs(context.Background(), &userlibclient.UserDocsRequest{
  113. UserId: "wh11",
  114. AppId:int64(10000),
  115. PageSize:2,
  116. Page:1,
  117. UserDocCategory:2,
  118. })
  119. if err != nil {
  120. log.Println("UserDoc Collect call error:", err)
  121. }
  122. log.Print("UserDoc Collect call response:", resp)
  123. }
  124. //用户转存单元测试
  125. func Test_UserDocDwnload(t *testing.T) {
  126. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  127. Etcd: discov.EtcdConf{
  128. Hosts: []string{"127.0.0.1:2380"},
  129. Key: "jydocs.userlib.rpc",
  130. },
  131. })
  132. userLib := userlibclient.NewUserLib(client)
  133. resp, err := userLib.DocDownload(context.Background(), &userlibclient.UserCollectRequest{
  134. DocId: "123",
  135. UserId: "wh11",
  136. Cost:1000,
  137. AppId:int64(10000),
  138. })
  139. if err != nil {
  140. log.Println("UserDoc Collect call error:", err)
  141. }
  142. log.Print("UserDoc Collect call response:", resp.Message,resp.Code)
  143. }