userLib_test.go 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122
  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(10),
  57. AppId:int64(10000),
  58. })
  59. if err != nil {
  60. log.Println("UserDoc Collect call error:", err)
  61. }
  62. log.Print("UserDoc Collect call response:", resp)
  63. }
  64. //删除找回
  65. func Test_UserDocRestore(t *testing.T) {
  66. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  67. Etcd: discov.EtcdConf{
  68. Hosts: []string{"127.0.0.1:2380"},
  69. Key: "jydocs.userlib.rpc",
  70. },
  71. })
  72. userLib := userlibclient.NewUserLib(client)
  73. resp, err := userLib.DocRestore(context.Background(), &userlibclient.UserRecoveryRequest{
  74. UserDocId: int32(10),
  75. AppId:int64(10000),
  76. })
  77. if err != nil {
  78. log.Println("UserDoc Collect call error:", err)
  79. }
  80. log.Print("UserDoc Collect call response:", resp)
  81. }
  82. //永久删除
  83. func Test_UserDocPermanentDelete(t *testing.T) {
  84. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  85. Etcd: discov.EtcdConf{
  86. Hosts: []string{"127.0.0.1:2380"},
  87. Key: "jydocs.userlib.rpc",
  88. },
  89. })
  90. userLib := userlibclient.NewUserLib(client)
  91. resp, err := userLib.DocPermanentDelete(context.Background(), &userlibclient.UserRecoveryRequest{
  92. UserDocId: int32(10),
  93. AppId:int64(10000),
  94. })
  95. if err != nil {
  96. log.Println("UserDoc Collect call error:", err)
  97. }
  98. log.Print("UserDoc Collect call response:", resp)
  99. }
  100. //用户转存单元测试
  101. func Test_UserDocDwnload(t *testing.T) {
  102. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  103. Etcd: discov.EtcdConf{
  104. Hosts: []string{"127.0.0.1:2380"},
  105. Key: "jydocs.userlib.rpc",
  106. },
  107. })
  108. userLib := userlibclient.NewUserLib(client)
  109. resp, err := userLib.DocDownload(context.Background(), &userlibclient.UserCollectRequest{
  110. DocId: "123",
  111. UserId: "wh5",
  112. Cost:1000,
  113. AppId:int64(10000),
  114. })
  115. if err != nil {
  116. log.Println("UserDoc Collect call error:", err)
  117. }
  118. log.Print("UserDoc Collect call response:", resp.Message,resp.Code)
  119. }