feedbag_list_test.go 42 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326
  1. package state
  2. import (
  3. "math"
  4. "testing"
  5. "github.com/mk6i/open-oscar-server/wire"
  6. "github.com/stretchr/testify/assert"
  7. )
  8. func TestFeedbagList_upsertItem(t *testing.T) {
  9. t.Run("generates unique ItemID", func(t *testing.T) {
  10. fl := NewFeedbagList([]wire.FeedbagItem{
  11. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  12. }, func(n int) int { return 42 })
  13. result, inserted := fl.upsertItem(wire.FeedbagItem{
  14. Name: "alice",
  15. ClassID: wire.FeedbagClassIdBuddy,
  16. GroupID: 1,
  17. })
  18. assert.True(t, inserted)
  19. assert.Equal(t, uint16(42), result.ItemID)
  20. assert.Equal(t, "alice", result.Name)
  21. assert.Equal(t, wire.FeedbagClassIdBuddy, result.ClassID)
  22. assert.Equal(t, uint16(1), result.GroupID)
  23. })
  24. t.Run("subsequent insert avoids collision", func(t *testing.T) {
  25. fl := NewFeedbagList([]wire.FeedbagItem{
  26. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  27. }, func(n int) int { return 42 })
  28. fl.upsertItem(wire.FeedbagItem{
  29. Name: "alice",
  30. ClassID: wire.FeedbagClassIdBuddy,
  31. GroupID: 1,
  32. })
  33. result, inserted := fl.upsertItem(wire.FeedbagItem{
  34. Name: "bob",
  35. ClassID: wire.FeedbagClassIdBuddy,
  36. GroupID: 1,
  37. })
  38. assert.True(t, inserted)
  39. assert.Equal(t, uint16(43), result.ItemID)
  40. })
  41. t.Run("non-buddy item does not update group order", func(t *testing.T) {
  42. fl := NewFeedbagList(nil, func(n int) int { return 5 })
  43. fl.upsertItem(wire.FeedbagItem{
  44. Name: "alice",
  45. ClassID: wire.FeedbagClassIDPermit,
  46. })
  47. upserts := fl.PendingUpdates()
  48. assert.Len(t, upserts, 1)
  49. assert.Equal(t, wire.FeedbagClassIDPermit, upserts[0].ClassID)
  50. })
  51. t.Run("updates existing non-buddy item in place", func(t *testing.T) {
  52. fl := NewFeedbagList([]wire.FeedbagItem{
  53. {
  54. Name: "alice",
  55. ClassID: wire.FeedbagClassIDPermit,
  56. ItemID: 7,
  57. TLVLBlock: wire.TLVLBlock{
  58. TLVList: wire.TLVList{
  59. wire.NewTLVBE(0x01, uint16(100)),
  60. },
  61. },
  62. },
  63. }, nil)
  64. _, inserted := fl.upsertItem(wire.FeedbagItem{
  65. Name: "alice",
  66. ClassID: wire.FeedbagClassIDPermit,
  67. TLVLBlock: wire.TLVLBlock{
  68. TLVList: wire.TLVList{
  69. wire.NewTLVBE(0x01, uint16(200)),
  70. },
  71. },
  72. })
  73. assert.False(t, inserted)
  74. upserts := fl.PendingUpdates()
  75. assert.Len(t, upserts, 1)
  76. assert.Equal(t, uint16(7), upserts[0].ItemID)
  77. val, ok := upserts[0].Uint16BE(0x01)
  78. assert.True(t, ok)
  79. assert.Equal(t, uint16(200), val)
  80. })
  81. t.Run("updates existing buddy item matched by group", func(t *testing.T) {
  82. fl := NewFeedbagList([]wire.FeedbagItem{
  83. {Name: "Group1", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  84. {Name: "Group2", ClassID: wire.FeedbagClassIdGroup, GroupID: 2},
  85. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 10},
  86. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 2, ItemID: 20},
  87. }, nil)
  88. _, inserted := fl.upsertItem(wire.FeedbagItem{
  89. Name: "alice",
  90. ClassID: wire.FeedbagClassIdBuddy,
  91. GroupID: 2,
  92. TLVLBlock: wire.TLVLBlock{
  93. TLVList: wire.TLVList{
  94. wire.NewTLVBE(0x01, uint16(999)),
  95. },
  96. },
  97. })
  98. assert.False(t, inserted)
  99. upserts := fl.PendingUpdates()
  100. assert.Len(t, upserts, 1)
  101. assert.Equal(t, uint16(2), upserts[0].GroupID)
  102. assert.Equal(t, "alice", upserts[0].Name)
  103. assert.Equal(t, uint16(20), upserts[0].ItemID)
  104. val, ok := upserts[0].Uint16BE(0x01)
  105. assert.True(t, ok)
  106. assert.Equal(t, uint16(999), val)
  107. })
  108. t.Run("skips update when existing item is identical", func(t *testing.T) {
  109. fl := NewFeedbagList([]wire.FeedbagItem{
  110. {
  111. Name: "alice",
  112. ClassID: wire.FeedbagClassIDPermit,
  113. ItemID: 7,
  114. TLVLBlock: wire.TLVLBlock{
  115. TLVList: wire.TLVList{
  116. wire.NewTLVBE(0x01, uint16(100)),
  117. },
  118. },
  119. },
  120. }, nil)
  121. _, inserted := fl.upsertItem(wire.FeedbagItem{
  122. Name: "alice",
  123. ClassID: wire.FeedbagClassIDPermit,
  124. ItemID: 7,
  125. TLVLBlock: wire.TLVLBlock{
  126. TLVList: wire.TLVList{
  127. wire.NewTLVBE(0x01, uint16(100)),
  128. },
  129. },
  130. })
  131. assert.False(t, inserted)
  132. assert.Nil(t, fl.PendingUpdates())
  133. })
  134. t.Run("normalized screen name: buddy stored with lowercase name", func(t *testing.T) {
  135. fl := NewFeedbagList([]wire.FeedbagItem{
  136. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  137. }, func(n int) int { return 10 })
  138. result, inserted := fl.upsertItem(wire.FeedbagItem{
  139. Name: "Alice",
  140. ClassID: wire.FeedbagClassIdBuddy,
  141. GroupID: 1,
  142. })
  143. assert.True(t, inserted)
  144. assert.Equal(t, "alice", result.Name)
  145. })
  146. t.Run("normalized screen name: buddy upsert with different case matches existing", func(t *testing.T) {
  147. fl := NewFeedbagList([]wire.FeedbagItem{
  148. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  149. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 99},
  150. }, nil)
  151. result, inserted := fl.upsertItem(wire.FeedbagItem{
  152. Name: "ALICE",
  153. ClassID: wire.FeedbagClassIdBuddy,
  154. GroupID: 1,
  155. })
  156. assert.False(t, inserted)
  157. assert.Equal(t, uint16(99), result.ItemID)
  158. assert.Equal(t, "alice", result.Name)
  159. })
  160. t.Run("normalized screen name: permit stored with lowercase name and spaces stripped", func(t *testing.T) {
  161. fl := NewFeedbagList(nil, func(n int) int { return 1 })
  162. result, _ := fl.upsertItem(wire.FeedbagItem{
  163. Name: " Bob Smith ",
  164. ClassID: wire.FeedbagClassIDPermit,
  165. })
  166. assert.Equal(t, "bobsmith", result.Name)
  167. })
  168. t.Run("normalized screen name: permit upsert with different case matches existing", func(t *testing.T) {
  169. fl := NewFeedbagList([]wire.FeedbagItem{
  170. {Name: "bob", ClassID: wire.FeedbagClassIDPermit, ItemID: 5},
  171. }, nil)
  172. result, inserted := fl.upsertItem(wire.FeedbagItem{
  173. Name: "BOB",
  174. ClassID: wire.FeedbagClassIDPermit,
  175. })
  176. assert.False(t, inserted)
  177. assert.Equal(t, "bob", result.Name)
  178. assert.Equal(t, uint16(5), result.ItemID)
  179. })
  180. t.Run("normalized screen name: deny upsert with different case matches existing", func(t *testing.T) {
  181. fl := NewFeedbagList([]wire.FeedbagItem{
  182. {Name: "charlie", ClassID: wire.FeedbagClassIDDeny, ItemID: 3},
  183. }, nil)
  184. result, inserted := fl.upsertItem(wire.FeedbagItem{
  185. Name: "Charlie",
  186. ClassID: wire.FeedbagClassIDDeny,
  187. })
  188. assert.False(t, inserted)
  189. assert.Equal(t, "charlie", result.Name)
  190. assert.Equal(t, uint16(3), result.ItemID)
  191. })
  192. }
  193. func TestFeedbagList_deleteItem(t *testing.T) {
  194. t.Run("non-buddy item does not update group order", func(t *testing.T) {
  195. fl := NewFeedbagList([]wire.FeedbagItem{
  196. {Name: "alice", ClassID: wire.FeedbagClassIDPermit, ItemID: 5},
  197. }, nil)
  198. fl.deleteItem(wire.FeedbagItem{Name: "alice", ClassID: wire.FeedbagClassIDPermit, ItemID: 5})
  199. deletes := fl.PendingDeletes()
  200. assert.Len(t, deletes, 1)
  201. assert.Nil(t, fl.PendingUpdates())
  202. })
  203. t.Run("removes item from items list", func(t *testing.T) {
  204. fl := NewFeedbagList([]wire.FeedbagItem{
  205. {Name: "alice", ClassID: wire.FeedbagClassIDPermit, ItemID: 1},
  206. {Name: "bob", ClassID: wire.FeedbagClassIDPermit, ItemID: 2},
  207. {Name: "charlie", ClassID: wire.FeedbagClassIDPermit, ItemID: 3},
  208. }, nil)
  209. fl.deleteItem(wire.FeedbagItem{Name: "bob", ClassID: wire.FeedbagClassIDPermit, ItemID: 2})
  210. deletes := fl.PendingDeletes()
  211. assert.Len(t, deletes, 1)
  212. assert.Equal(t, "bob", deletes[0].Name)
  213. assert.Equal(t, wire.FeedbagClassIDPermit, deletes[0].ClassID)
  214. })
  215. t.Run("multiple deletes accumulate", func(t *testing.T) {
  216. fl := NewFeedbagList([]wire.FeedbagItem{
  217. {Name: "alice", ClassID: wire.FeedbagClassIDDeny, ItemID: 1},
  218. {Name: "bob", ClassID: wire.FeedbagClassIDDeny, ItemID: 2},
  219. }, nil)
  220. fl.deleteItem(wire.FeedbagItem{Name: "alice", ClassID: wire.FeedbagClassIDDeny, ItemID: 1})
  221. fl.deleteItem(wire.FeedbagItem{Name: "bob", ClassID: wire.FeedbagClassIDDeny, ItemID: 2})
  222. deletes := fl.PendingDeletes()
  223. assert.Len(t, deletes, 2)
  224. })
  225. t.Run("normalized screen name: delete buddy by different case", func(t *testing.T) {
  226. fl := NewFeedbagList([]wire.FeedbagItem{
  227. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  228. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 50},
  229. }, nil)
  230. deleted, found := fl.deleteItem(wire.FeedbagItem{
  231. Name: "ALICE",
  232. ClassID: wire.FeedbagClassIdBuddy,
  233. GroupID: 1,
  234. })
  235. assert.True(t, found)
  236. assert.Equal(t, "alice", deleted.Name)
  237. assert.Equal(t, uint16(50), deleted.ItemID)
  238. })
  239. t.Run("normalized screen name: delete permit by different case", func(t *testing.T) {
  240. fl := NewFeedbagList([]wire.FeedbagItem{
  241. {Name: "bob", ClassID: wire.FeedbagClassIDPermit, ItemID: 7},
  242. }, nil)
  243. deleted, found := fl.deleteItem(wire.FeedbagItem{Name: "Bob", ClassID: wire.FeedbagClassIDPermit})
  244. assert.True(t, found)
  245. assert.Equal(t, "bob", deleted.Name)
  246. })
  247. t.Run("normalized screen name: delete deny by different case", func(t *testing.T) {
  248. fl := NewFeedbagList([]wire.FeedbagItem{
  249. {Name: "charlie", ClassID: wire.FeedbagClassIDDeny, ItemID: 8},
  250. }, nil)
  251. deleted, found := fl.deleteItem(wire.FeedbagItem{Name: "CHARLIE", ClassID: wire.FeedbagClassIDDeny})
  252. assert.True(t, found)
  253. assert.Equal(t, "charlie", deleted.Name)
  254. })
  255. }
  256. func TestFeedbagList_genID(t *testing.T) {
  257. tt := []struct {
  258. name string
  259. randInt func(n int) int
  260. items []wire.FeedbagItem
  261. want uint16
  262. }{
  263. {
  264. name: "empty items list returns random ID",
  265. randInt: func(n int) int { return 1000 },
  266. items: []wire.FeedbagItem{},
  267. want: 1000,
  268. },
  269. {
  270. name: "finds next available ID when starting ID conflicts with ItemID",
  271. randInt: func(n int) int { return 100 },
  272. items: []wire.FeedbagItem{
  273. {ItemID: 100, GroupID: 1},
  274. {ItemID: 101, GroupID: 1},
  275. },
  276. want: 102,
  277. },
  278. {
  279. name: "finds next available ID when starting ID conflicts with GroupID",
  280. randInt: func(n int) int { return 50 },
  281. items: []wire.FeedbagItem{
  282. {ItemID: 1, GroupID: 50},
  283. {ItemID: 2, GroupID: 51},
  284. },
  285. want: 52,
  286. },
  287. {
  288. name: "wraps around and skips 0 to find next available ID",
  289. randInt: func(n int) int { return math.MaxUint16 - 2 },
  290. items: []wire.FeedbagItem{
  291. {ItemID: math.MaxUint16 - 2, GroupID: 1},
  292. {ItemID: math.MaxUint16 - 1, GroupID: 1},
  293. {ItemID: math.MaxUint16, GroupID: 1},
  294. },
  295. want: 2,
  296. },
  297. {
  298. name: "skips 0 when starting from 0 and finds next available",
  299. randInt: func(n int) int { return 0 },
  300. items: []wire.FeedbagItem{},
  301. want: 1,
  302. },
  303. {
  304. name: "returns 0 when all IDs are taken",
  305. randInt: func(n int) int { return 100 },
  306. items: func() []wire.FeedbagItem {
  307. items := make([]wire.FeedbagItem, 0, math.MaxUint16+1)
  308. for i := 0; i <= math.MaxUint16; i++ {
  309. items = append(items, wire.FeedbagItem{
  310. ItemID: uint16(i),
  311. GroupID: uint16(i),
  312. })
  313. }
  314. return items
  315. }(),
  316. want: 0,
  317. },
  318. {
  319. name: "finds ID that conflicts with both ItemID and GroupID",
  320. randInt: func(n int) int { return 200 },
  321. items: []wire.FeedbagItem{
  322. {ItemID: 200, GroupID: 201},
  323. {ItemID: 201, GroupID: 200},
  324. },
  325. want: 202,
  326. },
  327. {
  328. name: "finds available ID immediately when no conflicts",
  329. randInt: func(n int) int { return 500 },
  330. items: []wire.FeedbagItem{
  331. {ItemID: 100, GroupID: 1},
  332. {ItemID: 200, GroupID: 2},
  333. {ItemID: 300, GroupID: 3},
  334. },
  335. want: 500,
  336. },
  337. {
  338. name: "handles single conflict and finds next",
  339. randInt: func(n int) int { return 42 },
  340. items: []wire.FeedbagItem{
  341. {ItemID: 42, GroupID: 1},
  342. },
  343. want: 43,
  344. },
  345. {
  346. name: "finds ID before starting point when wrapping",
  347. randInt: func(n int) int { return 5 },
  348. items: []wire.FeedbagItem{
  349. {ItemID: 5, GroupID: 1},
  350. {ItemID: 6, GroupID: 1},
  351. {ItemID: 7, GroupID: 1},
  352. },
  353. want: 8,
  354. },
  355. }
  356. for _, tc := range tt {
  357. t.Run(tc.name, func(t *testing.T) {
  358. fl := NewFeedbagList(tc.items, tc.randInt)
  359. got := fl.genID()
  360. assert.Equal(t, tc.want, got)
  361. })
  362. }
  363. }
  364. func TestFeedbagList_AddGroup(t *testing.T) {
  365. t.Run("creates group and root group when none exists", func(t *testing.T) {
  366. fl := NewFeedbagList(nil, func(n int) int { return 5 })
  367. group := fl.AddGroup("Buddies")
  368. assert.Equal(t, uint16(5), group.GroupID)
  369. assert.Equal(t, "Buddies", group.Name)
  370. assert.Equal(t, wire.FeedbagClassIdGroup, group.ClassID)
  371. upserts := fl.PendingUpdates()
  372. assert.Len(t, upserts, 2)
  373. root := upserts[0]
  374. assert.Equal(t, wire.FeedbagClassIdGroup, root.ClassID)
  375. assert.Equal(t, uint16(0), root.GroupID)
  376. order, ok := root.Uint16SliceBE(wire.FeedbagAttributesOrder)
  377. assert.True(t, ok)
  378. assert.Equal(t, []uint16{5}, order)
  379. })
  380. t.Run("updates existing root group order", func(t *testing.T) {
  381. existing := []wire.FeedbagItem{
  382. {
  383. ClassID: wire.FeedbagClassIdGroup,
  384. GroupID: 0,
  385. TLVLBlock: wire.TLVLBlock{
  386. TLVList: wire.TLVList{
  387. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1}),
  388. },
  389. },
  390. },
  391. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  392. }
  393. callCount := 0
  394. fl := NewFeedbagList(existing, func(n int) int {
  395. callCount++
  396. return callCount + 1
  397. })
  398. group := fl.AddGroup("Coworkers")
  399. upserts := fl.PendingUpdates()
  400. assert.Len(t, upserts, 2)
  401. order, ok := upserts[0].Uint16SliceBE(wire.FeedbagAttributesOrder)
  402. assert.True(t, ok)
  403. assert.Equal(t, []uint16{1, group.GroupID}, order)
  404. })
  405. t.Run("multiple AddGroup calls accumulate in root order", func(t *testing.T) {
  406. callCount := 0
  407. fl := NewFeedbagList(nil, func(n int) int {
  408. callCount++
  409. return callCount * 10
  410. })
  411. g1 := fl.AddGroup("Group1")
  412. g2 := fl.AddGroup("Group2")
  413. upserts := fl.PendingUpdates()
  414. assert.Len(t, upserts, 3)
  415. order, ok := upserts[0].Uint16SliceBE(wire.FeedbagAttributesOrder)
  416. assert.True(t, ok)
  417. assert.Equal(t, []uint16{g1.GroupID, g2.GroupID}, order)
  418. })
  419. }
  420. func TestFeedbagList_SetMode(t *testing.T) {
  421. t.Run("upserts pdinfo item with mode TLV", func(t *testing.T) {
  422. fl := NewFeedbagList(nil, func(n int) int { return 0 })
  423. fl.SetMode(2)
  424. upserts := fl.PendingUpdates()
  425. assert.Len(t, upserts, 1)
  426. assert.Equal(t, wire.FeedbagClassIdPdinfo, upserts[0].ClassID)
  427. mode, ok := upserts[0].Uint8(wire.FeedbagAttributesPdMode)
  428. assert.True(t, ok)
  429. assert.Equal(t, uint8(2), mode)
  430. })
  431. t.Run("second SetMode updates existing item in place", func(t *testing.T) {
  432. fl := NewFeedbagList(nil, func(n int) int { return 0 })
  433. fl.SetMode(1)
  434. _ = fl.PendingUpdates()
  435. fl.SetMode(3)
  436. upserts := fl.PendingUpdates()
  437. assert.Len(t, upserts, 1)
  438. mode, ok := upserts[0].Uint8(wire.FeedbagAttributesPdMode)
  439. assert.True(t, ok)
  440. assert.Equal(t, uint8(3), mode)
  441. })
  442. }
  443. func TestFeedbagList_DeleteGroup(t *testing.T) {
  444. t.Run("updates root group order", func(t *testing.T) {
  445. fl := NewFeedbagList([]wire.FeedbagItem{
  446. {
  447. Name: "", ClassID: wire.FeedbagClassIdGroup, GroupID: 0,
  448. TLVLBlock: wire.TLVLBlock{
  449. TLVList: wire.TLVList{
  450. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1, 2, 3}),
  451. },
  452. },
  453. },
  454. {
  455. Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1,
  456. TLVLBlock: wire.TLVLBlock{
  457. TLVList: wire.TLVList{
  458. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1}),
  459. },
  460. },
  461. },
  462. {Name: "Jane", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 1},
  463. {
  464. Name: "Coworkers", ClassID: wire.FeedbagClassIdGroup, GroupID: 2,
  465. TLVLBlock: wire.TLVLBlock{
  466. TLVList: wire.TLVList{
  467. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{2, 3}),
  468. },
  469. },
  470. },
  471. {Name: "Joe", ClassID: wire.FeedbagClassIdBuddy, GroupID: 2, ItemID: 2},
  472. {Name: "Fred", ClassID: wire.FeedbagClassIdBuddy, GroupID: 2, ItemID: 3},
  473. {Name: "Family", ClassID: wire.FeedbagClassIdGroup, GroupID: 3,
  474. TLVLBlock: wire.TLVLBlock{
  475. TLVList: wire.TLVList{
  476. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{4}),
  477. },
  478. },
  479. },
  480. {Name: "Alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 3, ItemID: 4},
  481. }, nil)
  482. fl.DeleteGroup("Coworkers")
  483. deletes := fl.PendingDeletes()
  484. assert.Len(t, deletes, 3)
  485. assert.Equal(t, "Coworkers", deletes[0].Name)
  486. assert.Equal(t, uint16(2), deletes[0].GroupID)
  487. assert.Equal(t, "Joe", deletes[1].Name)
  488. assert.Equal(t, uint16(2), deletes[1].GroupID)
  489. assert.Equal(t, "Fred", deletes[2].Name)
  490. assert.Equal(t, uint16(2), deletes[2].GroupID)
  491. upserts := fl.PendingUpdates()
  492. assert.Len(t, upserts, 1)
  493. assert.Equal(t, uint16(0), upserts[0].GroupID)
  494. order, ok := upserts[0].Uint16SliceBE(wire.FeedbagAttributesOrder)
  495. assert.True(t, ok)
  496. assert.Equal(t, []uint16{1, 3}, order)
  497. })
  498. t.Run("deleting non-existent group is a no-op", func(t *testing.T) {
  499. fl := NewFeedbagList(nil, nil)
  500. fl.DeleteGroup("Nonexistent")
  501. assert.Empty(t, fl.PendingDeletes())
  502. assert.Nil(t, fl.PendingUpdates())
  503. })
  504. }
  505. func TestFeedbagList_AddBuddy(t *testing.T) {
  506. t.Run("updates parent group order", func(t *testing.T) {
  507. fl := NewFeedbagList([]wire.FeedbagItem{
  508. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  509. }, func(n int) int { return 50 })
  510. inserted, err := fl.AddBuddy("Buddies", "alice", "", "")
  511. assert.NoError(t, err)
  512. assert.True(t, inserted)
  513. upserts := fl.PendingUpdates()
  514. assert.Len(t, upserts, 2)
  515. assert.Equal(t, uint16(1), upserts[1].GroupID)
  516. order, ok := upserts[1].Uint16SliceBE(wire.FeedbagAttributesOrder)
  517. assert.True(t, ok)
  518. assert.Equal(t, []uint16{upserts[0].ItemID}, order)
  519. })
  520. t.Run("multiple buddies accumulate in parent group order", func(t *testing.T) {
  521. fl := NewFeedbagList([]wire.FeedbagItem{
  522. {
  523. Name: "Buddies",
  524. ClassID: wire.FeedbagClassIdGroup,
  525. GroupID: 1,
  526. TLVLBlock: wire.TLVLBlock{
  527. TLVList: wire.TLVList{
  528. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{10}),
  529. },
  530. },
  531. },
  532. }, func(n int) int { return 50 })
  533. _, err := fl.AddBuddy("Buddies", "alice", "", "")
  534. assert.NoError(t, err)
  535. _, err = fl.AddBuddy("Buddies", "bob", "", "")
  536. assert.NoError(t, err)
  537. upserts := fl.PendingUpdates()
  538. assert.Len(t, upserts, 3)
  539. order, ok := upserts[1].Uint16SliceBE(wire.FeedbagAttributesOrder)
  540. assert.True(t, ok)
  541. assert.Equal(t, []uint16{10, upserts[0].ItemID, upserts[2].ItemID}, order)
  542. })
  543. t.Run("returns error when parent group does not exist", func(t *testing.T) {
  544. fl := NewFeedbagList(nil, func(n int) int { return 5 })
  545. _, err := fl.AddBuddy("Nonexistent", "alice", "", "")
  546. assert.ErrorContains(t, err, "group \"Nonexistent\" not found")
  547. })
  548. t.Run("normalized screen name: stores buddy with lowercase name", func(t *testing.T) {
  549. fl := NewFeedbagList([]wire.FeedbagItem{
  550. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  551. }, func(n int) int { return 1 })
  552. inserted, err := fl.AddBuddy("Buddies", "Alice", "", "")
  553. assert.NoError(t, err)
  554. assert.True(t, inserted)
  555. upserts := fl.PendingUpdates()
  556. var buddy *wire.FeedbagItem
  557. for i := range upserts {
  558. if upserts[i].ClassID == wire.FeedbagClassIdBuddy {
  559. buddy = &upserts[i]
  560. break
  561. }
  562. }
  563. assert.NotNil(t, buddy)
  564. assert.Equal(t, "alice", buddy.Name)
  565. })
  566. t.Run("normalized screen name: second AddBuddy with different case does not insert duplicate", func(t *testing.T) {
  567. fl := NewFeedbagList([]wire.FeedbagItem{
  568. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  569. }, func(n int) int { return 1 })
  570. inserted1, err := fl.AddBuddy("Buddies", "alice", "", "")
  571. assert.NoError(t, err)
  572. assert.True(t, inserted1)
  573. inserted2, err := fl.AddBuddy("Buddies", "ALICE", "", "")
  574. assert.NoError(t, err)
  575. assert.False(t, inserted2)
  576. })
  577. t.Run("normalized screen name: DeleteBuddy finds buddy by different case", func(t *testing.T) {
  578. fl := NewFeedbagList([]wire.FeedbagItem{
  579. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  580. }, func(n int) int { return 1 })
  581. _, err := fl.AddBuddy("Buddies", "alice", "", "")
  582. assert.NoError(t, err)
  583. _ = fl.PendingUpdates()
  584. err = fl.DeleteBuddy("Buddies", "Alice")
  585. assert.NoError(t, err)
  586. deletes := fl.PendingDeletes()
  587. assert.Len(t, deletes, 1)
  588. assert.Equal(t, "alice", deletes[0].Name)
  589. })
  590. t.Run("alias is stored as TLV attribute", func(t *testing.T) {
  591. fl := NewFeedbagList([]wire.FeedbagItem{
  592. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  593. }, func(n int) int { return 1 })
  594. _, err := fl.AddBuddy("Buddies", "alice", "Al", "")
  595. assert.NoError(t, err)
  596. upserts := fl.PendingUpdates()
  597. var buddy *wire.FeedbagItem
  598. for i := range upserts {
  599. if upserts[i].ClassID == wire.FeedbagClassIdBuddy {
  600. buddy = &upserts[i]
  601. break
  602. }
  603. }
  604. assert.NotNil(t, buddy)
  605. alias, ok := buddy.Bytes(wire.FeedbagAttributesAlias)
  606. assert.True(t, ok)
  607. assert.Equal(t, []byte("Al"), alias)
  608. })
  609. t.Run("note is stored as TLV attribute", func(t *testing.T) {
  610. fl := NewFeedbagList([]wire.FeedbagItem{
  611. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  612. }, func(n int) int { return 1 })
  613. _, err := fl.AddBuddy("Buddies", "alice", "", "call first")
  614. assert.NoError(t, err)
  615. upserts := fl.PendingUpdates()
  616. var buddy *wire.FeedbagItem
  617. for i := range upserts {
  618. if upserts[i].ClassID == wire.FeedbagClassIdBuddy {
  619. buddy = &upserts[i]
  620. break
  621. }
  622. }
  623. assert.NotNil(t, buddy)
  624. note, ok := buddy.Bytes(wire.FeedbagAttributesNote)
  625. assert.True(t, ok)
  626. assert.Equal(t, []byte("call first"), note)
  627. })
  628. }
  629. func TestFeedbagList_DeleteBuddy(t *testing.T) {
  630. t.Run("updates parent group order", func(t *testing.T) {
  631. fl := NewFeedbagList([]wire.FeedbagItem{
  632. {
  633. Name: "Buddies",
  634. ClassID: wire.FeedbagClassIdGroup,
  635. GroupID: 1,
  636. TLVLBlock: wire.TLVLBlock{
  637. TLVList: wire.TLVList{
  638. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{10, 20, 30}),
  639. },
  640. },
  641. },
  642. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 10},
  643. {Name: "bob", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 20},
  644. {Name: "charlie", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 30},
  645. }, nil)
  646. err := fl.DeleteBuddy("Buddies", "bob")
  647. assert.NoError(t, err)
  648. deletes := fl.PendingDeletes()
  649. assert.Len(t, deletes, 1)
  650. assert.Equal(t, "bob", deletes[0].Name)
  651. assert.Equal(t, uint16(20), deletes[0].ItemID)
  652. upserts := fl.PendingUpdates()
  653. assert.Len(t, upserts, 1)
  654. order, ok := upserts[0].Uint16SliceBE(wire.FeedbagAttributesOrder)
  655. assert.True(t, ok)
  656. assert.Equal(t, []uint16{10, 30}, order)
  657. })
  658. t.Run("wildcard removes buddy from all groups", func(t *testing.T) {
  659. fl := NewFeedbagList([]wire.FeedbagItem{
  660. {
  661. Name: "Buddies",
  662. ClassID: wire.FeedbagClassIdGroup,
  663. GroupID: 1,
  664. TLVLBlock: wire.TLVLBlock{
  665. TLVList: wire.TLVList{
  666. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{10, 20}),
  667. },
  668. },
  669. },
  670. {
  671. Name: "Coworkers",
  672. ClassID: wire.FeedbagClassIdGroup,
  673. GroupID: 2,
  674. TLVLBlock: wire.TLVLBlock{
  675. TLVList: wire.TLVList{
  676. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{30, 40}),
  677. },
  678. },
  679. },
  680. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 10},
  681. {Name: "bob", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 20},
  682. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 2, ItemID: 30},
  683. {Name: "charlie", ClassID: wire.FeedbagClassIdBuddy, GroupID: 2, ItemID: 40},
  684. }, nil)
  685. err := fl.DeleteBuddy("*", "alice")
  686. assert.NoError(t, err)
  687. deletes := fl.PendingDeletes()
  688. assert.Len(t, deletes, 2)
  689. upserts := fl.PendingUpdates()
  690. assert.Len(t, upserts, 2)
  691. order1, ok := upserts[0].Uint16SliceBE(wire.FeedbagAttributesOrder)
  692. assert.True(t, ok)
  693. assert.Equal(t, []uint16{20}, order1)
  694. order2, ok := upserts[1].Uint16SliceBE(wire.FeedbagAttributesOrder)
  695. assert.True(t, ok)
  696. assert.Equal(t, []uint16{40}, order2)
  697. })
  698. t.Run("removes only buddy in specified group when same screen name in two groups", func(t *testing.T) {
  699. fl := NewFeedbagList([]wire.FeedbagItem{
  700. {
  701. Name: "Buddies",
  702. ClassID: wire.FeedbagClassIdGroup,
  703. GroupID: 1,
  704. TLVLBlock: wire.TLVLBlock{
  705. TLVList: wire.TLVList{
  706. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{10, 20}),
  707. },
  708. },
  709. },
  710. {
  711. Name: "Coworkers",
  712. ClassID: wire.FeedbagClassIdGroup,
  713. GroupID: 2,
  714. TLVLBlock: wire.TLVLBlock{
  715. TLVList: wire.TLVList{
  716. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{20}),
  717. },
  718. },
  719. },
  720. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 2, ItemID: 20},
  721. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 10},
  722. }, nil)
  723. err := fl.DeleteBuddy("Buddies", "alice")
  724. assert.NoError(t, err)
  725. deletes := fl.PendingDeletes()
  726. assert.Len(t, deletes, 1)
  727. assert.Equal(t, "alice", deletes[0].Name)
  728. assert.Equal(t, uint16(1), deletes[0].GroupID, "should delete from Buddies (group 1), not Coworkers (group 2)")
  729. assert.Equal(t, uint16(10), deletes[0].ItemID)
  730. })
  731. }
  732. func TestFeedbagList_PendingUpdates(t *testing.T) {
  733. t.Run("empty when nothing inserted", func(t *testing.T) {
  734. fl := NewFeedbagList(nil, func(n int) int { return 1 })
  735. assert.Nil(t, fl.PendingUpdates())
  736. })
  737. t.Run("includes inserts and updates", func(t *testing.T) {
  738. fl := NewFeedbagList([]wire.FeedbagItem{
  739. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  740. }, func(n int) int { return 50 })
  741. _, err := fl.AddBuddy("Buddies", "alice", "", "")
  742. assert.NoError(t, err)
  743. _, err = fl.AddBuddy("Buddies", "bob", "", "")
  744. assert.NoError(t, err)
  745. upserts := fl.PendingUpdates()
  746. assert.Len(t, upserts, 3)
  747. assert.Equal(t, wire.FeedbagClassIdBuddy, upserts[0].ClassID)
  748. assert.Equal(t, wire.FeedbagClassIdGroup, upserts[1].ClassID)
  749. assert.Equal(t, wire.FeedbagClassIdBuddy, upserts[2].ClassID)
  750. })
  751. t.Run("clears after retrieval", func(t *testing.T) {
  752. fl := NewFeedbagList(nil, func(n int) int { return 5 })
  753. fl.AddGroup("Buddies")
  754. assert.Len(t, fl.PendingUpdates(), 2)
  755. assert.Nil(t, fl.PendingUpdates())
  756. })
  757. }
  758. func TestFeedbagList_PendingDeletes(t *testing.T) {
  759. t.Run("empty when nothing deleted", func(t *testing.T) {
  760. fl := NewFeedbagList(nil, nil)
  761. assert.Nil(t, fl.PendingDeletes())
  762. })
  763. t.Run("clears after retrieval", func(t *testing.T) {
  764. fl := NewFeedbagList([]wire.FeedbagItem{
  765. {Name: "alice", ClassID: wire.FeedbagClassIDPermit, ItemID: 1},
  766. }, nil)
  767. fl.deleteItem(wire.FeedbagItem{Name: "alice", ClassID: wire.FeedbagClassIDPermit, ItemID: 1})
  768. assert.Len(t, fl.PendingDeletes(), 1)
  769. assert.Nil(t, fl.PendingDeletes())
  770. })
  771. }
  772. func TestFeedbagList_PendingUpdates_upsertsOnly(t *testing.T) {
  773. t.Run("tracks upserted items", func(t *testing.T) {
  774. fl := NewFeedbagList(nil, func(n int) int { return 5 })
  775. result, inserted := fl.upsertItem(wire.FeedbagItem{ClassID: wire.FeedbagClassIDPermit, Name: "alice"})
  776. assert.True(t, inserted)
  777. upserts := fl.PendingUpdates()
  778. assert.Len(t, upserts, 1)
  779. assert.Equal(t, "alice", upserts[0].Name)
  780. assert.Equal(t, uint16(5), result.ItemID)
  781. })
  782. t.Run("clears after retrieval", func(t *testing.T) {
  783. fl := NewFeedbagList(nil, func(n int) int { return 5 })
  784. fl.upsertItem(wire.FeedbagItem{ClassID: wire.FeedbagClassIDPermit})
  785. assert.Len(t, fl.PendingUpdates(), 1)
  786. assert.Nil(t, fl.PendingUpdates())
  787. })
  788. t.Run("multiple inserts accumulate", func(t *testing.T) {
  789. fl := NewFeedbagList(nil, func(n int) int { return 10 })
  790. fl.upsertItem(wire.FeedbagItem{ClassID: wire.FeedbagClassIDPermit, Name: "alice"})
  791. fl.upsertItem(wire.FeedbagItem{ClassID: wire.FeedbagClassIDPermit, Name: "bob"})
  792. upserts := fl.PendingUpdates()
  793. assert.Len(t, upserts, 2)
  794. })
  795. }
  796. func TestFeedbagList_PermitUser(t *testing.T) {
  797. t.Run("new permit entry is added", func(t *testing.T) {
  798. fl := NewFeedbagList(nil, func(n int) int { return 0 })
  799. fl.PermitUser("alice")
  800. upserts := fl.PendingUpdates()
  801. assert.Len(t, upserts, 1)
  802. assert.Equal(t, wire.FeedbagClassIDPermit, upserts[0].ClassID)
  803. assert.Equal(t, "alice", upserts[0].Name)
  804. })
  805. t.Run("duplicate permit is not re-added", func(t *testing.T) {
  806. fl := NewFeedbagList([]wire.FeedbagItem{
  807. {ClassID: wire.FeedbagClassIDPermit, Name: "alice", ItemID: 1},
  808. }, nil)
  809. fl.PermitUser("alice")
  810. assert.Nil(t, fl.PendingUpdates())
  811. })
  812. t.Run("name is normalized", func(t *testing.T) {
  813. fl := NewFeedbagList(nil, func(n int) int { return 0 })
  814. fl.PermitUser("Alice")
  815. upserts := fl.PendingUpdates()
  816. assert.Len(t, upserts, 1)
  817. assert.Equal(t, "alice", upserts[0].Name)
  818. })
  819. }
  820. func TestFeedbagList_DenyUser(t *testing.T) {
  821. t.Run("new deny entry is added", func(t *testing.T) {
  822. fl := NewFeedbagList(nil, func(n int) int { return 0 })
  823. fl.DenyUser("alice")
  824. upserts := fl.PendingUpdates()
  825. assert.Len(t, upserts, 1)
  826. assert.Equal(t, wire.FeedbagClassIDDeny, upserts[0].ClassID)
  827. assert.Equal(t, "alice", upserts[0].Name)
  828. })
  829. t.Run("duplicate deny is not re-added", func(t *testing.T) {
  830. fl := NewFeedbagList([]wire.FeedbagItem{
  831. {ClassID: wire.FeedbagClassIDDeny, Name: "alice", ItemID: 1},
  832. }, nil)
  833. fl.DenyUser("alice")
  834. assert.Nil(t, fl.PendingUpdates())
  835. })
  836. t.Run("name is normalized", func(t *testing.T) {
  837. fl := NewFeedbagList(nil, func(n int) int { return 0 })
  838. fl.DenyUser("Alice")
  839. upserts := fl.PendingUpdates()
  840. assert.Len(t, upserts, 1)
  841. assert.Equal(t, "alice", upserts[0].Name)
  842. })
  843. }
  844. func TestFeedbagList_DeletePermit(t *testing.T) {
  845. t.Run("existing permit is deleted", func(t *testing.T) {
  846. item := wire.FeedbagItem{ClassID: wire.FeedbagClassIDPermit, Name: "alice", ItemID: 1}
  847. fl := NewFeedbagList([]wire.FeedbagItem{item}, nil)
  848. fl.DeletePermit("alice")
  849. deletes := fl.PendingDeletes()
  850. assert.Len(t, deletes, 1)
  851. assert.Equal(t, item, deletes[0])
  852. })
  853. t.Run("deleting non-existent permit is a no-op", func(t *testing.T) {
  854. fl := NewFeedbagList(nil, nil)
  855. fl.DeletePermit("alice")
  856. assert.Empty(t, fl.PendingDeletes())
  857. })
  858. t.Run("name comparison is case-insensitive", func(t *testing.T) {
  859. item := wire.FeedbagItem{ClassID: wire.FeedbagClassIDPermit, Name: "alice", ItemID: 1}
  860. fl := NewFeedbagList([]wire.FeedbagItem{item}, nil)
  861. fl.DeletePermit("Alice")
  862. assert.Len(t, fl.PendingDeletes(), 1)
  863. })
  864. }
  865. func TestFeedbagList_DeleteDeny(t *testing.T) {
  866. t.Run("existing deny is deleted", func(t *testing.T) {
  867. item := wire.FeedbagItem{ClassID: wire.FeedbagClassIDDeny, Name: "alice", ItemID: 1}
  868. fl := NewFeedbagList([]wire.FeedbagItem{item}, nil)
  869. fl.DeleteDeny("alice")
  870. deletes := fl.PendingDeletes()
  871. assert.Len(t, deletes, 1)
  872. assert.Equal(t, item, deletes[0])
  873. })
  874. t.Run("deleting non-existent deny is a no-op", func(t *testing.T) {
  875. fl := NewFeedbagList(nil, nil)
  876. fl.DeleteDeny("alice")
  877. assert.Empty(t, fl.PendingDeletes())
  878. })
  879. t.Run("name comparison is case-insensitive", func(t *testing.T) {
  880. item := wire.FeedbagItem{ClassID: wire.FeedbagClassIDDeny, Name: "alice", ItemID: 1}
  881. fl := NewFeedbagList([]wire.FeedbagItem{item}, nil)
  882. fl.DeleteDeny("Alice")
  883. assert.Len(t, fl.PendingDeletes(), 1)
  884. })
  885. }
  886. func TestFeedbagList_AddLinkedScreenName(t *testing.T) {
  887. t.Run("new linked screen name is added", func(t *testing.T) {
  888. fl := NewFeedbagList(nil, func(n int) int { return 0 })
  889. fl.AddLinkedScreenName("alice")
  890. upserts := fl.PendingUpdates()
  891. assert.Len(t, upserts, 1)
  892. assert.Equal(t, wire.FeedbagClassIdAlInfo, upserts[0].ClassID)
  893. assert.Equal(t, "alice", upserts[0].Name)
  894. })
  895. t.Run("duplicate linked screen name is not re-added", func(t *testing.T) {
  896. fl := NewFeedbagList([]wire.FeedbagItem{
  897. {ClassID: wire.FeedbagClassIdAlInfo, Name: "alice", ItemID: 1},
  898. }, nil)
  899. fl.AddLinkedScreenName("alice")
  900. assert.Nil(t, fl.PendingUpdates())
  901. })
  902. t.Run("name is normalized", func(t *testing.T) {
  903. fl := NewFeedbagList(nil, func(n int) int { return 0 })
  904. fl.AddLinkedScreenName("Alice")
  905. upserts := fl.PendingUpdates()
  906. assert.Len(t, upserts, 1)
  907. assert.Equal(t, "alice", upserts[0].Name)
  908. })
  909. }
  910. func TestFeedbagList_DeleteLinkedScreenName(t *testing.T) {
  911. t.Run("existing linked screen name is deleted, root group created and pending", func(t *testing.T) {
  912. item := wire.FeedbagItem{ClassID: wire.FeedbagClassIdAlInfo, Name: "alice", ItemID: 1}
  913. fl := NewFeedbagList([]wire.FeedbagItem{item}, nil)
  914. fl.DeleteLinkedScreenName("alice")
  915. deletes := fl.PendingDeletes()
  916. assert.Len(t, deletes, 1)
  917. assert.Equal(t, item, deletes[0])
  918. upserts := fl.PendingUpdates()
  919. assert.Len(t, upserts, 1)
  920. assert.Equal(t, wire.FeedbagClassIdGroup, upserts[0].ClassID)
  921. assert.Equal(t, uint16(0), upserts[0].GroupID)
  922. })
  923. t.Run("existing root group is touched on delete", func(t *testing.T) {
  924. root := wire.FeedbagItem{ClassID: wire.FeedbagClassIdGroup, GroupID: 0, ItemID: 1}
  925. item := wire.FeedbagItem{ClassID: wire.FeedbagClassIdAlInfo, Name: "alice", ItemID: 2}
  926. fl := NewFeedbagList([]wire.FeedbagItem{root, item}, nil)
  927. fl.DeleteLinkedScreenName("alice")
  928. upserts := fl.PendingUpdates()
  929. assert.Len(t, upserts, 1)
  930. assert.Equal(t, wire.FeedbagClassIdGroup, upserts[0].ClassID)
  931. assert.Equal(t, uint16(0), upserts[0].GroupID)
  932. })
  933. t.Run("deleting non-existent linked screen name is a no-op", func(t *testing.T) {
  934. fl := NewFeedbagList(nil, nil)
  935. fl.DeleteLinkedScreenName("alice")
  936. assert.Empty(t, fl.PendingDeletes())
  937. assert.Nil(t, fl.PendingUpdates())
  938. })
  939. t.Run("name comparison is case-insensitive", func(t *testing.T) {
  940. item := wire.FeedbagItem{ClassID: wire.FeedbagClassIdAlInfo, Name: "alice", ItemID: 1}
  941. fl := NewFeedbagList([]wire.FeedbagItem{item}, nil)
  942. fl.DeleteLinkedScreenName("Alice")
  943. assert.Len(t, fl.PendingDeletes(), 1)
  944. })
  945. }
  946. func TestFeedbagList_LinkedScreenNames(t *testing.T) {
  947. t.Run("returns all linked screen names", func(t *testing.T) {
  948. fl := NewFeedbagList([]wire.FeedbagItem{
  949. {ClassID: wire.FeedbagClassIdAlInfo, Name: "alice", ItemID: 1},
  950. {ClassID: wire.FeedbagClassIdAlInfo, Name: "bob", ItemID: 2},
  951. {ClassID: wire.FeedbagClassIdBuddy, Name: "carol", ItemID: 3},
  952. }, nil)
  953. names := fl.LinkedScreenNames()
  954. assert.Equal(t, []IdentScreenName{
  955. NewIdentScreenName("alice"),
  956. NewIdentScreenName("bob"),
  957. }, names)
  958. })
  959. t.Run("returns nil when no linked screen names exist", func(t *testing.T) {
  960. fl := NewFeedbagList(nil, nil)
  961. assert.Nil(t, fl.LinkedScreenNames())
  962. })
  963. }
  964. func TestFeedbagList_HasLinkedScreenName(t *testing.T) {
  965. t.Run("returns true when linked screen name exists", func(t *testing.T) {
  966. fl := NewFeedbagList([]wire.FeedbagItem{
  967. {ClassID: wire.FeedbagClassIdAlInfo, Name: "alice", ItemID: 1},
  968. }, nil)
  969. assert.True(t, fl.HasLinkedScreenName("alice"))
  970. })
  971. t.Run("returns false when linked screen name does not exist", func(t *testing.T) {
  972. fl := NewFeedbagList(nil, nil)
  973. assert.False(t, fl.HasLinkedScreenName("alice"))
  974. })
  975. t.Run("match is case-insensitive", func(t *testing.T) {
  976. fl := NewFeedbagList([]wire.FeedbagItem{
  977. {ClassID: wire.FeedbagClassIdAlInfo, Name: "alice", ItemID: 1},
  978. }, nil)
  979. assert.True(t, fl.HasLinkedScreenName("Alice"))
  980. })
  981. }
  982. func TestFeedbagList_RenameGroup(t *testing.T) {
  983. t.Run("renames group in place preserving IDs", func(t *testing.T) {
  984. fl := NewFeedbagList([]wire.FeedbagItem{
  985. {Name: "Coworkers", ClassID: wire.FeedbagClassIdGroup, GroupID: 5, ItemID: 0},
  986. }, nil)
  987. err := fl.RenameGroup("Coworkers", "Colleagues")
  988. assert.NoError(t, err)
  989. upserts := fl.PendingUpdates()
  990. assert.Len(t, upserts, 1)
  991. assert.Equal(t, "Colleagues", upserts[0].Name)
  992. assert.Equal(t, uint16(5), upserts[0].GroupID)
  993. assert.Equal(t, wire.FeedbagClassIdGroup, upserts[0].ClassID)
  994. })
  995. t.Run("returns ErrGroupNotFound for missing group", func(t *testing.T) {
  996. fl := NewFeedbagList(nil, nil)
  997. err := fl.RenameGroup("Nope", "New")
  998. assert.ErrorIs(t, err, ErrGroupNotFound)
  999. assert.Nil(t, fl.PendingUpdates())
  1000. })
  1001. t.Run("returns ErrGroupExists when target name taken", func(t *testing.T) {
  1002. fl := NewFeedbagList([]wire.FeedbagItem{
  1003. {Name: "A", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  1004. {Name: "B", ClassID: wire.FeedbagClassIdGroup, GroupID: 2},
  1005. }, nil)
  1006. err := fl.RenameGroup("A", "B")
  1007. assert.ErrorIs(t, err, ErrGroupExists)
  1008. assert.Nil(t, fl.PendingUpdates())
  1009. })
  1010. t.Run("renaming to same name is a no-op", func(t *testing.T) {
  1011. fl := NewFeedbagList([]wire.FeedbagItem{
  1012. {Name: "A", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  1013. }, nil)
  1014. err := fl.RenameGroup("A", "A")
  1015. assert.NoError(t, err)
  1016. assert.Nil(t, fl.PendingUpdates())
  1017. })
  1018. }
  1019. func TestFeedbagList_MoveBuddy(t *testing.T) {
  1020. t.Run("moves buddy across groups carrying all attributes", func(t *testing.T) {
  1021. fl := NewFeedbagList([]wire.FeedbagItem{
  1022. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1,
  1023. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{
  1024. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{10}),
  1025. }}},
  1026. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 10,
  1027. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{
  1028. wire.NewTLVBE(wire.FeedbagAttributesAlias, "Al"),
  1029. wire.NewTLVBE(wire.FeedbagAttributesNote, "friend"),
  1030. // a non-alias/note attribute that must survive the move
  1031. wire.NewTLVBE(wire.FeedbagAttributesPending, []byte{}),
  1032. }}},
  1033. {Name: "Coworkers", ClassID: wire.FeedbagClassIdGroup, GroupID: 2},
  1034. }, func(n int) int { return 99 })
  1035. err := fl.MoveBuddy("Buddies", "Coworkers", "alice", "")
  1036. assert.NoError(t, err)
  1037. deletes := fl.PendingDeletes()
  1038. assert.Len(t, deletes, 1)
  1039. assert.Equal(t, "alice", deletes[0].Name)
  1040. assert.Equal(t, uint16(1), deletes[0].GroupID)
  1041. upserts := fl.PendingUpdates()
  1042. var newBuddy *wire.FeedbagItem
  1043. for i := range upserts {
  1044. if upserts[i].ClassID == wire.FeedbagClassIdBuddy {
  1045. newBuddy = &upserts[i]
  1046. }
  1047. }
  1048. assert.NotNil(t, newBuddy)
  1049. assert.Equal(t, uint16(2), newBuddy.GroupID)
  1050. alias, ok := newBuddy.Bytes(wire.FeedbagAttributesAlias)
  1051. assert.True(t, ok)
  1052. assert.Equal(t, []byte("Al"), alias)
  1053. note, ok := newBuddy.Bytes(wire.FeedbagAttributesNote)
  1054. assert.True(t, ok)
  1055. assert.Equal(t, []byte("friend"), note)
  1056. // the non-alias/note attribute is preserved rather than stripped
  1057. assert.True(t, newBuddy.HasTag(wire.FeedbagAttributesPending))
  1058. })
  1059. t.Run("reorders within a group before another buddy", func(t *testing.T) {
  1060. fl := NewFeedbagList([]wire.FeedbagItem{
  1061. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1,
  1062. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{
  1063. wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{10, 20, 30}),
  1064. }}},
  1065. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 10},
  1066. {Name: "bob", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 20},
  1067. {Name: "carol", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 30},
  1068. }, nil)
  1069. // move carol (30) before alice (10)
  1070. err := fl.MoveBuddy("Buddies", "", "carol", "alice")
  1071. assert.NoError(t, err)
  1072. upserts := fl.PendingUpdates()
  1073. assert.Len(t, upserts, 1)
  1074. order, ok := upserts[0].Uint16SliceBE(wire.FeedbagAttributesOrder)
  1075. assert.True(t, ok)
  1076. assert.Equal(t, []uint16{30, 10, 20}, order)
  1077. })
  1078. t.Run("returns ErrBuddyNotFound when buddy missing", func(t *testing.T) {
  1079. fl := NewFeedbagList([]wire.FeedbagItem{
  1080. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  1081. }, nil)
  1082. err := fl.MoveBuddy("Buddies", "", "ghost", "")
  1083. assert.ErrorIs(t, err, ErrBuddyNotFound)
  1084. })
  1085. t.Run("returns ErrGroupNotFound when destination missing", func(t *testing.T) {
  1086. fl := NewFeedbagList([]wire.FeedbagItem{
  1087. {Name: "Buddies", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  1088. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 10},
  1089. }, nil)
  1090. err := fl.MoveBuddy("Buddies", "Nowhere", "alice", "")
  1091. assert.ErrorIs(t, err, ErrGroupNotFound)
  1092. })
  1093. }
  1094. func TestFeedbagList_SetBuddyAlias(t *testing.T) {
  1095. t.Run("sets alias on all matching buddies", func(t *testing.T) {
  1096. fl := NewFeedbagList([]wire.FeedbagItem{
  1097. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 10},
  1098. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 2, ItemID: 20},
  1099. }, nil)
  1100. found, err := fl.SetBuddyAlias("alice", "Al")
  1101. assert.NoError(t, err)
  1102. assert.True(t, found)
  1103. upserts := fl.PendingUpdates()
  1104. assert.Len(t, upserts, 2)
  1105. for _, item := range upserts {
  1106. alias, ok := item.Bytes(wire.FeedbagAttributesAlias)
  1107. assert.True(t, ok)
  1108. assert.Equal(t, []byte("Al"), alias)
  1109. }
  1110. })
  1111. t.Run("clears alias when empty", func(t *testing.T) {
  1112. fl := NewFeedbagList([]wire.FeedbagItem{
  1113. {Name: "alice", ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, ItemID: 10,
  1114. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{
  1115. wire.NewTLVBE(wire.FeedbagAttributesAlias, "Al"),
  1116. }}},
  1117. }, nil)
  1118. found, err := fl.SetBuddyAlias("alice", "")
  1119. assert.NoError(t, err)
  1120. assert.True(t, found)
  1121. upserts := fl.PendingUpdates()
  1122. assert.Len(t, upserts, 1)
  1123. assert.False(t, upserts[0].HasTag(wire.FeedbagAttributesAlias))
  1124. })
  1125. t.Run("returns false when buddy not found", func(t *testing.T) {
  1126. fl := NewFeedbagList(nil, nil)
  1127. found, err := fl.SetBuddyAlias("ghost", "X")
  1128. assert.NoError(t, err)
  1129. assert.False(t, found)
  1130. assert.Nil(t, fl.PendingUpdates())
  1131. })
  1132. }
  1133. func TestFeedbagList_SetGroupCollapsed(t *testing.T) {
  1134. t.Run("sets collapsed attribute", func(t *testing.T) {
  1135. fl := NewFeedbagList([]wire.FeedbagItem{
  1136. {Name: "Coworkers", ClassID: wire.FeedbagClassIdGroup, GroupID: 1},
  1137. }, nil)
  1138. err := fl.SetGroupCollapsed("Coworkers", true)
  1139. assert.NoError(t, err)
  1140. upserts := fl.PendingUpdates()
  1141. assert.Len(t, upserts, 1)
  1142. assert.True(t, upserts[0].HasTag(wire.FeedbagAttributesCollapsed))
  1143. })
  1144. t.Run("clears collapsed attribute", func(t *testing.T) {
  1145. fl := NewFeedbagList([]wire.FeedbagItem{
  1146. {Name: "Coworkers", ClassID: wire.FeedbagClassIdGroup, GroupID: 1,
  1147. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{
  1148. wire.NewTLVBE(wire.FeedbagAttributesCollapsed, []byte{}),
  1149. }}},
  1150. }, nil)
  1151. err := fl.SetGroupCollapsed("Coworkers", false)
  1152. assert.NoError(t, err)
  1153. upserts := fl.PendingUpdates()
  1154. assert.Len(t, upserts, 1)
  1155. assert.False(t, upserts[0].HasTag(wire.FeedbagAttributesCollapsed))
  1156. })
  1157. t.Run("targets unnamed default group, not the root group", func(t *testing.T) {
  1158. fl := NewFeedbagList([]wire.FeedbagItem{
  1159. {Name: "", ClassID: wire.FeedbagClassIdGroup, GroupID: 0},
  1160. {Name: "", ClassID: wire.FeedbagClassIdGroup, GroupID: 3},
  1161. }, nil)
  1162. err := fl.SetGroupCollapsed("", true)
  1163. assert.NoError(t, err)
  1164. upserts := fl.PendingUpdates()
  1165. assert.Len(t, upserts, 1)
  1166. assert.Equal(t, uint16(3), upserts[0].GroupID)
  1167. assert.True(t, upserts[0].HasTag(wire.FeedbagAttributesCollapsed))
  1168. })
  1169. t.Run("returns ErrGroupNotFound for missing group", func(t *testing.T) {
  1170. fl := NewFeedbagList(nil, nil)
  1171. err := fl.SetGroupCollapsed("Nope", true)
  1172. assert.ErrorIs(t, err, ErrGroupNotFound)
  1173. })
  1174. }