mgmt_api_test.go 70 KB

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