mgmt_api_test.go 71 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448
  1. package http
  2. import (
  3. "errors"
  4. "fmt"
  5. "io"
  6. "log/slog"
  7. "net/http"
  8. "net/http/httptest"
  9. "net/mail"
  10. "net/netip"
  11. "strings"
  12. "testing"
  13. "time"
  14. "github.com/google/uuid"
  15. "github.com/stretchr/testify/assert"
  16. "github.com/stretchr/testify/mock"
  17. "github.com/mk6i/retro-aim-server/config"
  18. "github.com/mk6i/retro-aim-server/state"
  19. "github.com/mk6i/retro-aim-server/wire"
  20. )
  21. func TestSessionHandler_GET(t *testing.T) {
  22. fnNewSess := func(screenName string, uin uint32, signonComplete bool) *state.Session {
  23. sess := state.NewSession()
  24. sess.SetIdentScreenName(state.NewIdentScreenName(screenName))
  25. sess.SetDisplayScreenName(state.DisplayScreenName(screenName))
  26. sess.SetUIN(uin)
  27. if signonComplete {
  28. sess.SetSignonComplete()
  29. }
  30. ip, _ := netip.ParseAddrPort("1.2.3.4:1234")
  31. sess.SetRemoteAddr(&ip)
  32. return sess
  33. }
  34. tt := []struct {
  35. name string
  36. want string
  37. statusCode int
  38. timeSinceFunc func(t time.Time) time.Duration
  39. mockParams mockParams
  40. }{
  41. {
  42. name: "without sessions",
  43. want: `{"count":0,"sessions":[]}`,
  44. statusCode: http.StatusOK,
  45. mockParams: mockParams{
  46. sessionRetrieverParams: sessionRetrieverParams{
  47. sessionRetrieverAllSessionsParams: sessionRetrieverAllSessionsParams{
  48. {
  49. result: []*state.Session{},
  50. },
  51. },
  52. },
  53. },
  54. },
  55. {
  56. name: "with sessions",
  57. want: `{"count":3,"sessions":[{"id":"usera","screen_name":"userA","online_seconds":0,"away_message":"","idle_seconds":0,"is_icq":false,"remote_addr":"1.2.3.4","remote_port":1234},{"id":"userb","screen_name":"userB","online_seconds":0,"away_message":"","idle_seconds":0,"is_icq":false,"remote_addr":"1.2.3.4","remote_port":1234},{"id":"100003","screen_name":"100003","online_seconds":0,"away_message":"","idle_seconds":0,"is_icq":true,"remote_addr":"1.2.3.4","remote_port":1234}]}`,
  58. statusCode: http.StatusOK,
  59. timeSinceFunc: func(t time.Time) time.Duration { t0 := time.Now(); return t0.Sub(t0) },
  60. mockParams: mockParams{
  61. sessionRetrieverParams: sessionRetrieverParams{
  62. sessionRetrieverAllSessionsParams: sessionRetrieverAllSessionsParams{
  63. {
  64. result: []*state.Session{
  65. fnNewSess("userA", 0, true),
  66. fnNewSess("userB", 0, true),
  67. fnNewSess("100003", 100003, true),
  68. fnNewSess("userC", 0, false),
  69. },
  70. },
  71. },
  72. },
  73. },
  74. },
  75. }
  76. for _, tc := range tt {
  77. t.Run(tc.name, func(t *testing.T) {
  78. request := httptest.NewRequest(http.MethodGet, "/session", nil)
  79. responseRecorder := httptest.NewRecorder()
  80. sessionRetriever := newMockSessionRetriever(t)
  81. for _, params := range tc.mockParams.sessionRetrieverParams.sessionRetrieverAllSessionsParams {
  82. sessionRetriever.EXPECT().
  83. AllSessions().
  84. Return(params.result)
  85. }
  86. getSessionHandler(responseRecorder, request, sessionRetriever, tc.timeSinceFunc)
  87. if responseRecorder.Code != tc.statusCode {
  88. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  89. }
  90. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  91. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  92. }
  93. })
  94. }
  95. }
  96. func TestSessionHandlerScreenname_GET(t *testing.T) {
  97. fnNewSess := func(screenName string, uin uint32, signonComplete bool) *state.Session {
  98. sess := state.NewSession()
  99. sess.SetIdentScreenName(state.NewIdentScreenName(screenName))
  100. sess.SetDisplayScreenName(state.DisplayScreenName(screenName))
  101. sess.SetUIN(uin)
  102. if signonComplete {
  103. sess.SetSignonComplete()
  104. }
  105. ip, _ := netip.ParseAddrPort("1.2.3.4:1234")
  106. sess.SetRemoteAddr(&ip)
  107. return sess
  108. }
  109. tt := []struct {
  110. name string
  111. sessions []*state.Session
  112. requestScreenName state.IdentScreenName
  113. want string
  114. statusCode int
  115. timeSinceFunc func(t time.Time) time.Duration
  116. mockParams mockParams
  117. }{
  118. {
  119. name: "no session for screenname",
  120. sessions: []*state.Session{},
  121. requestScreenName: state.NewIdentScreenName("userA"),
  122. want: `session not found`,
  123. statusCode: http.StatusNotFound,
  124. mockParams: mockParams{
  125. sessionRetrieverParams: sessionRetrieverParams{
  126. retrieveSessionByNameParams: retrieveSessionByNameParams{
  127. {
  128. screenName: state.NewIdentScreenName("userA"),
  129. result: nil,
  130. },
  131. },
  132. },
  133. },
  134. },
  135. {
  136. name: "active session found for screenname",
  137. requestScreenName: state.NewIdentScreenName("userA"),
  138. want: `{"count":1,"sessions":[{"id":"usera","screen_name":"userA","online_seconds":0,"away_message":"","idle_seconds":0,"is_icq":false,"remote_addr":"1.2.3.4","remote_port":1234}]}`,
  139. statusCode: http.StatusOK,
  140. timeSinceFunc: func(t time.Time) time.Duration { t0 := time.Now(); return t0.Sub(t0) },
  141. mockParams: mockParams{
  142. sessionRetrieverParams: sessionRetrieverParams{
  143. retrieveSessionByNameParams: retrieveSessionByNameParams{
  144. {
  145. screenName: state.NewIdentScreenName("userA"),
  146. result: fnNewSess("userA", 0, true),
  147. },
  148. },
  149. },
  150. },
  151. },
  152. }
  153. for _, tc := range tt {
  154. t.Run(tc.name, func(t *testing.T) {
  155. request := httptest.NewRequest(http.MethodGet, "/session/"+tc.requestScreenName.String(), nil)
  156. request.SetPathValue("screenname", tc.requestScreenName.String())
  157. responseRecorder := httptest.NewRecorder()
  158. sessionRetriever := newMockSessionRetriever(t)
  159. for _, params := range tc.mockParams.sessionRetrieverParams.retrieveSessionByNameParams {
  160. sessionRetriever.EXPECT().
  161. RetrieveSession(params.screenName).
  162. Return(params.result)
  163. }
  164. getSessionHandler(responseRecorder, request, sessionRetriever, tc.timeSinceFunc)
  165. if responseRecorder.Code != tc.statusCode {
  166. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  167. }
  168. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  169. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  170. }
  171. })
  172. }
  173. }
  174. func TestSessionHandlerScreenname_DELETE(t *testing.T) {
  175. fnNewSess := func(screenName string) *state.Session {
  176. sess := state.NewSession()
  177. sess.SetIdentScreenName(state.NewIdentScreenName(screenName))
  178. sess.SetDisplayScreenName(state.DisplayScreenName(screenName))
  179. ip, _ := netip.ParseAddrPort("1.2.3.4:1234")
  180. sess.SetRemoteAddr(&ip)
  181. return sess
  182. }
  183. tt := []struct {
  184. name string
  185. session *state.Session
  186. requestScreenName state.IdentScreenName
  187. statusCode int
  188. mockParams mockParams
  189. }{
  190. {
  191. name: "delete an active session",
  192. requestScreenName: state.NewIdentScreenName("userA"),
  193. statusCode: http.StatusNoContent,
  194. mockParams: mockParams{
  195. sessionRetrieverParams: sessionRetrieverParams{
  196. retrieveSessionByNameParams: retrieveSessionByNameParams{
  197. {
  198. screenName: state.NewIdentScreenName("userA"),
  199. result: fnNewSess("userA"),
  200. },
  201. },
  202. },
  203. },
  204. },
  205. {
  206. name: "delete a non-existent session",
  207. requestScreenName: state.NewIdentScreenName("userA"),
  208. statusCode: http.StatusNotFound,
  209. mockParams: mockParams{
  210. sessionRetrieverParams: sessionRetrieverParams{
  211. retrieveSessionByNameParams: retrieveSessionByNameParams{
  212. {
  213. screenName: state.NewIdentScreenName("userA"),
  214. result: nil,
  215. },
  216. },
  217. },
  218. },
  219. },
  220. }
  221. for _, tc := range tt {
  222. t.Run(tc.name, func(t *testing.T) {
  223. request := httptest.NewRequest(http.MethodDelete, "/session/"+tc.requestScreenName.String(), nil)
  224. request.SetPathValue("screenname", tc.requestScreenName.String())
  225. responseRecorder := httptest.NewRecorder()
  226. sessionRetriever := newMockSessionRetriever(t)
  227. for _, params := range tc.mockParams.sessionRetrieverParams.retrieveSessionByNameParams {
  228. sessionRetriever.EXPECT().
  229. RetrieveSession(params.screenName).
  230. Return(params.result)
  231. }
  232. deleteSessionHandler(responseRecorder, request, sessionRetriever)
  233. if responseRecorder.Code != tc.statusCode {
  234. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  235. }
  236. })
  237. }
  238. }
  239. func TestUserAccountHandler_GET(t *testing.T) {
  240. tt := []struct {
  241. name string
  242. requestScreenName state.IdentScreenName
  243. want string
  244. statusCode int
  245. mockParams mockParams
  246. }{
  247. {
  248. name: "invalid account",
  249. requestScreenName: state.NewIdentScreenName("userA"),
  250. want: `user not found`,
  251. statusCode: http.StatusNotFound,
  252. mockParams: mockParams{
  253. userManagerParams: userManagerParams{
  254. getUserParams: getUserParams{
  255. {
  256. screenName: state.NewIdentScreenName("userA"),
  257. result: nil,
  258. },
  259. },
  260. },
  261. },
  262. },
  263. {
  264. name: "valid aim account",
  265. requestScreenName: state.NewIdentScreenName("userA"),
  266. want: `{"id":"usera","screen_name":"userA","profile":"My Profile Text","email_address":"\u003cuserA@aol.com\u003e","reg_status":2,"confirmed":true,"is_icq":false,"suspended_status":"","is_bot":false}`,
  267. statusCode: http.StatusOK,
  268. mockParams: mockParams{
  269. userManagerParams: userManagerParams{
  270. getUserParams: getUserParams{
  271. {
  272. screenName: state.NewIdentScreenName("userA"),
  273. result: &state.User{
  274. DisplayScreenName: "userA",
  275. IdentScreenName: state.NewIdentScreenName("userA"),
  276. SuspendedStatus: 0x0,
  277. },
  278. },
  279. },
  280. },
  281. accountManagerParams: accountManagerParams{
  282. EmailAddressParams: EmailAddressParams{
  283. {
  284. screenName: state.NewIdentScreenName("userA"),
  285. result: &mail.Address{
  286. Address: "userA@aol.com",
  287. },
  288. },
  289. },
  290. RegStatusParams: RegStatusParams{
  291. {
  292. screenName: state.NewIdentScreenName("userA"),
  293. result: uint16(0x02),
  294. },
  295. },
  296. ConfirmStatusParams: ConfirmStatusParams{
  297. {
  298. screenName: state.NewIdentScreenName("userA"),
  299. result: true,
  300. },
  301. },
  302. },
  303. profileRetrieverParams: profileRetrieverParams{
  304. retrieveProfileParams: retrieveProfileParams{
  305. {
  306. screenName: state.NewIdentScreenName("userA"),
  307. result: "My Profile Text",
  308. },
  309. },
  310. },
  311. },
  312. },
  313. {
  314. name: "valid aim bot account",
  315. requestScreenName: state.NewIdentScreenName("userA"),
  316. want: `{"id":"usera","screen_name":"userA","profile":"My Profile Text","email_address":"\u003cuserA@aol.com\u003e","reg_status":2,"confirmed":true,"is_icq":false,"suspended_status":"","is_bot":true}`,
  317. statusCode: http.StatusOK,
  318. mockParams: mockParams{
  319. userManagerParams: userManagerParams{
  320. getUserParams: getUserParams{
  321. {
  322. screenName: state.NewIdentScreenName("userA"),
  323. result: &state.User{
  324. DisplayScreenName: "userA",
  325. IdentScreenName: state.NewIdentScreenName("userA"),
  326. SuspendedStatus: 0x0,
  327. IsBot: true,
  328. },
  329. },
  330. },
  331. },
  332. accountManagerParams: accountManagerParams{
  333. EmailAddressParams: EmailAddressParams{
  334. {
  335. screenName: state.NewIdentScreenName("userA"),
  336. result: &mail.Address{
  337. Address: "userA@aol.com",
  338. },
  339. },
  340. },
  341. RegStatusParams: RegStatusParams{
  342. {
  343. screenName: state.NewIdentScreenName("userA"),
  344. result: uint16(0x02),
  345. },
  346. },
  347. ConfirmStatusParams: ConfirmStatusParams{
  348. {
  349. screenName: state.NewIdentScreenName("userA"),
  350. result: true,
  351. },
  352. },
  353. },
  354. profileRetrieverParams: profileRetrieverParams{
  355. retrieveProfileParams: retrieveProfileParams{
  356. {
  357. screenName: state.NewIdentScreenName("userA"),
  358. result: "My Profile Text",
  359. },
  360. },
  361. },
  362. },
  363. },
  364. {
  365. name: "suspended aim account",
  366. requestScreenName: state.NewIdentScreenName("userB"),
  367. want: `{"id":"userb","screen_name":"userB","profile":"My Profile Text","email_address":"\u003cuserB@aol.com\u003e","reg_status":2,"confirmed":true,"is_icq":false,"suspended_status":"suspended","is_bot":false}`,
  368. statusCode: http.StatusOK,
  369. mockParams: mockParams{
  370. userManagerParams: userManagerParams{
  371. getUserParams: getUserParams{
  372. {
  373. screenName: state.NewIdentScreenName("userB"),
  374. result: &state.User{
  375. DisplayScreenName: "userB",
  376. IdentScreenName: state.NewIdentScreenName("userB"),
  377. SuspendedStatus: wire.LoginErrSuspendedAccount,
  378. },
  379. },
  380. },
  381. },
  382. accountManagerParams: accountManagerParams{
  383. EmailAddressParams: EmailAddressParams{
  384. {
  385. screenName: state.NewIdentScreenName("userB"),
  386. result: &mail.Address{
  387. Address: "userB@aol.com",
  388. },
  389. },
  390. },
  391. RegStatusParams: RegStatusParams{
  392. {
  393. screenName: state.NewIdentScreenName("userB"),
  394. result: uint16(0x02),
  395. },
  396. },
  397. ConfirmStatusParams: ConfirmStatusParams{
  398. {
  399. screenName: state.NewIdentScreenName("userB"),
  400. result: true,
  401. },
  402. },
  403. },
  404. profileRetrieverParams: profileRetrieverParams{
  405. retrieveProfileParams: retrieveProfileParams{
  406. {
  407. screenName: state.NewIdentScreenName("userB"),
  408. result: "My Profile Text",
  409. },
  410. },
  411. },
  412. },
  413. },
  414. }
  415. for _, tc := range tt {
  416. t.Run(tc.name, func(t *testing.T) {
  417. request := httptest.NewRequest(http.MethodGet, "/user/"+tc.requestScreenName.String()+"/account", nil)
  418. request.SetPathValue("screenname", tc.requestScreenName.String())
  419. responseRecorder := httptest.NewRecorder()
  420. userManager := newMockUserManager(t)
  421. for _, params := range tc.mockParams.userManagerParams.getUserParams {
  422. userManager.EXPECT().
  423. User(matchContext(), params.screenName).
  424. Return(params.result, params.err)
  425. }
  426. accountManager := newMockAccountManager(t)
  427. for _, params := range tc.mockParams.accountManagerParams.EmailAddressParams {
  428. accountManager.EXPECT().
  429. EmailAddress(matchContext(), params.screenName).
  430. Return(params.result, params.err)
  431. }
  432. for _, params := range tc.mockParams.accountManagerParams.RegStatusParams {
  433. accountManager.EXPECT().
  434. RegStatus(matchContext(), params.screenName).
  435. Return(params.result, params.err)
  436. }
  437. for _, params := range tc.mockParams.accountManagerParams.ConfirmStatusParams {
  438. accountManager.EXPECT().
  439. ConfirmStatus(matchContext(), params.screenName).
  440. Return(params.result, params.err)
  441. }
  442. profileRetriever := newMockProfileRetriever(t)
  443. for _, params := range tc.mockParams.profileRetrieverParams.retrieveProfileParams {
  444. profileRetriever.EXPECT().
  445. Profile(matchContext(), params.screenName).
  446. Return(params.result, params.err)
  447. }
  448. getUserAccountHandler(responseRecorder, request, userManager, accountManager, profileRetriever, slog.Default())
  449. if responseRecorder.Code != tc.statusCode {
  450. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  451. }
  452. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  453. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  454. }
  455. })
  456. }
  457. }
  458. func TestUserAccountHandler_PATCH(t *testing.T) {
  459. tt := []struct {
  460. name string
  461. requestScreenName state.IdentScreenName
  462. want string
  463. body string
  464. statusCode int
  465. mockParams mockParams
  466. }{
  467. {
  468. name: "suspending a non-existent account",
  469. requestScreenName: state.NewIdentScreenName("userA"),
  470. body: `{"suspended_status":"suspended"}`,
  471. want: `user not found`,
  472. statusCode: http.StatusNotFound,
  473. mockParams: mockParams{
  474. userManagerParams: userManagerParams{
  475. getUserParams: getUserParams{
  476. {
  477. screenName: state.NewIdentScreenName("userA"),
  478. result: nil,
  479. },
  480. },
  481. },
  482. },
  483. },
  484. {
  485. name: "patching with invalid suspended_status value",
  486. requestScreenName: state.NewIdentScreenName("userA"),
  487. body: `{"suspended_status":"thisisinvalid"}`,
  488. want: `{"message":"suspended_status must be empty str or one of deleted,expired,suspended,suspended_age"}`,
  489. statusCode: http.StatusBadRequest,
  490. mockParams: mockParams{
  491. userManagerParams: userManagerParams{
  492. getUserParams: getUserParams{
  493. {
  494. screenName: state.NewIdentScreenName("userA"),
  495. result: &state.User{},
  496. },
  497. },
  498. },
  499. },
  500. },
  501. {
  502. name: "suspending an active aim account",
  503. requestScreenName: state.NewIdentScreenName("userA"),
  504. statusCode: http.StatusNoContent,
  505. body: `{"suspended_status":"suspended"}`,
  506. mockParams: mockParams{
  507. userManagerParams: userManagerParams{
  508. getUserParams: getUserParams{
  509. {
  510. screenName: state.NewIdentScreenName("userA"),
  511. result: &state.User{
  512. DisplayScreenName: "userA",
  513. IdentScreenName: state.NewIdentScreenName("userA"),
  514. SuspendedStatus: 0x0,
  515. },
  516. },
  517. },
  518. },
  519. accountManagerParams: accountManagerParams{
  520. updateSuspendedStatusParams: updateSuspendedStatusParams{
  521. {
  522. suspendedStatus: wire.LoginErrSuspendedAccount,
  523. screenName: state.NewIdentScreenName("userA"),
  524. err: nil,
  525. },
  526. },
  527. },
  528. },
  529. },
  530. {
  531. name: "unsuspending a suspended aim account",
  532. requestScreenName: state.NewIdentScreenName("userA"),
  533. statusCode: http.StatusNoContent,
  534. body: `{"suspended_status":""}`,
  535. mockParams: mockParams{
  536. userManagerParams: userManagerParams{
  537. getUserParams: getUserParams{
  538. {
  539. screenName: state.NewIdentScreenName("userA"),
  540. result: &state.User{
  541. DisplayScreenName: "userA",
  542. IdentScreenName: state.NewIdentScreenName("userA"),
  543. SuspendedStatus: wire.LoginErrSuspendedAccount,
  544. },
  545. },
  546. },
  547. },
  548. accountManagerParams: accountManagerParams{
  549. updateSuspendedStatusParams: updateSuspendedStatusParams{
  550. {
  551. suspendedStatus: 0x0,
  552. screenName: state.NewIdentScreenName("userA"),
  553. err: nil,
  554. },
  555. },
  556. },
  557. },
  558. },
  559. {
  560. name: "suspending an already suspended aim account",
  561. requestScreenName: state.NewIdentScreenName("userA"),
  562. statusCode: http.StatusNotModified,
  563. body: `{"suspended_status":"suspended"}`,
  564. mockParams: mockParams{
  565. userManagerParams: userManagerParams{
  566. getUserParams: getUserParams{
  567. {
  568. screenName: state.NewIdentScreenName("userA"),
  569. result: &state.User{
  570. DisplayScreenName: "userA",
  571. IdentScreenName: state.NewIdentScreenName("userA"),
  572. SuspendedStatus: wire.LoginErrSuspendedAccount,
  573. },
  574. },
  575. },
  576. },
  577. },
  578. },
  579. {
  580. name: "setting bot flag (before: false, after: true)",
  581. requestScreenName: state.NewIdentScreenName("userA"),
  582. statusCode: http.StatusNoContent,
  583. body: `{"is_bot":true}`,
  584. mockParams: mockParams{
  585. userManagerParams: userManagerParams{
  586. getUserParams: getUserParams{
  587. {
  588. screenName: state.NewIdentScreenName("userA"),
  589. result: &state.User{
  590. DisplayScreenName: "userA",
  591. IdentScreenName: state.NewIdentScreenName("userA"),
  592. SuspendedStatus: 0x0,
  593. IsBot: false,
  594. },
  595. },
  596. },
  597. },
  598. accountManagerParams: accountManagerParams{
  599. setBotStatusParams: setBotStatusParams{
  600. {
  601. isBot: true,
  602. screenName: state.NewIdentScreenName("userA"),
  603. err: nil,
  604. },
  605. },
  606. },
  607. },
  608. },
  609. {
  610. name: "setting bot flag (before: true, after: false)",
  611. requestScreenName: state.NewIdentScreenName("userA"),
  612. statusCode: http.StatusNoContent,
  613. body: `{"is_bot":false}`,
  614. mockParams: mockParams{
  615. userManagerParams: userManagerParams{
  616. getUserParams: getUserParams{
  617. {
  618. screenName: state.NewIdentScreenName("userA"),
  619. result: &state.User{
  620. DisplayScreenName: "userA",
  621. IdentScreenName: state.NewIdentScreenName("userA"),
  622. SuspendedStatus: 0x0,
  623. IsBot: true,
  624. },
  625. },
  626. },
  627. },
  628. accountManagerParams: accountManagerParams{
  629. setBotStatusParams: setBotStatusParams{
  630. {
  631. isBot: false,
  632. screenName: state.NewIdentScreenName("userA"),
  633. err: nil,
  634. },
  635. },
  636. },
  637. },
  638. },
  639. {
  640. name: "setting bot flag (before: true, after: true)",
  641. requestScreenName: state.NewIdentScreenName("userA"),
  642. statusCode: http.StatusNotModified,
  643. body: `{"is_bot":true}`,
  644. mockParams: mockParams{
  645. userManagerParams: userManagerParams{
  646. getUserParams: getUserParams{
  647. {
  648. screenName: state.NewIdentScreenName("userA"),
  649. result: &state.User{
  650. DisplayScreenName: "userA",
  651. IdentScreenName: state.NewIdentScreenName("userA"),
  652. SuspendedStatus: 0x0,
  653. IsBot: true,
  654. },
  655. },
  656. },
  657. },
  658. },
  659. },
  660. }
  661. for _, tc := range tt {
  662. t.Run(tc.name, func(t *testing.T) {
  663. request := httptest.NewRequest(http.MethodPatch, "/user/"+tc.requestScreenName.String()+"/account", strings.NewReader(tc.body))
  664. request.SetPathValue("screenname", tc.requestScreenName.String())
  665. responseRecorder := httptest.NewRecorder()
  666. userManager := newMockUserManager(t)
  667. for _, params := range tc.mockParams.userManagerParams.getUserParams {
  668. userManager.EXPECT().
  669. User(matchContext(), params.screenName).
  670. Return(params.result, params.err)
  671. }
  672. accountManager := newMockAccountManager(t)
  673. for _, params := range tc.mockParams.accountManagerParams.updateSuspendedStatusParams {
  674. accountManager.EXPECT().
  675. UpdateSuspendedStatus(matchContext(), params.suspendedStatus, params.screenName).
  676. Return(params.err)
  677. }
  678. for _, params := range tc.mockParams.accountManagerParams.setBotStatusParams {
  679. accountManager.EXPECT().
  680. SetBotStatus(matchContext(), params.isBot, params.screenName).
  681. Return(params.err)
  682. }
  683. patchUserAccountHandler(responseRecorder, request, userManager, accountManager, slog.Default())
  684. if responseRecorder.Code != tc.statusCode {
  685. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  686. }
  687. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  688. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  689. }
  690. })
  691. }
  692. }
  693. func TestUserBuddyIconHandler_GET(t *testing.T) {
  694. sampleGIF := []byte{
  695. 0x47, 0x49, 0x46, 0x38, 0x39, 0x61, 0x32, 0x00, 0x32, 0x00, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00,
  696. 0x00, 0x00, 0x00, 0x21, 0xf9, 0x04, 0x01, 0x00, 0x00, 0x00, 0x00, 0x2c, 0x00, 0x00, 0x00, 0x00,
  697. 0x32, 0x00, 0x32, 0x00, 0x00, 0x02, 0x02, 0x44, 0x01, 0x00, 0x3b}
  698. sampleJPG := []byte{0xFF, 0xD8, 0xFF, 0x43, 0x13, 0x37}
  699. tt := []struct {
  700. name string
  701. requestScreenName state.IdentScreenName
  702. want string
  703. statusCode int
  704. contentType string
  705. mockParams mockParams
  706. }{
  707. {
  708. name: "invalid account",
  709. requestScreenName: state.NewIdentScreenName("userA"),
  710. want: `user not found`,
  711. statusCode: http.StatusNotFound,
  712. contentType: "text/plain; charset=utf-8",
  713. mockParams: mockParams{
  714. userManagerParams: userManagerParams{
  715. getUserParams: getUserParams{
  716. {
  717. screenName: state.NewIdentScreenName("userA"),
  718. result: nil,
  719. },
  720. },
  721. },
  722. },
  723. },
  724. {
  725. name: "account with gif buddy icon",
  726. requestScreenName: state.NewIdentScreenName("userA"),
  727. want: string(sampleGIF),
  728. statusCode: http.StatusOK,
  729. contentType: "image/gif",
  730. mockParams: mockParams{
  731. userManagerParams: userManagerParams{
  732. getUserParams: getUserParams{
  733. {
  734. screenName: state.NewIdentScreenName("userA"),
  735. result: &state.User{
  736. DisplayScreenName: "userA",
  737. IdentScreenName: state.NewIdentScreenName("userA"),
  738. },
  739. },
  740. },
  741. },
  742. feedBagRetrieverParams: feedBagRetrieverParams{
  743. buddyIconMetadataParams: buddyIconMetadataParams{
  744. {
  745. screenName: state.NewIdentScreenName("userA"),
  746. result: &wire.BARTID{
  747. Type: wire.BARTTypesBuddyIcon,
  748. BARTInfo: wire.BARTInfo{
  749. Flags: 0x00,
  750. Hash: []byte{'t', 'h', 'e', 'h', 'a', 's', 'h'},
  751. },
  752. },
  753. },
  754. },
  755. },
  756. bartRetrieverParams: bartRetrieverParams{
  757. bartRetrieveParams: bartRetrieveParams{
  758. {
  759. itemHash: []byte{'t', 'h', 'e', 'h', 'a', 's', 'h'},
  760. result: sampleGIF,
  761. },
  762. },
  763. },
  764. },
  765. },
  766. {
  767. name: "account with jpg buddy icon",
  768. requestScreenName: state.NewIdentScreenName("userA"),
  769. want: string(sampleJPG),
  770. statusCode: http.StatusOK,
  771. contentType: "image/jpeg",
  772. mockParams: mockParams{
  773. userManagerParams: userManagerParams{
  774. getUserParams: getUserParams{
  775. {
  776. screenName: state.NewIdentScreenName("userA"),
  777. result: &state.User{
  778. DisplayScreenName: "userA",
  779. IdentScreenName: state.NewIdentScreenName("userA"),
  780. },
  781. },
  782. },
  783. },
  784. feedBagRetrieverParams: feedBagRetrieverParams{
  785. buddyIconMetadataParams: buddyIconMetadataParams{
  786. {
  787. screenName: state.NewIdentScreenName("userA"),
  788. result: &wire.BARTID{
  789. Type: wire.BARTTypesBuddyIcon,
  790. BARTInfo: wire.BARTInfo{
  791. Flags: 0x00,
  792. Hash: []byte{'t', 'h', 'e', 'h', 'a', 's', 'h'},
  793. },
  794. },
  795. },
  796. },
  797. },
  798. bartRetrieverParams: bartRetrieverParams{
  799. bartRetrieveParams: bartRetrieveParams{
  800. {
  801. itemHash: []byte{'t', 'h', 'e', 'h', 'a', 's', 'h'},
  802. result: sampleJPG,
  803. },
  804. },
  805. },
  806. },
  807. },
  808. {
  809. name: "account with unknown format buddy icon",
  810. requestScreenName: state.NewIdentScreenName("userA"),
  811. want: string([]byte{0x13, 0x37, 0x13, 0x37, 0x13, 0x37}),
  812. statusCode: http.StatusOK,
  813. contentType: "application/octet-stream",
  814. mockParams: mockParams{
  815. userManagerParams: userManagerParams{
  816. getUserParams: getUserParams{
  817. {
  818. screenName: state.NewIdentScreenName("userA"),
  819. result: &state.User{
  820. DisplayScreenName: "userA",
  821. IdentScreenName: state.NewIdentScreenName("userA"),
  822. },
  823. },
  824. },
  825. },
  826. feedBagRetrieverParams: feedBagRetrieverParams{
  827. buddyIconMetadataParams: buddyIconMetadataParams{
  828. {
  829. screenName: state.NewIdentScreenName("userA"),
  830. result: &wire.BARTID{
  831. Type: wire.BARTTypesBuddyIcon,
  832. BARTInfo: wire.BARTInfo{
  833. Flags: 0x00,
  834. Hash: []byte{'t', 'h', 'e', 'h', 'a', 's', 'h'},
  835. },
  836. },
  837. },
  838. },
  839. },
  840. bartRetrieverParams: bartRetrieverParams{
  841. bartRetrieveParams: bartRetrieveParams{
  842. {
  843. itemHash: []byte{'t', 'h', 'e', 'h', 'a', 's', 'h'},
  844. result: []byte{0x13, 0x37, 0x13, 0x37, 0x13, 0x37},
  845. },
  846. },
  847. },
  848. },
  849. },
  850. {
  851. name: "account with cleared buddy icon",
  852. requestScreenName: state.NewIdentScreenName("userA"),
  853. want: "icon not found",
  854. statusCode: http.StatusNotFound,
  855. contentType: "text/plain; charset=utf-8",
  856. mockParams: mockParams{
  857. userManagerParams: userManagerParams{
  858. getUserParams: getUserParams{
  859. {
  860. screenName: state.NewIdentScreenName("userA"),
  861. result: &state.User{
  862. DisplayScreenName: "userA",
  863. IdentScreenName: state.NewIdentScreenName("userA"),
  864. },
  865. },
  866. },
  867. },
  868. feedBagRetrieverParams: feedBagRetrieverParams{
  869. buddyIconMetadataParams: buddyIconMetadataParams{
  870. {
  871. screenName: state.NewIdentScreenName("userA"),
  872. result: &wire.BARTID{
  873. Type: wire.BARTTypesBuddyIcon,
  874. BARTInfo: wire.BARTInfo{
  875. Flags: 0x00,
  876. Hash: wire.GetClearIconHash(),
  877. },
  878. },
  879. },
  880. },
  881. },
  882. },
  883. },
  884. {
  885. name: "account with no buddy icon",
  886. requestScreenName: state.NewIdentScreenName("userA"),
  887. want: "icon not found",
  888. statusCode: http.StatusNotFound,
  889. contentType: "text/plain; charset=utf-8",
  890. mockParams: mockParams{
  891. userManagerParams: userManagerParams{
  892. getUserParams: getUserParams{
  893. {
  894. screenName: state.NewIdentScreenName("userA"),
  895. result: &state.User{
  896. DisplayScreenName: "userA",
  897. IdentScreenName: state.NewIdentScreenName("userA"),
  898. },
  899. },
  900. },
  901. },
  902. feedBagRetrieverParams: feedBagRetrieverParams{
  903. buddyIconMetadataParams: buddyIconMetadataParams{
  904. {
  905. screenName: state.NewIdentScreenName("userA"),
  906. result: nil,
  907. },
  908. },
  909. },
  910. },
  911. },
  912. }
  913. for _, tc := range tt {
  914. t.Run(tc.name, func(t *testing.T) {
  915. request := httptest.NewRequest(http.MethodGet, "/user/"+tc.requestScreenName.String()+"/icon", nil)
  916. request.SetPathValue("screenname", tc.requestScreenName.String())
  917. responseRecorder := httptest.NewRecorder()
  918. userManager := newMockUserManager(t)
  919. for _, params := range tc.mockParams.userManagerParams.getUserParams {
  920. userManager.EXPECT().
  921. User(matchContext(), params.screenName).
  922. Return(params.result, params.err)
  923. }
  924. feedbagRetriever := newMockFeedBagRetriever(t)
  925. for _, params := range tc.mockParams.feedBagRetrieverParams.buddyIconMetadataParams {
  926. feedbagRetriever.EXPECT().
  927. BuddyIconMetadata(matchContext(), params.screenName).
  928. Return(params.result, params.err)
  929. }
  930. bartRetriever := newMockBuddyIconRetriever(t)
  931. for _, params := range tc.mockParams.bartRetrieverParams.bartRetrieveParams {
  932. bartRetriever.EXPECT().
  933. BuddyIcon(matchContext(), params.itemHash).
  934. Return(params.result, params.err)
  935. }
  936. getUserBuddyIconHandler(responseRecorder, request, userManager, feedbagRetriever, bartRetriever, slog.Default())
  937. if responseRecorder.Code != tc.statusCode {
  938. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  939. }
  940. contentType := responseRecorder.Header().Get("Content-Type")
  941. if contentType != tc.contentType {
  942. t.Errorf("Want content type '%s', got '%s'", tc.contentType, contentType)
  943. }
  944. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  945. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  946. }
  947. })
  948. }
  949. }
  950. func TestUserHandler_GET(t *testing.T) {
  951. tt := []struct {
  952. name string
  953. want string
  954. statusCode int
  955. mockParams mockParams
  956. }{
  957. {
  958. name: "empty user store",
  959. want: `[]`,
  960. statusCode: http.StatusOK,
  961. mockParams: mockParams{
  962. userManagerParams: userManagerParams{
  963. allUsersParams: allUsersParams{
  964. {
  965. result: []state.User{},
  966. },
  967. },
  968. },
  969. },
  970. },
  971. {
  972. name: "user store containing 3 users",
  973. want: `[{"id":"usera","screen_name":"userA","is_icq":false,"suspended_status":"","is_bot":false},{"id":"userb","screen_name":"userB","is_icq":false,"suspended_status":"","is_bot":true},{"id":"100003","screen_name":"100003","is_icq":true,"suspended_status":"","is_bot":false}]`,
  974. statusCode: http.StatusOK,
  975. mockParams: mockParams{
  976. userManagerParams: userManagerParams{
  977. allUsersParams: allUsersParams{
  978. {
  979. result: []state.User{
  980. {
  981. DisplayScreenName: "userA",
  982. IdentScreenName: state.NewIdentScreenName("userA"),
  983. },
  984. {
  985. DisplayScreenName: "userB",
  986. IdentScreenName: state.NewIdentScreenName("userB"),
  987. IsBot: true,
  988. },
  989. {
  990. DisplayScreenName: "100003",
  991. IdentScreenName: state.NewIdentScreenName("100003"),
  992. IsICQ: true,
  993. },
  994. },
  995. },
  996. },
  997. },
  998. },
  999. },
  1000. {
  1001. name: "user handler error",
  1002. want: `internal server error`,
  1003. statusCode: http.StatusInternalServerError,
  1004. mockParams: mockParams{
  1005. userManagerParams: userManagerParams{
  1006. allUsersParams: allUsersParams{
  1007. {
  1008. result: []state.User{},
  1009. err: io.EOF,
  1010. },
  1011. },
  1012. },
  1013. },
  1014. },
  1015. }
  1016. for _, tc := range tt {
  1017. t.Run(tc.name, func(t *testing.T) {
  1018. request := httptest.NewRequest(http.MethodGet, "/user", nil)
  1019. responseRecorder := httptest.NewRecorder()
  1020. userManager := newMockUserManager(t)
  1021. for _, params := range tc.mockParams.userManagerParams.allUsersParams {
  1022. userManager.EXPECT().
  1023. AllUsers(matchContext()).
  1024. Return(params.result, params.err)
  1025. }
  1026. getUserHandler(responseRecorder, request, userManager, slog.Default())
  1027. if responseRecorder.Code != tc.statusCode {
  1028. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  1029. }
  1030. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  1031. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  1032. }
  1033. })
  1034. }
  1035. }
  1036. func TestUserHandler_POST(t *testing.T) {
  1037. tt := []struct {
  1038. name string
  1039. body string
  1040. UUID uuid.UUID
  1041. want string
  1042. password string
  1043. statusCode int
  1044. mockParams mockParams
  1045. }{
  1046. {
  1047. name: "with valid AIM user",
  1048. body: `{"screen_name":"userA", "password":"thepassword"}`,
  1049. UUID: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b"),
  1050. want: `User account created successfully.`,
  1051. password: "thepassword",
  1052. statusCode: http.StatusCreated,
  1053. mockParams: mockParams{
  1054. userManagerParams: userManagerParams{
  1055. insertUserParams: insertUserParams{
  1056. {
  1057. u: state.User{
  1058. AuthKey: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b").String(),
  1059. DisplayScreenName: "userA",
  1060. IdentScreenName: state.NewIdentScreenName("userA"),
  1061. },
  1062. err: nil,
  1063. },
  1064. },
  1065. },
  1066. },
  1067. },
  1068. {
  1069. name: "with valid ICQ user",
  1070. body: `{"screen_name":"100003", "password":"thepass"}`,
  1071. UUID: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b"),
  1072. want: `User account created successfully.`,
  1073. password: "thepass",
  1074. statusCode: http.StatusCreated,
  1075. mockParams: mockParams{
  1076. userManagerParams: userManagerParams{
  1077. insertUserParams: insertUserParams{
  1078. {
  1079. u: state.User{
  1080. AuthKey: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b").String(),
  1081. DisplayScreenName: "100003",
  1082. IdentScreenName: state.NewIdentScreenName("100003"),
  1083. IsICQ: true,
  1084. },
  1085. err: nil,
  1086. },
  1087. },
  1088. },
  1089. },
  1090. },
  1091. {
  1092. name: "with malformed body",
  1093. body: `{"screen_name":"userA", "password":"thepassword"`, // missing closing }
  1094. want: `malformed input`,
  1095. statusCode: http.StatusBadRequest,
  1096. },
  1097. {
  1098. name: "user handler error",
  1099. body: `{"screen_name":"userA", "password":"thepassword"}`,
  1100. UUID: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b"),
  1101. want: `internal server error`,
  1102. password: "thepassword",
  1103. statusCode: http.StatusInternalServerError,
  1104. mockParams: mockParams{
  1105. userManagerParams: userManagerParams{
  1106. insertUserParams: insertUserParams{
  1107. {
  1108. u: state.User{
  1109. AuthKey: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b").String(),
  1110. DisplayScreenName: "userA",
  1111. IdentScreenName: state.NewIdentScreenName("userA"),
  1112. },
  1113. err: io.EOF,
  1114. },
  1115. },
  1116. },
  1117. },
  1118. },
  1119. {
  1120. name: "duplicate user",
  1121. body: `{"screen_name":"userA", "password":"thepassword"}`,
  1122. UUID: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b"),
  1123. want: `user already exists`,
  1124. password: "thepassword",
  1125. statusCode: http.StatusConflict,
  1126. mockParams: mockParams{
  1127. userManagerParams: userManagerParams{
  1128. insertUserParams: insertUserParams{
  1129. {
  1130. u: state.User{
  1131. AuthKey: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b").String(),
  1132. DisplayScreenName: "userA",
  1133. IdentScreenName: state.NewIdentScreenName("userA"),
  1134. },
  1135. err: state.ErrDupUser,
  1136. },
  1137. },
  1138. },
  1139. },
  1140. },
  1141. {
  1142. name: "invalid AIM screen name",
  1143. body: `{"screen_name":"a", "password":"thepassword"}`,
  1144. UUID: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b"),
  1145. want: `invalid screen name: screen name must be between 3 and 16 characters`,
  1146. statusCode: http.StatusBadRequest,
  1147. },
  1148. {
  1149. name: "invalid AIM password",
  1150. body: `{"screen_name":"userA", "password":"1"}`,
  1151. UUID: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b"),
  1152. want: `invalid password: invalid password length: password length must be between 4-16 characters`,
  1153. statusCode: http.StatusBadRequest,
  1154. },
  1155. {
  1156. name: "invalid ICQ UIN",
  1157. body: `{"screen_name":"1000", "password":"thepass"}`,
  1158. UUID: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b"),
  1159. want: `invalid uin: uin must be a number in the range 10000-2147483646`,
  1160. statusCode: http.StatusBadRequest,
  1161. },
  1162. {
  1163. name: "invalid ICQ password",
  1164. body: `{"screen_name":"100003", "password":"thelongpassword"}`,
  1165. UUID: uuid.MustParse("07c70701-ba68-49a9-9f9b-67a53816e37b"),
  1166. want: `invalid password: invalid password length: password must be between 6-8 characters`,
  1167. statusCode: http.StatusBadRequest,
  1168. },
  1169. }
  1170. for _, tc := range tt {
  1171. t.Run(tc.name, func(t *testing.T) {
  1172. request := httptest.NewRequest(http.MethodPost, "/user", strings.NewReader(tc.body))
  1173. responseRecorder := httptest.NewRecorder()
  1174. userManager := newMockUserManager(t)
  1175. for _, params := range tc.mockParams.userManagerParams.insertUserParams {
  1176. assert.NoError(t, params.u.HashPassword(tc.password))
  1177. userManager.EXPECT().
  1178. InsertUser(matchContext(), params.u).
  1179. Return(params.err)
  1180. }
  1181. newUUID := func() uuid.UUID { return tc.UUID }
  1182. postUserHandler(responseRecorder, request, userManager, newUUID, slog.Default())
  1183. if responseRecorder.Code != tc.statusCode {
  1184. t.Errorf("want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  1185. }
  1186. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  1187. t.Errorf("want '%s', got '%s'", tc.want, responseRecorder.Body)
  1188. }
  1189. })
  1190. }
  1191. }
  1192. func TestUserHandler_DELETE(t *testing.T) {
  1193. tt := []struct {
  1194. name string
  1195. body string
  1196. want string
  1197. statusCode int
  1198. mockParams mockParams
  1199. }{
  1200. {
  1201. name: "with valid user",
  1202. body: `{"screen_name":"userA"}`,
  1203. want: `User account successfully deleted.`,
  1204. statusCode: http.StatusNoContent,
  1205. mockParams: mockParams{
  1206. userManagerParams: userManagerParams{
  1207. deleteUserParams: deleteUserParams{
  1208. {
  1209. screenName: state.NewIdentScreenName("userA"),
  1210. },
  1211. },
  1212. },
  1213. },
  1214. },
  1215. {
  1216. name: "with non-existent user",
  1217. body: `{"screen_name":"userA"}`,
  1218. want: `user does not exist`,
  1219. statusCode: http.StatusNotFound,
  1220. mockParams: mockParams{
  1221. userManagerParams: userManagerParams{
  1222. deleteUserParams: deleteUserParams{
  1223. {
  1224. screenName: state.NewIdentScreenName("userA"),
  1225. err: state.ErrNoUser,
  1226. },
  1227. },
  1228. },
  1229. },
  1230. },
  1231. {
  1232. name: "with malformed body",
  1233. body: `{"screen_name":"userA"`, // missing closing }
  1234. want: `malformed input`,
  1235. statusCode: http.StatusBadRequest,
  1236. },
  1237. {
  1238. name: "user handler error",
  1239. body: `{"screen_name":"userA"}`,
  1240. want: `internal server error`,
  1241. statusCode: http.StatusInternalServerError,
  1242. mockParams: mockParams{
  1243. userManagerParams: userManagerParams{
  1244. deleteUserParams: deleteUserParams{
  1245. {
  1246. screenName: state.NewIdentScreenName("userA"),
  1247. err: io.EOF,
  1248. },
  1249. },
  1250. },
  1251. },
  1252. },
  1253. }
  1254. for _, tc := range tt {
  1255. t.Run(tc.name, func(t *testing.T) {
  1256. request := httptest.NewRequest(http.MethodDelete, "/user", strings.NewReader(tc.body))
  1257. responseRecorder := httptest.NewRecorder()
  1258. userManager := newMockUserManager(t)
  1259. for _, params := range tc.mockParams.userManagerParams.deleteUserParams {
  1260. userManager.EXPECT().
  1261. DeleteUser(matchContext(), params.screenName).
  1262. Return(params.err)
  1263. }
  1264. deleteUserHandler(responseRecorder, request, userManager, slog.Default())
  1265. if responseRecorder.Code != tc.statusCode {
  1266. t.Errorf("want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  1267. }
  1268. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  1269. t.Errorf("want '%s', got '%s'", tc.want, responseRecorder.Body)
  1270. }
  1271. })
  1272. }
  1273. }
  1274. func TestUserPasswordHandler_PUT(t *testing.T) {
  1275. tt := []struct {
  1276. name string
  1277. body string
  1278. want string
  1279. statusCode int
  1280. mockParams mockParams
  1281. }{
  1282. {
  1283. name: "user with valid password",
  1284. body: `{"screen_name":"userA", "password":"thenewpassword"}`,
  1285. want: `Password successfully reset.`,
  1286. statusCode: http.StatusNoContent,
  1287. mockParams: mockParams{
  1288. userManagerParams: userManagerParams{
  1289. setUserPasswordParams: setUserPasswordParams{
  1290. {
  1291. screenName: state.NewIdentScreenName("userA"),
  1292. newPassword: "thenewpassword",
  1293. },
  1294. },
  1295. },
  1296. },
  1297. },
  1298. {
  1299. name: "user with invalid password",
  1300. body: `{"screen_name":"userA", "password":"a"}`,
  1301. want: `invalid password length`,
  1302. statusCode: http.StatusBadRequest,
  1303. mockParams: mockParams{
  1304. userManagerParams: userManagerParams{
  1305. setUserPasswordParams: setUserPasswordParams{
  1306. {
  1307. screenName: state.NewIdentScreenName("userA"),
  1308. newPassword: "a",
  1309. err: state.ErrPasswordInvalid,
  1310. },
  1311. },
  1312. },
  1313. },
  1314. },
  1315. {
  1316. name: "with malformed body",
  1317. body: `{"screen_name":"userA", "password":"thepassword"`, // missing closing }
  1318. want: `malformed input`,
  1319. statusCode: http.StatusBadRequest,
  1320. },
  1321. {
  1322. name: "password updater returns runtime error",
  1323. body: `{"screen_name":"userA", "password":"thepassword"}`,
  1324. want: `internal server error`,
  1325. statusCode: http.StatusInternalServerError,
  1326. mockParams: mockParams{
  1327. userManagerParams: userManagerParams{
  1328. setUserPasswordParams: setUserPasswordParams{
  1329. {
  1330. screenName: state.NewIdentScreenName("userA"),
  1331. newPassword: "thepassword",
  1332. err: io.EOF,
  1333. },
  1334. },
  1335. },
  1336. },
  1337. },
  1338. {
  1339. name: "user doesn't exist",
  1340. body: `{"screen_name":"userA", "password":"thepassword"}`,
  1341. want: `user does not exist`,
  1342. statusCode: http.StatusNotFound,
  1343. mockParams: mockParams{
  1344. userManagerParams: userManagerParams{
  1345. setUserPasswordParams: setUserPasswordParams{
  1346. {
  1347. screenName: state.NewIdentScreenName("userA"),
  1348. newPassword: "thepassword",
  1349. err: state.ErrNoUser,
  1350. },
  1351. },
  1352. },
  1353. },
  1354. },
  1355. }
  1356. for _, tc := range tt {
  1357. t.Run(tc.name, func(t *testing.T) {
  1358. request := httptest.NewRequest(http.MethodPut, "/user", strings.NewReader(tc.body))
  1359. responseRecorder := httptest.NewRecorder()
  1360. userManager := newMockUserManager(t)
  1361. for _, params := range tc.mockParams.userManagerParams.setUserPasswordParams {
  1362. userManager.EXPECT().
  1363. SetUserPassword(matchContext(), params.screenName, params.newPassword).
  1364. Return(params.err)
  1365. }
  1366. putUserPasswordHandler(responseRecorder, request, userManager, slog.Default())
  1367. if responseRecorder.Code != tc.statusCode {
  1368. t.Errorf("want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  1369. }
  1370. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  1371. t.Errorf("want '%s', got '%s'", tc.want, responseRecorder.Body)
  1372. }
  1373. })
  1374. }
  1375. }
  1376. func TestPublicChatHandler_GET(t *testing.T) {
  1377. fnNewSess := func(screenName string) *state.Session {
  1378. sess := state.NewSession()
  1379. sess.SetIdentScreenName(state.NewIdentScreenName(screenName))
  1380. sess.SetDisplayScreenName(state.DisplayScreenName(screenName))
  1381. return sess
  1382. }
  1383. chatRoom1 := state.NewChatRoom("chat-room-1-name", state.NewIdentScreenName("chat-room-1-creator"), state.PublicExchange)
  1384. chatRoom2 := state.NewChatRoom("chat-room-2-name", state.NewIdentScreenName("chat-room-1-creator"), state.PublicExchange)
  1385. tt := []struct {
  1386. name string
  1387. want string
  1388. statusCode int
  1389. mockParams mockParams
  1390. }{
  1391. {
  1392. name: "multiple chat rooms with participants",
  1393. want: `[{"name":"chat-room-1-name","create_time":"0001-01-01T00:00:00Z","url":"aim:gochat?roomname=chat-room-1-name&exchange=5","participants":[{"id":"usera","screen_name":"userA"},{"id":"userb","screen_name":"userB"}]},{"name":"chat-room-2-name","create_time":"0001-01-01T00:00:00Z","url":"aim:gochat?roomname=chat-room-2-name&exchange=5","participants":[{"id":"userc","screen_name":"userC"},{"id":"userd","screen_name":"userD"}]}]`,
  1394. statusCode: http.StatusOK,
  1395. mockParams: mockParams{
  1396. chatRoomRetrieverParams: chatRoomRetrieverParams{
  1397. allChatRoomsParams: allChatRoomsParams{
  1398. {
  1399. exchange: state.PublicExchange,
  1400. result: []state.ChatRoom{
  1401. chatRoom1,
  1402. chatRoom2,
  1403. },
  1404. },
  1405. },
  1406. },
  1407. chatSessionRetrieverParams: chatSessionRetrieverParams{
  1408. chatSessionRetrieverAllSessionsParams: chatSessionRetrieverAllSessionsParams{
  1409. {
  1410. cookie: chatRoom1.Cookie(),
  1411. result: []*state.Session{
  1412. fnNewSess("userA"),
  1413. fnNewSess("userB"),
  1414. },
  1415. },
  1416. {
  1417. cookie: chatRoom2.Cookie(),
  1418. result: []*state.Session{
  1419. fnNewSess("userC"),
  1420. fnNewSess("userD"),
  1421. },
  1422. },
  1423. },
  1424. },
  1425. },
  1426. },
  1427. {
  1428. name: "chat room without participants",
  1429. want: `[{"name":"chat-room-1-name","create_time":"0001-01-01T00:00:00Z","url":"aim:gochat?roomname=chat-room-1-name&exchange=5","participants":[]}]`,
  1430. statusCode: http.StatusOK,
  1431. mockParams: mockParams{
  1432. chatRoomRetrieverParams: chatRoomRetrieverParams{
  1433. allChatRoomsParams: allChatRoomsParams{
  1434. {
  1435. exchange: state.PublicExchange,
  1436. result: []state.ChatRoom{
  1437. chatRoom1,
  1438. },
  1439. },
  1440. },
  1441. },
  1442. chatSessionRetrieverParams: chatSessionRetrieverParams{
  1443. chatSessionRetrieverAllSessionsParams: chatSessionRetrieverAllSessionsParams{
  1444. {
  1445. cookie: chatRoom1.Cookie(),
  1446. result: []*state.Session{},
  1447. },
  1448. },
  1449. },
  1450. },
  1451. },
  1452. {
  1453. name: "no chat rooms",
  1454. want: `[]`,
  1455. statusCode: http.StatusOK,
  1456. mockParams: mockParams{
  1457. chatRoomRetrieverParams: chatRoomRetrieverParams{
  1458. allChatRoomsParams: allChatRoomsParams{
  1459. {
  1460. exchange: state.PublicExchange,
  1461. result: []state.ChatRoom{},
  1462. },
  1463. },
  1464. },
  1465. },
  1466. },
  1467. }
  1468. for _, tc := range tt {
  1469. t.Run(tc.name, func(t *testing.T) {
  1470. request := httptest.NewRequest(http.MethodGet, "/chat/room/public", nil)
  1471. responseRecorder := httptest.NewRecorder()
  1472. chatRoomRetriever := newMockChatRoomRetriever(t)
  1473. for _, params := range tc.mockParams.chatRoomRetrieverParams.allChatRoomsParams {
  1474. chatRoomRetriever.EXPECT().
  1475. AllChatRooms(matchContext(), params.exchange).
  1476. Return(params.result, params.err)
  1477. }
  1478. chatSessionRetriever := newMockChatSessionRetriever(t)
  1479. for _, params := range tc.mockParams.chatSessionRetrieverParams.chatSessionRetrieverAllSessionsParams {
  1480. chatSessionRetriever.EXPECT().
  1481. AllSessions(params.cookie).
  1482. Return(params.result)
  1483. }
  1484. getPublicChatHandler(responseRecorder, request, chatRoomRetriever, chatSessionRetriever, slog.Default())
  1485. if responseRecorder.Code != tc.statusCode {
  1486. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  1487. }
  1488. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  1489. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  1490. }
  1491. })
  1492. }
  1493. }
  1494. func TestPrivateChatHandler_GET(t *testing.T) {
  1495. fnNewSess := func(screenName string) *state.Session {
  1496. sess := state.NewSession()
  1497. sess.SetIdentScreenName(state.NewIdentScreenName(screenName))
  1498. sess.SetDisplayScreenName(state.DisplayScreenName(screenName))
  1499. return sess
  1500. }
  1501. chatRoom1 := state.NewChatRoom("chat-room-1-name", state.NewIdentScreenName("chat-room-1-creator"), state.PrivateExchange)
  1502. chatRoom2 := state.NewChatRoom("chat-room-2-name", state.NewIdentScreenName("chat-room-2-creator"), state.PrivateExchange)
  1503. tt := []struct {
  1504. name string
  1505. want string
  1506. statusCode int
  1507. mockParams mockParams
  1508. }{
  1509. {
  1510. name: "multiple chat rooms with participants",
  1511. want: `[{"name":"chat-room-1-name","create_time":"0001-01-01T00:00:00Z","creator_id":"chat-room-1-creator","url":"aim:gochat?roomname=chat-room-1-name&exchange=4","participants":[{"id":"usera","screen_name":"userA"},{"id":"userb","screen_name":"userB"}]},{"name":"chat-room-2-name","create_time":"0001-01-01T00:00:00Z","creator_id":"chat-room-2-creator","url":"aim:gochat?roomname=chat-room-2-name&exchange=4","participants":[{"id":"userc","screen_name":"userC"},{"id":"userd","screen_name":"userD"}]}]`,
  1512. statusCode: http.StatusOK,
  1513. mockParams: mockParams{
  1514. chatRoomRetrieverParams: chatRoomRetrieverParams{
  1515. allChatRoomsParams: allChatRoomsParams{
  1516. {
  1517. exchange: state.PrivateExchange,
  1518. result: []state.ChatRoom{
  1519. chatRoom1,
  1520. chatRoom2,
  1521. },
  1522. },
  1523. },
  1524. },
  1525. chatSessionRetrieverParams: chatSessionRetrieverParams{
  1526. chatSessionRetrieverAllSessionsParams: chatSessionRetrieverAllSessionsParams{
  1527. {
  1528. cookie: chatRoom1.Cookie(),
  1529. result: []*state.Session{
  1530. fnNewSess("userA"),
  1531. fnNewSess("userB"),
  1532. },
  1533. },
  1534. {
  1535. cookie: chatRoom2.Cookie(),
  1536. result: []*state.Session{
  1537. fnNewSess("userC"),
  1538. fnNewSess("userD"),
  1539. },
  1540. },
  1541. },
  1542. },
  1543. },
  1544. },
  1545. {
  1546. name: "chat room without participants",
  1547. want: `[{"name":"chat-room-1-name","create_time":"0001-01-01T00:00:00Z","creator_id":"chat-room-1-creator","url":"aim:gochat?roomname=chat-room-1-name&exchange=4","participants":[]}]`,
  1548. statusCode: http.StatusOK,
  1549. mockParams: mockParams{
  1550. chatRoomRetrieverParams: chatRoomRetrieverParams{
  1551. allChatRoomsParams: allChatRoomsParams{
  1552. {
  1553. exchange: state.PrivateExchange,
  1554. result: []state.ChatRoom{
  1555. chatRoom1,
  1556. },
  1557. },
  1558. },
  1559. },
  1560. chatSessionRetrieverParams: chatSessionRetrieverParams{
  1561. chatSessionRetrieverAllSessionsParams: chatSessionRetrieverAllSessionsParams{
  1562. {
  1563. cookie: chatRoom1.Cookie(),
  1564. result: []*state.Session{},
  1565. },
  1566. },
  1567. },
  1568. },
  1569. },
  1570. {
  1571. name: "no chat rooms",
  1572. want: `[]`,
  1573. statusCode: http.StatusOK,
  1574. mockParams: mockParams{
  1575. chatRoomRetrieverParams: chatRoomRetrieverParams{
  1576. allChatRoomsParams: allChatRoomsParams{
  1577. {
  1578. exchange: state.PrivateExchange,
  1579. result: []state.ChatRoom{},
  1580. },
  1581. },
  1582. },
  1583. },
  1584. },
  1585. }
  1586. for _, tc := range tt {
  1587. t.Run(tc.name, func(t *testing.T) {
  1588. request := httptest.NewRequest(http.MethodGet, "/chat/room/private", nil)
  1589. responseRecorder := httptest.NewRecorder()
  1590. chatRoomRetriever := newMockChatRoomRetriever(t)
  1591. for _, params := range tc.mockParams.chatRoomRetrieverParams.allChatRoomsParams {
  1592. chatRoomRetriever.EXPECT().
  1593. AllChatRooms(matchContext(), params.exchange).
  1594. Return(params.result, params.err)
  1595. }
  1596. chatSessionRetriever := newMockChatSessionRetriever(t)
  1597. for _, params := range tc.mockParams.chatSessionRetrieverParams.chatSessionRetrieverAllSessionsParams {
  1598. chatSessionRetriever.EXPECT().
  1599. AllSessions(params.cookie).
  1600. Return(params.result)
  1601. }
  1602. getPrivateChatHandler(responseRecorder, request, chatRoomRetriever, chatSessionRetriever, slog.Default())
  1603. if responseRecorder.Code != tc.statusCode {
  1604. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  1605. }
  1606. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  1607. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  1608. }
  1609. })
  1610. }
  1611. }
  1612. func TestInstantMessageHandler_POST(t *testing.T) {
  1613. type relayToScreenNameInputs struct {
  1614. sender state.IdentScreenName
  1615. recipient state.IdentScreenName
  1616. msg string
  1617. }
  1618. tt := []struct {
  1619. name string
  1620. relayToScreenNameInputs []relayToScreenNameInputs
  1621. body string
  1622. want string
  1623. statusCode int
  1624. }{
  1625. {
  1626. name: "send an instant message",
  1627. relayToScreenNameInputs: []relayToScreenNameInputs{
  1628. {
  1629. sender: state.NewIdentScreenName("sender_sn"),
  1630. recipient: state.NewIdentScreenName("recip_sn"),
  1631. msg: "hello world!",
  1632. },
  1633. },
  1634. body: `{"from":"sender_sn","to":"recip_sn","text":"hello world!"}`,
  1635. want: `Message sent successfully.`,
  1636. statusCode: http.StatusOK,
  1637. },
  1638. {
  1639. name: "with malformed body",
  1640. body: `{"screen_name":"userA", "password":"thepassword"`,
  1641. want: `malformed input`,
  1642. statusCode: http.StatusBadRequest,
  1643. },
  1644. }
  1645. for _, tc := range tt {
  1646. t.Run(tc.name, func(t *testing.T) {
  1647. request := httptest.NewRequest(http.MethodPost, "/user", strings.NewReader(tc.body))
  1648. responseRecorder := httptest.NewRecorder()
  1649. messageRelayer := newMockMessageRelayer(t)
  1650. for _, params := range tc.relayToScreenNameInputs {
  1651. validateSNAC := func(msg wire.SNACMessage) bool {
  1652. body := msg.Body.(wire.SNAC_0x04_0x07_ICBMChannelMsgToClient)
  1653. assert.Equal(t, params.sender.String(), body.TLVUserInfo.ScreenName)
  1654. b, ok := body.Bytes(wire.ICBMTLVAOLIMData)
  1655. assert.True(t, ok)
  1656. txt, err := wire.UnmarshalICBMMessageText(b)
  1657. assert.NoError(t, err)
  1658. assert.Equal(t, params.msg, txt)
  1659. return true
  1660. }
  1661. messageRelayer.EXPECT().
  1662. RelayToScreenName(mock.Anything, params.recipient, mock.MatchedBy(validateSNAC))
  1663. }
  1664. postInstantMessageHandler(responseRecorder, request, messageRelayer, slog.Default())
  1665. if responseRecorder.Code != tc.statusCode {
  1666. t.Errorf("want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  1667. }
  1668. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  1669. t.Errorf("want '%s', got '%s'", tc.want, responseRecorder.Body)
  1670. }
  1671. })
  1672. }
  1673. }
  1674. func TestVersionHandler_GET(t *testing.T) {
  1675. tt := []struct {
  1676. name string
  1677. want string
  1678. statusCode int
  1679. buildInfo config.Build
  1680. }{
  1681. {
  1682. name: "get ras version",
  1683. want: `{"version":"13.3.7","commit":"asdfASDF12345678","date":"2024-03-01"}`,
  1684. statusCode: http.StatusOK,
  1685. buildInfo: config.Build{
  1686. Version: "13.3.7",
  1687. Commit: "asdfASDF12345678",
  1688. Date: "2024-03-01",
  1689. },
  1690. },
  1691. }
  1692. for _, tc := range tt {
  1693. t.Run(tc.name, func(t *testing.T) {
  1694. responseRecorder := httptest.NewRecorder()
  1695. getVersionHandler(responseRecorder, tc.buildInfo)
  1696. if responseRecorder.Code != tc.statusCode {
  1697. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  1698. }
  1699. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  1700. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  1701. }
  1702. })
  1703. }
  1704. }
  1705. func TestDirectoryCategoryHandler_GET(t *testing.T) {
  1706. tt := []struct {
  1707. name string
  1708. want string
  1709. statusCode int
  1710. mockParams mockParams
  1711. }{
  1712. {
  1713. name: "no categories",
  1714. want: `[]`,
  1715. statusCode: http.StatusOK,
  1716. mockParams: mockParams{
  1717. directoryManagerParams: directoryManagerParams{
  1718. categoriesParams: categoriesParams{
  1719. {
  1720. result: nil,
  1721. },
  1722. },
  1723. },
  1724. },
  1725. },
  1726. {
  1727. name: "error fetching categories",
  1728. want: `{"message":"internal server error"}`,
  1729. statusCode: http.StatusInternalServerError,
  1730. mockParams: mockParams{
  1731. directoryManagerParams: directoryManagerParams{
  1732. categoriesParams: categoriesParams{
  1733. {
  1734. result: nil,
  1735. err: errors.New("error fetching categories"),
  1736. },
  1737. },
  1738. },
  1739. },
  1740. },
  1741. {
  1742. name: "fetch some categories",
  1743. want: `[{"id":1,"name":"category-1"},{"id":2,"name":"category-2"}]`,
  1744. statusCode: http.StatusOK,
  1745. mockParams: mockParams{
  1746. directoryManagerParams: directoryManagerParams{
  1747. categoriesParams: categoriesParams{
  1748. {
  1749. result: []state.Category{
  1750. {
  1751. ID: 1,
  1752. Name: "category-1",
  1753. },
  1754. {
  1755. ID: 2,
  1756. Name: "category-2",
  1757. },
  1758. },
  1759. },
  1760. },
  1761. },
  1762. },
  1763. },
  1764. }
  1765. for _, tc := range tt {
  1766. t.Run(tc.name, func(t *testing.T) {
  1767. responseRecorder := httptest.NewRecorder()
  1768. request := httptest.NewRequest(http.MethodGet, "/directory/category", nil)
  1769. directoryManager := newMockDirectoryManager(t)
  1770. for _, params := range tc.mockParams.categoriesParams {
  1771. directoryManager.EXPECT().
  1772. Categories(matchContext()).
  1773. Return(params.result, params.err)
  1774. }
  1775. getDirectoryCategoryHandler(responseRecorder, request, directoryManager, slog.Default())
  1776. if responseRecorder.Code != tc.statusCode {
  1777. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  1778. }
  1779. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  1780. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  1781. }
  1782. })
  1783. }
  1784. }
  1785. func TestDirectoryCategoryKeywordHandler_GET(t *testing.T) {
  1786. tt := []struct {
  1787. name string
  1788. categoryID int
  1789. want string
  1790. statusCode int
  1791. mockParams mockParams
  1792. }{
  1793. {
  1794. name: "category not found",
  1795. categoryID: 1,
  1796. want: `{"message":"category not found"}`,
  1797. statusCode: http.StatusNotFound,
  1798. mockParams: mockParams{
  1799. directoryManagerParams: directoryManagerParams{
  1800. keywordsByCategoryParams: keywordsByCategoryParams{
  1801. {
  1802. categoryID: 1,
  1803. result: nil,
  1804. err: state.ErrKeywordCategoryNotFound,
  1805. },
  1806. },
  1807. },
  1808. },
  1809. },
  1810. {
  1811. name: "error fetching keywords by category",
  1812. categoryID: 1,
  1813. want: `{"message":"internal server error"}`,
  1814. statusCode: http.StatusInternalServerError,
  1815. mockParams: mockParams{
  1816. directoryManagerParams: directoryManagerParams{
  1817. keywordsByCategoryParams: keywordsByCategoryParams{
  1818. {
  1819. categoryID: 1,
  1820. result: nil,
  1821. err: errors.New("error fetching keywords by category"),
  1822. },
  1823. },
  1824. },
  1825. },
  1826. },
  1827. {
  1828. name: "invalid category ID",
  1829. categoryID: -1,
  1830. want: `{"message":"invalid category ID"}`,
  1831. statusCode: http.StatusBadRequest,
  1832. mockParams: mockParams{
  1833. directoryManagerParams: directoryManagerParams{
  1834. keywordsByCategoryParams: keywordsByCategoryParams{},
  1835. },
  1836. },
  1837. },
  1838. {
  1839. name: "no keywords",
  1840. categoryID: 1,
  1841. want: `[]`,
  1842. statusCode: http.StatusOK,
  1843. mockParams: mockParams{
  1844. directoryManagerParams: directoryManagerParams{
  1845. keywordsByCategoryParams: keywordsByCategoryParams{
  1846. {
  1847. categoryID: 1,
  1848. result: nil,
  1849. },
  1850. },
  1851. },
  1852. },
  1853. },
  1854. {
  1855. name: "fetch some keywords by category",
  1856. categoryID: 1,
  1857. want: `[{"id":1,"name":"keyword-1"},{"id":2,"name":"keyword-2"}]`,
  1858. statusCode: http.StatusOK,
  1859. mockParams: mockParams{
  1860. directoryManagerParams: directoryManagerParams{
  1861. keywordsByCategoryParams: keywordsByCategoryParams{
  1862. {
  1863. categoryID: 1,
  1864. result: []state.Keyword{
  1865. {
  1866. ID: 1,
  1867. Name: "keyword-1",
  1868. },
  1869. {
  1870. ID: 2,
  1871. Name: "keyword-2",
  1872. },
  1873. },
  1874. },
  1875. },
  1876. },
  1877. },
  1878. },
  1879. }
  1880. for _, tc := range tt {
  1881. t.Run(tc.name, func(t *testing.T) {
  1882. request := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/directory/category/%d/keyword", tc.categoryID), nil)
  1883. request.SetPathValue("id", fmt.Sprintf("%d", tc.categoryID))
  1884. responseRecorder := httptest.NewRecorder()
  1885. directoryManager := newMockDirectoryManager(t)
  1886. for _, params := range tc.mockParams.keywordsByCategoryParams {
  1887. directoryManager.EXPECT().
  1888. KeywordsByCategory(matchContext(), params.categoryID).
  1889. Return(params.result, params.err)
  1890. }
  1891. getDirectoryCategoryKeywordHandler(responseRecorder, request, directoryManager, slog.Default())
  1892. if responseRecorder.Code != tc.statusCode {
  1893. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  1894. }
  1895. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  1896. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  1897. }
  1898. })
  1899. }
  1900. }
  1901. func TestDirectoryCategoryHandler_DELETE(t *testing.T) {
  1902. tt := []struct {
  1903. name string
  1904. categoryID int
  1905. want string
  1906. statusCode int
  1907. mockParams mockParams
  1908. }{
  1909. {
  1910. name: "category not found",
  1911. categoryID: 1,
  1912. want: `{"message":"category not found"}`,
  1913. statusCode: http.StatusNotFound,
  1914. mockParams: mockParams{
  1915. directoryManagerParams: directoryManagerParams{
  1916. deleteCategoryParams: deleteCategoryParams{
  1917. {
  1918. categoryID: 1,
  1919. err: state.ErrKeywordCategoryNotFound,
  1920. },
  1921. },
  1922. },
  1923. },
  1924. },
  1925. {
  1926. name: "keyword in use by user",
  1927. categoryID: 1,
  1928. want: `{"message":"can't delete because category in use by a user"}`,
  1929. statusCode: http.StatusConflict,
  1930. mockParams: mockParams{
  1931. directoryManagerParams: directoryManagerParams{
  1932. deleteCategoryParams: deleteCategoryParams{
  1933. {
  1934. categoryID: 1,
  1935. err: state.ErrKeywordInUse,
  1936. },
  1937. },
  1938. },
  1939. },
  1940. },
  1941. {
  1942. name: "runtime error",
  1943. categoryID: 1,
  1944. want: `{"message":"internal server error"}`,
  1945. statusCode: http.StatusInternalServerError,
  1946. mockParams: mockParams{
  1947. directoryManagerParams: directoryManagerParams{
  1948. deleteCategoryParams: deleteCategoryParams{
  1949. {
  1950. categoryID: 1,
  1951. err: errors.New("error deleting keyword"),
  1952. },
  1953. },
  1954. },
  1955. },
  1956. },
  1957. {
  1958. name: "successful deletion",
  1959. categoryID: 1,
  1960. want: ``,
  1961. statusCode: http.StatusNoContent,
  1962. mockParams: mockParams{
  1963. directoryManagerParams: directoryManagerParams{
  1964. deleteCategoryParams: deleteCategoryParams{
  1965. {
  1966. categoryID: 1,
  1967. },
  1968. },
  1969. },
  1970. },
  1971. },
  1972. {
  1973. name: "invalid category ID",
  1974. categoryID: -1,
  1975. want: `invalid category ID`,
  1976. statusCode: http.StatusBadRequest,
  1977. mockParams: mockParams{
  1978. directoryManagerParams: directoryManagerParams{
  1979. deleteCategoryParams: deleteCategoryParams{},
  1980. },
  1981. },
  1982. },
  1983. }
  1984. for _, tc := range tt {
  1985. t.Run(tc.name, func(t *testing.T) {
  1986. request := httptest.NewRequest(http.MethodDelete, fmt.Sprintf("/directory/category/%d/keyword", tc.categoryID), nil)
  1987. request.SetPathValue("id", fmt.Sprintf("%d", tc.categoryID))
  1988. responseRecorder := httptest.NewRecorder()
  1989. directoryManager := newMockDirectoryManager(t)
  1990. for _, params := range tc.mockParams.deleteCategoryParams {
  1991. directoryManager.EXPECT().
  1992. DeleteCategory(matchContext(), params.categoryID).
  1993. Return(params.err)
  1994. }
  1995. deleteDirectoryCategoryHandler(responseRecorder, request, directoryManager, slog.Default())
  1996. if responseRecorder.Code != tc.statusCode {
  1997. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  1998. }
  1999. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  2000. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  2001. }
  2002. })
  2003. }
  2004. }
  2005. func TestDirectoryCategoryHandler_POST(t *testing.T) {
  2006. tt := []struct {
  2007. name string
  2008. body string
  2009. want string
  2010. statusCode int
  2011. mockParams mockParams
  2012. }{
  2013. {
  2014. name: "category already exists",
  2015. body: `{"name":"the_category"}`,
  2016. want: `{"message":"category already exists"}`,
  2017. statusCode: http.StatusConflict,
  2018. mockParams: mockParams{
  2019. directoryManagerParams: directoryManagerParams{
  2020. createCategoryParams: createCategoryParams{
  2021. {
  2022. name: "the_category",
  2023. err: state.ErrKeywordCategoryExists,
  2024. },
  2025. },
  2026. },
  2027. },
  2028. },
  2029. {
  2030. name: "runtime error",
  2031. body: `{"name":"the_category"}`,
  2032. want: `{"message":"internal server error"}`,
  2033. statusCode: http.StatusInternalServerError,
  2034. mockParams: mockParams{
  2035. directoryManagerParams: directoryManagerParams{
  2036. createCategoryParams: createCategoryParams{
  2037. {
  2038. name: "the_category",
  2039. err: errors.New("error creating category"),
  2040. },
  2041. },
  2042. },
  2043. },
  2044. },
  2045. {
  2046. name: "bad input",
  2047. body: `{"name":"the_category"`,
  2048. want: `{"message":"malformed input"}`,
  2049. statusCode: http.StatusBadRequest,
  2050. mockParams: mockParams{
  2051. directoryManagerParams: directoryManagerParams{
  2052. createCategoryParams: createCategoryParams{},
  2053. },
  2054. },
  2055. },
  2056. {
  2057. name: "successful creation",
  2058. body: `{"name":"the_category"}`,
  2059. want: `{"id":1,"name":"the_category"}`,
  2060. statusCode: http.StatusCreated,
  2061. mockParams: mockParams{
  2062. directoryManagerParams: directoryManagerParams{
  2063. createCategoryParams: createCategoryParams{
  2064. {
  2065. name: "the_category",
  2066. result: state.Category{
  2067. ID: 1,
  2068. Name: "the_category",
  2069. },
  2070. },
  2071. },
  2072. },
  2073. },
  2074. },
  2075. }
  2076. for _, tc := range tt {
  2077. t.Run(tc.name, func(t *testing.T) {
  2078. request := httptest.NewRequest(http.MethodPost, "/directory/category", strings.NewReader(tc.body))
  2079. responseRecorder := httptest.NewRecorder()
  2080. directoryManager := newMockDirectoryManager(t)
  2081. for _, params := range tc.mockParams.createCategoryParams {
  2082. directoryManager.EXPECT().
  2083. CreateCategory(matchContext(), params.name).
  2084. Return(params.result, params.err)
  2085. }
  2086. postDirectoryCategoryHandler(responseRecorder, request, directoryManager, slog.Default())
  2087. if responseRecorder.Code != tc.statusCode {
  2088. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  2089. }
  2090. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  2091. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  2092. }
  2093. })
  2094. }
  2095. }
  2096. func TestDirectoryKeywordHandler_POST(t *testing.T) {
  2097. tt := []struct {
  2098. name string
  2099. body string
  2100. want string
  2101. statusCode int
  2102. mockParams mockParams
  2103. }{
  2104. {
  2105. name: "keyword already exists",
  2106. body: `{"category_id":1,"name":"the_keyword"}`,
  2107. want: `{"message":"keyword already exists"}`,
  2108. statusCode: http.StatusConflict,
  2109. mockParams: mockParams{
  2110. directoryManagerParams: directoryManagerParams{
  2111. createKeywordParams: createKeywordParams{
  2112. {
  2113. name: "the_keyword",
  2114. categoryID: 1,
  2115. err: state.ErrKeywordExists,
  2116. },
  2117. },
  2118. },
  2119. },
  2120. },
  2121. {
  2122. name: "category not found",
  2123. body: `{"category_id":1,"name":"the_keyword"}`,
  2124. want: `{"message":"category not found"}`,
  2125. statusCode: http.StatusNotFound,
  2126. mockParams: mockParams{
  2127. directoryManagerParams: directoryManagerParams{
  2128. createKeywordParams: createKeywordParams{
  2129. {
  2130. name: "the_keyword",
  2131. categoryID: 1,
  2132. err: state.ErrKeywordCategoryNotFound,
  2133. },
  2134. },
  2135. },
  2136. },
  2137. },
  2138. {
  2139. name: "runtime error",
  2140. body: `{"category_id":1,"name":"the_keyword"}`,
  2141. want: `{"message":"internal server error"}`,
  2142. statusCode: http.StatusInternalServerError,
  2143. mockParams: mockParams{
  2144. directoryManagerParams: directoryManagerParams{
  2145. createKeywordParams: createKeywordParams{
  2146. {
  2147. name: "the_keyword",
  2148. categoryID: 1,
  2149. err: errors.New("error creating keyword"),
  2150. },
  2151. },
  2152. },
  2153. },
  2154. },
  2155. {
  2156. name: "bad input",
  2157. body: `{"category_id":1,"name":"the_keyword"`,
  2158. want: `{"message":"malformed input"}`,
  2159. statusCode: http.StatusBadRequest,
  2160. mockParams: mockParams{
  2161. directoryManagerParams: directoryManagerParams{
  2162. createKeywordParams: createKeywordParams{},
  2163. },
  2164. },
  2165. },
  2166. {
  2167. name: "successful creation",
  2168. body: `{"category_id":1,"name":"the_keyword"}`,
  2169. want: `{"id":1,"name":"the_keyword"}`,
  2170. statusCode: http.StatusCreated,
  2171. mockParams: mockParams{
  2172. directoryManagerParams: directoryManagerParams{
  2173. createKeywordParams: createKeywordParams{
  2174. {
  2175. name: "the_keyword",
  2176. categoryID: 1,
  2177. result: state.Keyword{
  2178. ID: 1,
  2179. Name: "the_keyword",
  2180. },
  2181. },
  2182. },
  2183. },
  2184. },
  2185. },
  2186. }
  2187. for _, tc := range tt {
  2188. t.Run(tc.name, func(t *testing.T) {
  2189. request := httptest.NewRequest(http.MethodPost, "/directory/keyword", strings.NewReader(tc.body))
  2190. responseRecorder := httptest.NewRecorder()
  2191. directoryManager := newMockDirectoryManager(t)
  2192. for _, params := range tc.mockParams.createKeywordParams {
  2193. directoryManager.EXPECT().
  2194. CreateKeyword(matchContext(), params.name, params.categoryID).
  2195. Return(params.result, params.err)
  2196. }
  2197. postDirectoryKeywordHandler(responseRecorder, request, directoryManager, slog.Default())
  2198. if responseRecorder.Code != tc.statusCode {
  2199. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  2200. }
  2201. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  2202. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  2203. }
  2204. })
  2205. }
  2206. }
  2207. func TestDirectoryKeywordHandler_DELETE(t *testing.T) {
  2208. tt := []struct {
  2209. name string
  2210. categoryID int
  2211. want string
  2212. statusCode int
  2213. mockParams mockParams
  2214. }{
  2215. {
  2216. name: "keyword not found",
  2217. categoryID: 1,
  2218. want: `{"message":"keyword not found"}`,
  2219. statusCode: http.StatusNotFound,
  2220. mockParams: mockParams{
  2221. directoryManagerParams: directoryManagerParams{
  2222. deleteKeywordParams: deleteKeywordParams{
  2223. {
  2224. id: 1,
  2225. err: state.ErrKeywordNotFound,
  2226. },
  2227. },
  2228. },
  2229. },
  2230. },
  2231. {
  2232. name: "keyword in use by user",
  2233. categoryID: 1,
  2234. want: `{"message":"can't delete because category in use by a user"}`,
  2235. statusCode: http.StatusConflict,
  2236. mockParams: mockParams{
  2237. directoryManagerParams: directoryManagerParams{
  2238. deleteKeywordParams: deleteKeywordParams{
  2239. {
  2240. id: 1,
  2241. err: state.ErrKeywordInUse,
  2242. },
  2243. },
  2244. },
  2245. },
  2246. },
  2247. {
  2248. name: "runtime error",
  2249. categoryID: 1,
  2250. want: `{"message":"internal server error"}`,
  2251. statusCode: http.StatusInternalServerError,
  2252. mockParams: mockParams{
  2253. directoryManagerParams: directoryManagerParams{
  2254. deleteKeywordParams: deleteKeywordParams{
  2255. {
  2256. id: 1,
  2257. err: errors.New("error deleting keyword"),
  2258. },
  2259. },
  2260. },
  2261. },
  2262. },
  2263. {
  2264. name: "successful deletion",
  2265. categoryID: 1,
  2266. want: ``,
  2267. statusCode: http.StatusNoContent,
  2268. mockParams: mockParams{
  2269. directoryManagerParams: directoryManagerParams{
  2270. deleteKeywordParams: deleteKeywordParams{
  2271. {
  2272. id: 1,
  2273. },
  2274. },
  2275. },
  2276. },
  2277. },
  2278. {
  2279. name: "invalid keyword ID",
  2280. categoryID: -1,
  2281. want: `{"message":"invalid keyword ID"}`,
  2282. statusCode: http.StatusBadRequest,
  2283. mockParams: mockParams{
  2284. directoryManagerParams: directoryManagerParams{
  2285. deleteKeywordParams: deleteKeywordParams{},
  2286. },
  2287. },
  2288. },
  2289. }
  2290. for _, tc := range tt {
  2291. t.Run(tc.name, func(t *testing.T) {
  2292. request := httptest.NewRequest(http.MethodDelete, fmt.Sprintf("/directory/keyword/%d", tc.categoryID), nil)
  2293. request.SetPathValue("id", fmt.Sprintf("%d", tc.categoryID))
  2294. responseRecorder := httptest.NewRecorder()
  2295. directoryManager := newMockDirectoryManager(t)
  2296. for _, params := range tc.mockParams.deleteKeywordParams {
  2297. directoryManager.EXPECT().
  2298. DeleteKeyword(matchContext(), params.id).
  2299. Return(params.err)
  2300. }
  2301. deleteDirectoryKeywordHandler(responseRecorder, request, directoryManager, slog.Default())
  2302. if responseRecorder.Code != tc.statusCode {
  2303. t.Errorf("Want status '%d', got '%d'", tc.statusCode, responseRecorder.Code)
  2304. }
  2305. if strings.TrimSpace(responseRecorder.Body.String()) != tc.want {
  2306. t.Errorf("Want '%s', got '%s'", tc.want, responseRecorder.Body)
  2307. }
  2308. })
  2309. }
  2310. }