server_account_test.go 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781
  1. package server
  2. import (
  3. "fmt"
  4. "github.com/stretchr/testify/require"
  5. "heckel.io/ntfy/v2/log"
  6. "heckel.io/ntfy/v2/user"
  7. "heckel.io/ntfy/v2/util"
  8. "io"
  9. "net/netip"
  10. "path/filepath"
  11. "strings"
  12. "testing"
  13. "time"
  14. )
  15. func TestAccount_Signup_Success(t *testing.T) {
  16. conf := newTestConfigWithAuthFile(t)
  17. conf.EnableSignup = true
  18. s := newTestServer(t, conf)
  19. defer s.closeDatabases()
  20. rr := request(t, s, "POST", "/v1/account", `{"username":"phil", "password":"mypass"}`, nil)
  21. require.Equal(t, 200, rr.Code)
  22. rr = request(t, s, "POST", "/v1/account/token", "", map[string]string{
  23. "Authorization": util.BasicAuth("phil", "mypass"),
  24. })
  25. require.Equal(t, 200, rr.Code)
  26. token, _ := util.UnmarshalJSON[apiAccountTokenResponse](io.NopCloser(rr.Body))
  27. require.NotEmpty(t, token.Token)
  28. require.True(t, time.Now().Add(71*time.Hour).Unix() < token.Expires)
  29. require.True(t, strings.HasPrefix(token.Token, "tk_"))
  30. require.Equal(t, "9.9.9.9", token.LastOrigin)
  31. require.True(t, token.LastAccess > time.Now().Unix()-2)
  32. require.True(t, token.LastAccess < time.Now().Unix()+2)
  33. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  34. "Authorization": util.BearerAuth(token.Token),
  35. })
  36. require.Equal(t, 200, rr.Code)
  37. account, _ := util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  38. require.Equal(t, "phil", account.Username)
  39. require.Equal(t, "user", account.Role)
  40. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  41. "Authorization": util.BasicAuth("", token.Token), // We allow a fake basic auth to make curl-ing easier (curl -u :<token>)
  42. })
  43. require.Equal(t, 200, rr.Code)
  44. account, _ = util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  45. require.Equal(t, "phil", account.Username)
  46. }
  47. func TestAccount_Signup_UserExists(t *testing.T) {
  48. conf := newTestConfigWithAuthFile(t)
  49. conf.EnableSignup = true
  50. s := newTestServer(t, conf)
  51. defer s.closeDatabases()
  52. rr := request(t, s, "POST", "/v1/account", `{"username":"phil", "password":"mypass"}`, nil)
  53. require.Equal(t, 200, rr.Code)
  54. rr = request(t, s, "POST", "/v1/account", `{"username":"phil", "password":"mypass"}`, nil)
  55. require.Equal(t, 409, rr.Code)
  56. require.Equal(t, 40901, toHTTPError(t, rr.Body.String()).Code)
  57. }
  58. func TestAccount_Signup_LimitReached(t *testing.T) {
  59. conf := newTestConfigWithAuthFile(t)
  60. conf.EnableSignup = true
  61. s := newTestServer(t, conf)
  62. defer s.closeDatabases()
  63. for i := 0; i < 3; i++ {
  64. rr := request(t, s, "POST", "/v1/account", fmt.Sprintf(`{"username":"phil%d", "password":"mypass"}`, i), nil)
  65. require.Equal(t, 200, rr.Code)
  66. }
  67. rr := request(t, s, "POST", "/v1/account", `{"username":"thiswontwork", "password":"mypass"}`, nil)
  68. require.Equal(t, 429, rr.Code)
  69. require.Equal(t, 42906, toHTTPError(t, rr.Body.String()).Code)
  70. }
  71. func TestAccount_Signup_AsUser(t *testing.T) {
  72. conf := newTestConfigWithAuthFile(t)
  73. conf.EnableSignup = true
  74. s := newTestServer(t, conf)
  75. defer s.closeDatabases()
  76. log.Info("1")
  77. require.Nil(t, s.userManager.AddUser("phil", "phil", user.RoleAdmin, false))
  78. log.Info("2")
  79. require.Nil(t, s.userManager.AddUser("ben", "ben", user.RoleUser, false))
  80. log.Info("3")
  81. rr := request(t, s, "POST", "/v1/account", `{"username":"emma", "password":"emma"}`, map[string]string{
  82. "Authorization": util.BasicAuth("phil", "phil"),
  83. })
  84. require.Equal(t, 200, rr.Code)
  85. log.Info("4")
  86. rr = request(t, s, "POST", "/v1/account", `{"username":"marian", "password":"marian"}`, map[string]string{
  87. "Authorization": util.BasicAuth("ben", "ben"),
  88. })
  89. require.Equal(t, 401, rr.Code)
  90. }
  91. func TestAccount_Signup_Disabled(t *testing.T) {
  92. conf := newTestConfigWithAuthFile(t)
  93. conf.EnableSignup = false
  94. s := newTestServer(t, conf)
  95. defer s.closeDatabases()
  96. rr := request(t, s, "POST", "/v1/account", `{"username":"phil", "password":"mypass"}`, nil)
  97. require.Equal(t, 400, rr.Code)
  98. require.Equal(t, 40022, toHTTPError(t, rr.Body.String()).Code)
  99. }
  100. func TestAccount_Signup_Rate_Limit(t *testing.T) {
  101. conf := newTestConfigWithAuthFile(t)
  102. conf.EnableSignup = true
  103. s := newTestServer(t, conf)
  104. for i := 0; i < 3; i++ {
  105. rr := request(t, s, "POST", "/v1/account", fmt.Sprintf(`{"username":"phil%d", "password":"mypass"}`, i), nil)
  106. require.Equal(t, 200, rr.Code, "failed on iteration %d", i)
  107. }
  108. rr := request(t, s, "POST", "/v1/account", `{"username":"notallowed", "password":"mypass"}`, nil)
  109. require.Equal(t, 429, rr.Code)
  110. require.Equal(t, 42906, toHTTPError(t, rr.Body.String()).Code)
  111. }
  112. func TestAccount_Get_Anonymous(t *testing.T) {
  113. conf := newTestConfigWithAuthFile(t)
  114. conf.VisitorRequestLimitReplenish = 86 * time.Second
  115. conf.VisitorEmailLimitReplenish = time.Hour
  116. conf.VisitorAttachmentTotalSizeLimit = 5123
  117. conf.AttachmentFileSizeLimit = 512
  118. s := newTestServer(t, conf)
  119. s.smtpSender = &testMailer{}
  120. defer s.closeDatabases()
  121. rr := request(t, s, "GET", "/v1/account", "", nil)
  122. require.Equal(t, 200, rr.Code)
  123. account, _ := util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  124. require.Equal(t, "*", account.Username)
  125. require.Equal(t, string(user.RoleAnonymous), account.Role)
  126. require.Equal(t, "ip", account.Limits.Basis)
  127. require.Equal(t, int64(1004), account.Limits.Messages) // I hate this
  128. require.Equal(t, int64(24), account.Limits.Emails) // I hate this
  129. require.Equal(t, int64(5123), account.Limits.AttachmentTotalSize)
  130. require.Equal(t, int64(512), account.Limits.AttachmentFileSize)
  131. require.Equal(t, int64(0), account.Stats.Messages)
  132. require.Equal(t, int64(1004), account.Stats.MessagesRemaining)
  133. require.Equal(t, int64(0), account.Stats.Emails)
  134. require.Equal(t, int64(24), account.Stats.EmailsRemaining)
  135. require.Equal(t, int64(0), account.Stats.Calls)
  136. require.Equal(t, int64(0), account.Stats.CallsRemaining)
  137. rr = request(t, s, "POST", "/mytopic", "", nil)
  138. require.Equal(t, 200, rr.Code)
  139. rr = request(t, s, "POST", "/mytopic", "", map[string]string{
  140. "Email": "phil@ntfy.sh",
  141. })
  142. require.Equal(t, 200, rr.Code)
  143. rr = request(t, s, "GET", "/v1/account", "", nil)
  144. require.Equal(t, 200, rr.Code)
  145. account, _ = util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  146. require.Equal(t, int64(2), account.Stats.Messages)
  147. require.Equal(t, int64(1002), account.Stats.MessagesRemaining)
  148. require.Equal(t, int64(1), account.Stats.Emails)
  149. require.Equal(t, int64(23), account.Stats.EmailsRemaining)
  150. }
  151. func TestAccount_ChangeSettings(t *testing.T) {
  152. s := newTestServer(t, newTestConfigWithAuthFile(t))
  153. defer s.closeDatabases()
  154. require.Nil(t, s.userManager.AddUser("phil", "phil", user.RoleUser, false))
  155. u, _ := s.userManager.User("phil")
  156. token, _ := s.userManager.CreateToken(u.ID, "", time.Unix(0, 0), netip.IPv4Unspecified(), false)
  157. rr := request(t, s, "PATCH", "/v1/account/settings", `{"notification": {"sound": "juntos"},"ignored": true}`, map[string]string{
  158. "Authorization": util.BasicAuth("phil", "phil"),
  159. })
  160. require.Equal(t, 200, rr.Code)
  161. rr = request(t, s, "PATCH", "/v1/account/settings", `{"notification": {"delete_after": 86400}, "language": "de"}`, map[string]string{
  162. "Authorization": util.BearerAuth(token.Value),
  163. })
  164. require.Equal(t, 200, rr.Code)
  165. rr = request(t, s, "GET", "/v1/account", `{"username":"marian", "password":"marian"}`, map[string]string{
  166. "Authorization": util.BearerAuth(token.Value),
  167. })
  168. require.Equal(t, 200, rr.Code)
  169. account, _ := util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  170. require.Equal(t, "de", account.Language)
  171. require.Equal(t, util.Int(86400), account.Notification.DeleteAfter)
  172. require.Equal(t, util.String("juntos"), account.Notification.Sound)
  173. require.Nil(t, account.Notification.MinPriority) // Not set
  174. }
  175. func TestAccount_Subscription_AddUpdateDelete(t *testing.T) {
  176. s := newTestServer(t, newTestConfigWithAuthFile(t))
  177. defer s.closeDatabases()
  178. require.Nil(t, s.userManager.AddUser("phil", "phil", user.RoleUser, false))
  179. rr := request(t, s, "POST", "/v1/account/subscription", `{"base_url": "http://abc.com", "topic": "def"}`, map[string]string{
  180. "Authorization": util.BasicAuth("phil", "phil"),
  181. })
  182. require.Equal(t, 200, rr.Code)
  183. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  184. "Authorization": util.BasicAuth("phil", "phil"),
  185. })
  186. require.Equal(t, 200, rr.Code)
  187. account, _ := util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  188. require.Equal(t, 1, len(account.Subscriptions))
  189. require.Equal(t, "http://abc.com", account.Subscriptions[0].BaseURL)
  190. require.Equal(t, "def", account.Subscriptions[0].Topic)
  191. require.Nil(t, account.Subscriptions[0].DisplayName)
  192. rr = request(t, s, "PATCH", "/v1/account/subscription", `{"base_url": "http://abc.com", "topic": "def", "display_name": "ding dong"}`, map[string]string{
  193. "Authorization": util.BasicAuth("phil", "phil"),
  194. })
  195. require.Equal(t, 200, rr.Code)
  196. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  197. "Authorization": util.BasicAuth("phil", "phil"),
  198. })
  199. require.Equal(t, 200, rr.Code)
  200. account, _ = util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  201. require.Equal(t, 1, len(account.Subscriptions))
  202. require.Equal(t, "http://abc.com", account.Subscriptions[0].BaseURL)
  203. require.Equal(t, "def", account.Subscriptions[0].Topic)
  204. require.Equal(t, util.String("ding dong"), account.Subscriptions[0].DisplayName)
  205. rr = request(t, s, "DELETE", "/v1/account/subscription", "", map[string]string{
  206. "Authorization": util.BasicAuth("phil", "phil"),
  207. "X-BaseURL": "http://abc.com",
  208. "X-Topic": "def",
  209. })
  210. require.Equal(t, 200, rr.Code)
  211. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  212. "Authorization": util.BasicAuth("phil", "phil"),
  213. })
  214. require.Equal(t, 200, rr.Code)
  215. account, _ = util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  216. require.Equal(t, 0, len(account.Subscriptions))
  217. }
  218. func TestAccount_ChangePassword(t *testing.T) {
  219. conf := newTestConfigWithAuthFile(t)
  220. conf.AuthUsers = []*user.User{
  221. {Name: "philuser", Hash: "$2a$10$U4WSIYY6evyGmZaraavM2e2JeVG6EMGUKN1uUwufUeeRd4Jpg6cGC", Role: user.RoleUser}, // philuser:philpass
  222. }
  223. s := newTestServer(t, conf)
  224. defer s.closeDatabases()
  225. require.Nil(t, s.userManager.AddUser("phil", "phil", user.RoleUser, false))
  226. rr := request(t, s, "POST", "/v1/account/password", `{"password": "WRONG", "new_password": ""}`, map[string]string{
  227. "Authorization": util.BasicAuth("phil", "phil"),
  228. })
  229. require.Equal(t, 400, rr.Code)
  230. rr = request(t, s, "POST", "/v1/account/password", `{"password": "WRONG", "new_password": "new password"}`, map[string]string{
  231. "Authorization": util.BasicAuth("phil", "phil"),
  232. })
  233. require.Equal(t, 400, rr.Code)
  234. require.Equal(t, 40026, toHTTPError(t, rr.Body.String()).Code)
  235. rr = request(t, s, "POST", "/v1/account/password", `{"password": "phil", "new_password": "new password"}`, map[string]string{
  236. "Authorization": util.BasicAuth("phil", "phil"),
  237. })
  238. require.Equal(t, 200, rr.Code)
  239. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  240. "Authorization": util.BasicAuth("phil", "phil"),
  241. })
  242. require.Equal(t, 401, rr.Code)
  243. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  244. "Authorization": util.BasicAuth("phil", "new password"),
  245. })
  246. require.Equal(t, 200, rr.Code)
  247. // Cannot change password of provisioned user
  248. rr = request(t, s, "POST", "/v1/account/password", `{"password": "philpass", "new_password": "new password"}`, map[string]string{
  249. "Authorization": util.BasicAuth("philuser", "philpass"),
  250. })
  251. require.Equal(t, 409, rr.Code)
  252. }
  253. func TestAccount_ChangePassword_NoAccount(t *testing.T) {
  254. s := newTestServer(t, newTestConfigWithAuthFile(t))
  255. defer s.closeDatabases()
  256. rr := request(t, s, "POST", "/v1/account/password", `{"password": "new password"}`, nil)
  257. require.Equal(t, 401, rr.Code)
  258. }
  259. func TestAccount_ExtendToken(t *testing.T) {
  260. t.Parallel()
  261. s := newTestServer(t, newTestConfigWithAuthFile(t))
  262. defer s.closeDatabases()
  263. require.Nil(t, s.userManager.AddUser("phil", "phil", user.RoleUser, false))
  264. rr := request(t, s, "POST", "/v1/account/token", "", map[string]string{
  265. "Authorization": util.BasicAuth("phil", "phil"),
  266. })
  267. require.Equal(t, 200, rr.Code)
  268. token, err := util.UnmarshalJSON[apiAccountTokenResponse](io.NopCloser(rr.Body))
  269. require.Nil(t, err)
  270. time.Sleep(time.Second)
  271. rr = request(t, s, "PATCH", "/v1/account/token", "", map[string]string{
  272. "Authorization": util.BearerAuth(token.Token),
  273. })
  274. require.Equal(t, 200, rr.Code)
  275. extendedToken, err := util.UnmarshalJSON[apiAccountTokenResponse](io.NopCloser(rr.Body))
  276. require.Nil(t, err)
  277. require.Equal(t, token.Token, extendedToken.Token)
  278. require.True(t, token.Expires < extendedToken.Expires)
  279. expires := time.Now().Add(999 * time.Hour)
  280. body := fmt.Sprintf(`{"token":"%s", "label":"some label", "expires": %d}`, token.Token, expires.Unix())
  281. rr = request(t, s, "PATCH", "/v1/account/token", body, map[string]string{
  282. "Authorization": util.BearerAuth(token.Token),
  283. })
  284. require.Equal(t, 200, rr.Code)
  285. token, err = util.UnmarshalJSON[apiAccountTokenResponse](io.NopCloser(rr.Body))
  286. require.Nil(t, err)
  287. require.Equal(t, "some label", token.Label)
  288. require.Equal(t, expires.Unix(), token.Expires)
  289. }
  290. func TestAccount_ExtendToken_NoTokenProvided(t *testing.T) {
  291. s := newTestServer(t, newTestConfigWithAuthFile(t))
  292. defer s.closeDatabases()
  293. require.Nil(t, s.userManager.AddUser("phil", "phil", user.RoleUser, false))
  294. rr := request(t, s, "PATCH", "/v1/account/token", "", map[string]string{
  295. "Authorization": util.BasicAuth("phil", "phil"), // Not Bearer!
  296. })
  297. require.Equal(t, 400, rr.Code)
  298. require.Equal(t, 40023, toHTTPError(t, rr.Body.String()).Code)
  299. }
  300. func TestAccount_DeleteToken(t *testing.T) {
  301. s := newTestServer(t, newTestConfigWithAuthFile(t))
  302. defer s.closeDatabases()
  303. require.Nil(t, s.userManager.AddUser("phil", "phil", user.RoleUser, false))
  304. rr := request(t, s, "POST", "/v1/account/token", "", map[string]string{
  305. "Authorization": util.BasicAuth("phil", "phil"),
  306. })
  307. require.Equal(t, 200, rr.Code)
  308. token, err := util.UnmarshalJSON[apiAccountTokenResponse](io.NopCloser(rr.Body))
  309. require.Nil(t, err)
  310. require.True(t, token.Expires > time.Now().Add(71*time.Hour).Unix())
  311. // Delete token failure (using basic auth)
  312. rr = request(t, s, "DELETE", "/v1/account/token", "", map[string]string{
  313. "Authorization": util.BasicAuth("phil", "phil"), // Not Bearer!
  314. })
  315. require.Equal(t, 400, rr.Code)
  316. require.Equal(t, 40023, toHTTPError(t, rr.Body.String()).Code)
  317. // Delete token with wrong token
  318. rr = request(t, s, "DELETE", "/v1/account/token", "", map[string]string{
  319. "Authorization": util.BearerAuth("invalidtoken"),
  320. })
  321. require.Equal(t, 401, rr.Code)
  322. // Delete token with correct token
  323. rr = request(t, s, "DELETE", "/v1/account/token", "", map[string]string{
  324. "Authorization": util.BearerAuth(token.Token),
  325. })
  326. require.Equal(t, 200, rr.Code)
  327. // Cannot get account anymore
  328. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  329. "Authorization": util.BearerAuth(token.Token),
  330. })
  331. require.Equal(t, 401, rr.Code)
  332. }
  333. func TestAccount_Delete_Success(t *testing.T) {
  334. conf := newTestConfigWithAuthFile(t)
  335. conf.EnableSignup = true
  336. s := newTestServer(t, conf)
  337. rr := request(t, s, "POST", "/v1/account", `{"username":"phil", "password":"mypass"}`, nil)
  338. require.Equal(t, 200, rr.Code)
  339. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  340. "Authorization": util.BasicAuth("phil", "mypass"),
  341. })
  342. require.Equal(t, 200, rr.Code)
  343. rr = request(t, s, "DELETE", "/v1/account", `{"password":"mypass"}`, map[string]string{
  344. "Authorization": util.BasicAuth("phil", "mypass"),
  345. })
  346. require.Equal(t, 200, rr.Code)
  347. // Account was marked deleted
  348. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  349. "Authorization": util.BasicAuth("phil", "mypass"),
  350. })
  351. require.Equal(t, 401, rr.Code)
  352. // Cannot re-create account, since still exists
  353. rr = request(t, s, "POST", "/v1/account", `{"username":"phil", "password":"mypass"}`, nil)
  354. require.Equal(t, 409, rr.Code)
  355. }
  356. func TestAccount_Delete_Not_Allowed(t *testing.T) {
  357. conf := newTestConfigWithAuthFile(t)
  358. conf.EnableSignup = true
  359. s := newTestServer(t, conf)
  360. rr := request(t, s, "POST", "/v1/account", `{"username":"phil", "password":"mypass"}`, nil)
  361. require.Equal(t, 200, rr.Code)
  362. rr = request(t, s, "DELETE", "/v1/account", "", nil)
  363. require.Equal(t, 401, rr.Code)
  364. rr = request(t, s, "DELETE", "/v1/account", `{"password":"mypass"}`, nil)
  365. require.Equal(t, 401, rr.Code)
  366. rr = request(t, s, "DELETE", "/v1/account", `{"password":"INCORRECT"}`, map[string]string{
  367. "Authorization": util.BasicAuth("phil", "mypass"),
  368. })
  369. require.Equal(t, 400, rr.Code)
  370. require.Equal(t, 40026, toHTTPError(t, rr.Body.String()).Code)
  371. }
  372. func TestAccount_Reservation_AddWithoutTierFails(t *testing.T) {
  373. conf := newTestConfigWithAuthFile(t)
  374. conf.EnableSignup = true
  375. s := newTestServer(t, conf)
  376. rr := request(t, s, "POST", "/v1/account", `{"username":"phil", "password":"mypass"}`, nil)
  377. require.Equal(t, 200, rr.Code)
  378. rr = request(t, s, "POST", "/v1/account/reservation", `{"topic":"mytopic", "everyone":"deny-all"}`, map[string]string{
  379. "Authorization": util.BasicAuth("phil", "mypass"),
  380. })
  381. require.Equal(t, 401, rr.Code)
  382. }
  383. func TestAccount_Reservation_AddAdminSuccess(t *testing.T) {
  384. conf := newTestConfigWithAuthFile(t)
  385. conf.EnableSignup = true
  386. s := newTestServer(t, conf)
  387. // A user, an admin, and a reservation walk into a bar
  388. require.Nil(t, s.userManager.AddTier(&user.Tier{
  389. Code: "pro",
  390. ReservationLimit: 2,
  391. }))
  392. require.Nil(t, s.userManager.AddUser("noadmin1", "pass", user.RoleUser, false))
  393. require.Nil(t, s.userManager.ChangeTier("noadmin1", "pro"))
  394. require.Nil(t, s.userManager.AddReservation("noadmin1", "mytopic", user.PermissionDenyAll))
  395. require.Nil(t, s.userManager.AddUser("noadmin2", "pass", user.RoleUser, false))
  396. require.Nil(t, s.userManager.ChangeTier("noadmin2", "pro"))
  397. require.Nil(t, s.userManager.AddUser("phil", "adminpass", user.RoleAdmin, false))
  398. // Admin can reserve topic
  399. rr := request(t, s, "POST", "/v1/account/reservation", `{"topic":"sometopic","everyone":"deny-all"}`, map[string]string{
  400. "Authorization": util.BasicAuth("phil", "adminpass"),
  401. })
  402. require.Equal(t, 200, rr.Code)
  403. // User cannot reserve already reserved topic
  404. rr = request(t, s, "POST", "/v1/account/reservation", `{"topic":"mytopic","everyone":"deny-all"}`, map[string]string{
  405. "Authorization": util.BasicAuth("noadmin2", "pass"),
  406. })
  407. require.Equal(t, 409, rr.Code)
  408. // Admin cannot reserve already reserved topic
  409. rr = request(t, s, "POST", "/v1/account/reservation", `{"topic":"mytopic","everyone":"deny-all"}`, map[string]string{
  410. "Authorization": util.BasicAuth("phil", "adminpass"),
  411. })
  412. require.Equal(t, 409, rr.Code)
  413. reservations, err := s.userManager.Reservations("phil")
  414. require.Nil(t, err)
  415. require.Equal(t, 1, len(reservations))
  416. require.Equal(t, "sometopic", reservations[0].Topic)
  417. reservations, err = s.userManager.Reservations("noadmin1")
  418. require.Nil(t, err)
  419. require.Equal(t, 1, len(reservations))
  420. require.Equal(t, "mytopic", reservations[0].Topic)
  421. reservations, err = s.userManager.Reservations("noadmin2")
  422. require.Nil(t, err)
  423. require.Equal(t, 0, len(reservations))
  424. }
  425. func TestAccount_Reservation_AddRemoveUserWithTierSuccess(t *testing.T) {
  426. conf := newTestConfigWithAuthFile(t)
  427. conf.EnableSignup = true
  428. conf.EnableReservations = true
  429. conf.TwilioAccount = "dummy"
  430. s := newTestServer(t, conf)
  431. // Create user
  432. rr := request(t, s, "POST", "/v1/account", `{"username":"phil", "password":"mypass"}`, nil)
  433. require.Equal(t, 200, rr.Code)
  434. // Create a tier
  435. require.Nil(t, s.userManager.AddTier(&user.Tier{
  436. Code: "pro",
  437. MessageLimit: 123,
  438. MessageExpiryDuration: 86400 * time.Second,
  439. EmailLimit: 32,
  440. CallLimit: 10,
  441. ReservationLimit: 2,
  442. AttachmentFileSizeLimit: 1231231,
  443. AttachmentTotalSizeLimit: 123123,
  444. AttachmentExpiryDuration: 10800 * time.Second,
  445. AttachmentBandwidthLimit: 21474836480,
  446. }))
  447. require.Nil(t, s.userManager.ChangeTier("phil", "pro"))
  448. // Reserve two topics
  449. rr = request(t, s, "POST", "/v1/account/reservation", `{"topic": "mytopic", "everyone":"deny-all"}`, map[string]string{
  450. "Authorization": util.BasicAuth("phil", "mypass"),
  451. })
  452. require.Equal(t, 200, rr.Code)
  453. rr = request(t, s, "POST", "/v1/account/reservation", `{"topic": "another", "everyone":"read-only"}`, map[string]string{
  454. "Authorization": util.BasicAuth("phil", "mypass"),
  455. })
  456. require.Equal(t, 200, rr.Code)
  457. // Trying to reserve a third should fail
  458. rr = request(t, s, "POST", "/v1/account/reservation", `{"topic": "yet-another", "everyone":"deny-all"}`, map[string]string{
  459. "Authorization": util.BasicAuth("phil", "mypass"),
  460. })
  461. require.Equal(t, 429, rr.Code)
  462. // Modify existing should still work
  463. rr = request(t, s, "POST", "/v1/account/reservation", `{"topic": "another", "everyone":"write-only"}`, map[string]string{
  464. "Authorization": util.BasicAuth("phil", "mypass"),
  465. })
  466. require.Equal(t, 200, rr.Code)
  467. // Check account result
  468. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  469. "Authorization": util.BasicAuth("phil", "mypass"),
  470. })
  471. require.Equal(t, 200, rr.Code)
  472. account, _ := util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  473. require.Equal(t, "pro", account.Tier.Code)
  474. require.Equal(t, int64(123), account.Limits.Messages)
  475. require.Equal(t, int64(86400), account.Limits.MessagesExpiryDuration)
  476. require.Equal(t, int64(32), account.Limits.Emails)
  477. require.Equal(t, int64(10), account.Limits.Calls)
  478. require.Equal(t, int64(2), account.Limits.Reservations)
  479. require.Equal(t, int64(1231231), account.Limits.AttachmentFileSize)
  480. require.Equal(t, int64(123123), account.Limits.AttachmentTotalSize)
  481. require.Equal(t, int64(10800), account.Limits.AttachmentExpiryDuration)
  482. require.Equal(t, int64(21474836480), account.Limits.AttachmentBandwidth)
  483. require.Equal(t, 2, len(account.Reservations))
  484. require.Equal(t, "another", account.Reservations[0].Topic)
  485. require.Equal(t, "write-only", account.Reservations[0].Everyone)
  486. require.Equal(t, "mytopic", account.Reservations[1].Topic)
  487. require.Equal(t, "deny-all", account.Reservations[1].Everyone)
  488. // Delete and re-check
  489. rr = request(t, s, "DELETE", "/v1/account/reservation/another", "", map[string]string{
  490. "Authorization": util.BasicAuth("phil", "mypass"),
  491. })
  492. require.Equal(t, 200, rr.Code)
  493. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  494. "Authorization": util.BasicAuth("phil", "mypass"),
  495. })
  496. require.Equal(t, 200, rr.Code)
  497. account, _ = util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  498. require.Equal(t, 1, len(account.Reservations))
  499. require.Equal(t, "mytopic", account.Reservations[0].Topic)
  500. }
  501. func TestAccount_Reservation_PublishByAnonymousFails(t *testing.T) {
  502. conf := newTestConfigWithAuthFile(t)
  503. conf.AuthDefault = user.PermissionReadWrite
  504. conf.EnableSignup = true
  505. s := newTestServer(t, conf)
  506. // Create user with tier
  507. rr := request(t, s, "POST", "/v1/account", `{"username":"phil", "password":"mypass"}`, nil)
  508. require.Equal(t, 200, rr.Code)
  509. require.Nil(t, s.userManager.AddTier(&user.Tier{
  510. Code: "pro",
  511. MessageLimit: 20,
  512. ReservationLimit: 2,
  513. }))
  514. require.Nil(t, s.userManager.ChangeTier("phil", "pro"))
  515. // Reserve a topic
  516. rr = request(t, s, "POST", "/v1/account/reservation", `{"topic": "mytopic", "everyone":"deny-all"}`, map[string]string{
  517. "Authorization": util.BasicAuth("phil", "mypass"),
  518. })
  519. require.Equal(t, 200, rr.Code)
  520. // Publish a message
  521. rr = request(t, s, "POST", "/mytopic", `Howdy`, map[string]string{
  522. "Authorization": util.BasicAuth("phil", "mypass"),
  523. })
  524. require.Equal(t, 200, rr.Code)
  525. // Publish a message (as anonymous)
  526. rr = request(t, s, "POST", "/mytopic", `Howdy`, nil)
  527. require.Equal(t, 403, rr.Code)
  528. }
  529. func TestAccount_Reservation_Delete_Messages_And_Attachments(t *testing.T) {
  530. t.Parallel()
  531. conf := newTestConfigWithAuthFile(t)
  532. conf.AuthDefault = user.PermissionReadWrite
  533. s := newTestServer(t, conf)
  534. // Create user with tier
  535. require.Nil(t, s.userManager.AddUser("phil", "mypass", user.RoleUser, false))
  536. require.Nil(t, s.userManager.AddTier(&user.Tier{
  537. Code: "pro",
  538. MessageLimit: 20,
  539. MessageExpiryDuration: time.Hour,
  540. ReservationLimit: 2,
  541. AttachmentTotalSizeLimit: 10000,
  542. AttachmentFileSizeLimit: 10000,
  543. AttachmentExpiryDuration: time.Hour,
  544. AttachmentBandwidthLimit: 10000,
  545. }))
  546. require.Nil(t, s.userManager.ChangeTier("phil", "pro"))
  547. // Reserve two topics "mytopic1" and "mytopic2"
  548. rr := request(t, s, "POST", "/v1/account/reservation", `{"topic": "mytopic1", "everyone":"deny-all"}`, map[string]string{
  549. "Authorization": util.BasicAuth("phil", "mypass"),
  550. })
  551. require.Equal(t, 200, rr.Code)
  552. rr = request(t, s, "POST", "/v1/account/reservation", `{"topic": "mytopic2", "everyone":"deny-all"}`, map[string]string{
  553. "Authorization": util.BasicAuth("phil", "mypass"),
  554. })
  555. require.Equal(t, 200, rr.Code)
  556. // Publish a message with attachment to each topic
  557. rr = request(t, s, "POST", "/mytopic1?f=attach.txt", `Howdy`, map[string]string{
  558. "Authorization": util.BasicAuth("phil", "mypass"),
  559. })
  560. require.Equal(t, 200, rr.Code)
  561. m1 := toMessage(t, rr.Body.String())
  562. require.FileExists(t, filepath.Join(s.config.AttachmentCacheDir, m1.ID))
  563. rr = request(t, s, "POST", "/mytopic2?f=attach.txt", `Howdy`, map[string]string{
  564. "Authorization": util.BasicAuth("phil", "mypass"),
  565. })
  566. require.Equal(t, 200, rr.Code)
  567. m2 := toMessage(t, rr.Body.String())
  568. require.FileExists(t, filepath.Join(s.config.AttachmentCacheDir, m2.ID))
  569. // Pre-verify message count and file
  570. ms, err := s.messageCache.Messages("mytopic1", sinceAllMessages, false)
  571. require.Nil(t, err)
  572. require.Equal(t, 1, len(ms))
  573. require.FileExists(t, filepath.Join(s.config.AttachmentCacheDir, m1.ID))
  574. ms, err = s.messageCache.Messages("mytopic2", sinceAllMessages, false)
  575. require.Nil(t, err)
  576. require.Equal(t, 1, len(ms))
  577. require.FileExists(t, filepath.Join(s.config.AttachmentCacheDir, m2.ID))
  578. // Delete reservation
  579. rr = request(t, s, "DELETE", "/v1/account/reservation/mytopic1", ``, map[string]string{
  580. "X-Delete-Messages": "true",
  581. "Authorization": util.BasicAuth("phil", "mypass"),
  582. })
  583. require.Equal(t, 200, rr.Code)
  584. rr = request(t, s, "DELETE", "/v1/account/reservation/mytopic2", ``, map[string]string{
  585. "X-Delete-Messages": "false",
  586. "Authorization": util.BasicAuth("phil", "mypass"),
  587. })
  588. require.Equal(t, 200, rr.Code)
  589. // Verify that messages and attachments were deleted
  590. // This does not explicitly call the manager!
  591. waitFor(t, func() bool {
  592. ms, err := s.messageCache.Messages("mytopic1", sinceAllMessages, false)
  593. require.Nil(t, err)
  594. return len(ms) == 0 && !util.FileExists(filepath.Join(s.config.AttachmentCacheDir, m1.ID))
  595. })
  596. ms, err = s.messageCache.Messages("mytopic1", sinceAllMessages, false)
  597. require.Nil(t, err)
  598. require.Equal(t, 0, len(ms))
  599. require.NoFileExists(t, filepath.Join(s.config.AttachmentCacheDir, m1.ID))
  600. ms, err = s.messageCache.Messages("mytopic2", sinceAllMessages, false)
  601. require.Nil(t, err)
  602. require.Equal(t, 1, len(ms))
  603. require.Equal(t, m2.ID, ms[0].ID)
  604. require.FileExists(t, filepath.Join(s.config.AttachmentCacheDir, m2.ID))
  605. }
  606. /*func TestAccount_Persist_UserStats_After_Tier_Change(t *testing.T) {
  607. conf := newTestConfigWithAuthFile(t)
  608. conf.AuthDefault = user.PermissionReadWrite
  609. conf.AuthStatsQueueWriterInterval = 300 * time.Millisecond
  610. s := newTestServer(t, conf)
  611. defer s.closeDatabases()
  612. // Create user with tier
  613. require.Nil(t, s.userManager.AddUser("phil", "phil", user.RoleUser))
  614. require.Nil(t, s.userManager.AddTier(&user.Tier{
  615. Code: "starter",
  616. MessageSizeLimit: 10,
  617. }))
  618. require.Nil(t, s.userManager.AddTier(&user.Tier{
  619. Code: "pro",
  620. MessageSizeLimit: 20,
  621. }))
  622. require.Nil(t, s.userManager.ChangeTier("phil", "starter"))
  623. // Publish a message
  624. rr := request(t, s, "POST", "/mytopic", "hi", map[string]string{
  625. "Authorization": util.BasicAuth("phil", "phil"),
  626. })
  627. require.Equal(t, 200, rr.Code)
  628. // Wait for stats queue writer, verify that message stats were persisted
  629. waitFor(t, func() bool {
  630. u, err := s.userManager.User("phil")
  631. require.Nil(t, err)
  632. return int64(1) == u.Stats.Messages
  633. })
  634. // Change tier, make a request (to reset limiters)
  635. require.Nil(t, s.userManager.ChangeTier("phil", "pro"))
  636. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  637. "Authorization": util.BasicAuth("phil", "phil"),
  638. })
  639. require.Equal(t, 200, rr.Code)
  640. account, _ := util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  641. require.Equal(t, int64(1), account.Stats.Messages) // Is not reset!
  642. // Publish another message
  643. rr = request(t, s, "POST", "/mytopic", "hi", map[string]string{
  644. "Authorization": util.BasicAuth("phil", "phil"),
  645. })
  646. require.Equal(t, 200, rr.Code)
  647. // Verify that message stats were persisted
  648. waitFor(t, func() bool {
  649. u, err := s.userManager.User("phil")
  650. require.Nil(t, err)
  651. return int64(2) == u.Stats.Messages // v.EnqueueUserStats had run!
  652. })
  653. // Stats keep counting
  654. rr = request(t, s, "GET", "/v1/account", "", map[string]string{
  655. "Authorization": util.BasicAuth("phil", "phil"),
  656. })
  657. require.Equal(t, 200, rr.Code)
  658. account, _ = util.UnmarshalJSON[apiAccountResponse](io.NopCloser(rr.Body))
  659. require.Equal(t, int64(2), account.Stats.Messages) // Is not reset!
  660. }*/