mgmt_api_test.go 66 KB

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