/server/poll/poll_test.go

https://github.com/matterpoll/matterpoll · Go · 799 lines · 758 code · 41 blank · 0 comment · 15 complexity · aefa390456092f27bd320fc9dce4b239 MD5 · raw file 

    

  1. package poll_test
    2. 

  2. 

  3. import (
    4. 

  4. 	"fmt"
    5. 

  5. 	"testing"
    6. 

  6. 

  7. 	"bou.ke/monkey"
    8. 

  8. 	"github.com/mattermost/mattermost-server/v5/model"
    9. 

  9. 	"github.com/stretchr/testify/assert"
    10. 

  10. 	"github.com/stretchr/testify/require"
    11. 

  11. 

  12. 	"github.com/matterpoll/matterpoll/server/poll"
    13. 

  13. 	"github.com/matterpoll/matterpoll/server/utils/testutils"
    14. 

  14. )
    15. 

  15. 

  16. func TestNewPoll(t *testing.T) {
    17. 

  17. 	t.Run("all fine", func(t *testing.T) {
    18. 

  18. 		assert := assert.New(t)
    19. 

  19. 		patch1 := monkey.Patch(model.GetMillis, func() int64 { return 1234567890 })
    20. 

  20. 		patch2 := monkey.Patch(model.NewId, testutils.GetPollID)
    21. 

  21. 		defer patch1.Unpatch()
    22. 

  22. 		defer patch2.Unpatch()
    23. 

  23. 

  24. 		creator := model.NewRandomString(10)
    25. 

  25. 		question := model.NewRandomString(10)
    26. 

  26. 		answerOptions := []string{model.NewRandomString(10), model.NewRandomString(10), model.NewRandomString(10)}
    27. 

  27. 		p, err := poll.NewPoll(creator, question, answerOptions, poll.Settings{
    28. 

  28. 			Anonymous:       true,
    29. 

  29. 			Progress:        true,
    30. 

  30. 			PublicAddOption: true,
    31. 

  31. 			MaxVotes:        3,
    32. 

  32. 		})
    33. 

  33. 

  34. 		require.Nil(t, err)
    35. 

  35. 		require.NotNil(t, p)
    36. 

  36. 		assert.Equal(testutils.GetPollID(), p.ID)
    37. 

  37. 		assert.Equal(int64(1234567890), p.CreatedAt)
    38. 

  38. 		assert.Equal(creator, p.Creator)
    39. 

  39. 		assert.Equal(question, p.Question)
    40. 

  40. 		assert.Equal(&poll.AnswerOption{Answer: answerOptions[0], Voter: []string{}}, p.AnswerOptions[0])
    41. 

  41. 		assert.Equal(&poll.AnswerOption{Answer: answerOptions[1], Voter: []string{}}, p.AnswerOptions[1])
    42. 

  42. 		assert.Equal(&poll.AnswerOption{Answer: answerOptions[2], Voter: []string{}}, p.AnswerOptions[2])
    43. 

  43. 		assert.Equal(poll.Settings{Anonymous: true, Progress: true, PublicAddOption: true, MaxVotes: 3}, p.Settings)
    44. 

  44. 	})
    45. 

  45. 

  46. 	t.Run("error, invalid votes setting", func(t *testing.T) {
    47. 

  47. 		assert := assert.New(t)
    48. 

  48. 		patch1 := monkey.Patch(model.GetMillis, func() int64 { return 1234567890 })
    49. 

  49. 		patch2 := monkey.Patch(model.NewId, testutils.GetPollID)
    50. 

  50. 		defer patch1.Unpatch()
    51. 

  51. 		defer patch2.Unpatch()
    52. 

  52. 

  53. 		creator := model.NewRandomString(10)
    54. 

  54. 		question := model.NewRandomString(10)
    55. 

  55. 		answerOptions := []string{model.NewRandomString(10), model.NewRandomString(10), model.NewRandomString(10)}
    56. 

  56. 		p, err := poll.NewPoll(creator, question, answerOptions, poll.Settings{
    57. 

  57. 			Anonymous:       true,
    58. 

  58. 			Progress:        true,
    59. 

  59. 			PublicAddOption: true,
    60. 

  60. 			MaxVotes:        4,
    61. 

  61. 		})
    62. 

  62. 

  63. 		assert.Nil(p)
    64. 

  64. 		assert.NotNil(err)
    65. 

  65. 	})
    66. 

  66. 

  67. 	t.Run("error, duplicate option", func(t *testing.T) {
    68. 

  68. 		assert := assert.New(t)
    69. 

  69. 

  70. 		creator := model.NewRandomString(10)
    71. 

  71. 		question := model.NewRandomString(10)
    72. 

  72. 		option := model.NewRandomString(10)
    73. 

  73. 		answerOptions := []string{option, model.NewRandomString(10), option}
    74. 

  74. 		p, err := poll.NewPoll(creator, question, answerOptions, poll.Settings{MaxVotes: 1})
    75. 

  75. 

  76. 		assert.Nil(p)
    77. 

  77. 		assert.NotNil(err)
    78. 

  78. 	})
    79. 

  79. }
    80. 

  80. 

  81. func TestNewSettingsFromStrings(t *testing.T) {
    82. 

  82. 	for name, test := range map[string]struct {
    83. 

  83. 		Strs             []string
    84. 

  84. 		ShouldError      bool
    85. 

  85. 		ExpectedSettings poll.Settings
    86. 

  86. 	}{
    87. 

  87. 		"no settings": {
    88. 

  88. 			Strs:        []string{},
    89. 

  89. 			ShouldError: false,
    90. 

  90. 			ExpectedSettings: poll.Settings{
    91. 

  91. 				Anonymous:       false,
    92. 

  92. 				Progress:        false,
    93. 

  93. 				PublicAddOption: false,
    94. 

  94. 				MaxVotes:        1,
    95. 

  95. 			},
    96. 

  96. 		},
    97. 

  97. 		"full settings": {
    98. 

  98. 			Strs:        []string{"anonymous", "progress", "public-add-option", "votes=4"},
    99. 

  99. 			ShouldError: false,
    100. 

  100. 			ExpectedSettings: poll.Settings{
    101. 

  101. 				Anonymous:       true,
    102. 

  102. 				Progress:        true,
    103. 

  103. 				PublicAddOption: true,
    104. 

  104. 				MaxVotes:        4,
    105. 

  105. 			},
    106. 

  106. 		},
    107. 

  107. 		"without votes settings": {
    108. 

  108. 			Strs:        []string{"anonymous", "progress", "public-add-option"},
    109. 

  109. 			ShouldError: false,
    110. 

  110. 			ExpectedSettings: poll.Settings{
    111. 

  111. 				Anonymous:       true,
    112. 

  112. 				Progress:        true,
    113. 

  113. 				PublicAddOption: true,
    114. 

  114. 				MaxVotes:        1,
    115. 

  115. 			},
    116. 

  116. 		},
    117. 

  117. 		"invalid votes setting": {
    118. 

  118. 			Strs:        []string{"votes=9223372036854775808"}, // Exceed math.MaxInt64
    119. 

  119. 			ShouldError: true,
    120. 

  120. 			ExpectedSettings: poll.Settings{
    121. 

  121. 				Anonymous:       false,
    122. 

  122. 				Progress:        false,
    123. 

  123. 				PublicAddOption: false,
    124. 

  124. 				MaxVotes:        1,
    125. 

  125. 			},
    126. 

  126. 		},
    127. 

  127. 		"invalid setting": {
    128. 

  128. 			Strs:        []string{"anonymous", "progress", "public-add-option", "invalid"},
    129. 

  129. 			ShouldError: true,
    130. 

  130. 			ExpectedSettings: poll.Settings{
    131. 

  131. 				Anonymous:       true,
    132. 

  132. 				Progress:        true,
    133. 

  133. 				PublicAddOption: true,
    134. 

  134. 				MaxVotes:        1,
    135. 

  135. 			},
    136. 

  136. 		},
    137. 

  137. 	} {
    138. 

  138. 		t.Run(name, func(t *testing.T) {
    139. 

  139. 			assert := assert.New(t)
    140. 

  140. 

  141. 			settings, errMsg := poll.NewSettingsFromStrings(test.Strs)
    142. 

  142. 			if test.ShouldError {
    143. 

  143. 				assert.NotNil(errMsg)
    144. 

  144. 			} else {
    145. 

  145. 				assert.Nil(errMsg)
    146. 

  146. 			}
    147. 

  147. 			assert.Equal(test.ExpectedSettings, settings)
    148. 

  148. 		})
    149. 

  149. 	}
    150. 

  150. }
    151. 

  151. 

  152. func TestNewSettingsFromSubmission(t *testing.T) {
    153. 

  153. 	for name, test := range map[string]struct {
    154. 

  154. 		Submission       map[string]interface{}
    155. 

  155. 		ExpectedSettings poll.Settings
    156. 

  156. 	}{
    157. 

  157. 		"no settings": {
    158. 

  158. 			Submission: map[string]interface{}{},
    159. 

  159. 			ExpectedSettings: poll.Settings{
    160. 

  160. 				Anonymous:       false,
    161. 

  161. 				Progress:        false,
    162. 

  162. 				PublicAddOption: false,
    163. 

  163. 				MaxVotes:        1,
    164. 

  164. 			},
    165. 

  165. 		},
    166. 

  166. 		"full settings": {
    167. 

  167. 			Submission: map[string]interface{}{
    168. 

  168. 				"setting-anonymous":         true,
    169. 

  169. 				"setting-progress":          true,
    170. 

  170. 				"setting-public-add-option": true,
    171. 

  171. 				"setting-multi":             float64(4),
    172. 

  172. 			},
    173. 

  173. 			ExpectedSettings: poll.Settings{
    174. 

  174. 				Anonymous:       true,
    175. 

  175. 				Progress:        true,
    176. 

  176. 				PublicAddOption: true,
    177. 

  177. 				MaxVotes:        4,
    178. 

  178. 			},
    179. 

  179. 		},
    180. 

  180. 		"without votes settings": {
    181. 

  181. 			Submission: map[string]interface{}{
    182. 

  182. 				"setting-anonymous":         false,
    183. 

  183. 				"setting-progress":          false,
    184. 

  184. 				"setting-public-add-option": false,
    185. 

  185. 			},
    186. 

  186. 			ExpectedSettings: poll.Settings{
    187. 

  187. 				Anonymous:       false,
    188. 

  188. 				Progress:        false,
    189. 

  189. 				PublicAddOption: false,
    190. 

  190. 				MaxVotes:        1,
    191. 

  191. 			},
    192. 

  192. 		},
    193. 

  193. 	} {
    194. 

  194. 		t.Run(name, func(t *testing.T) {
    195. 

  195. 			assert := assert.New(t)
    196. 

  196. 

  197. 			settings := poll.NewSettingsFromSubmission(test.Submission)
    198. 

  198. 			assert.Equal(test.ExpectedSettings, settings)
    199. 

  199. 		})
    200. 

  200. 	}
    201. 

  201. }
    202. 

  202. 

  203. func TestAddAnswerOption(t *testing.T) {
    204. 

  204. 	assert := assert.New(t)
    205. 

  205. 

  206. 	t.Run("all fine", func(t *testing.T) {
    207. 

  207. 		p := testutils.GetPollWithVotes()
    208. 

  208. 

  209. 		err := p.AddAnswerOption("new option")
    210. 

  210. 		assert.Nil(err)
    211. 

  211. 		assert.Equal("new option", p.AnswerOptions[len(p.AnswerOptions)-1].Answer)
    212. 

  212. 	})
    213. 

  213. 	t.Run("dublicant options", func(t *testing.T) {
    214. 

  214. 		p := testutils.GetPollWithVotes()
    215. 

  215. 

  216. 		err := p.AddAnswerOption(p.AnswerOptions[0].Answer)
    217. 

  217. 		assert.NotNil(err)
    218. 

  218. 	})
    219. 

  219. 	t.Run("dublicant options with spaces", func(t *testing.T) {
    220. 

  220. 		p := testutils.GetPollWithVotes()
    221. 

  221. 

  222. 		err := p.AddAnswerOption(fmt.Sprintf(" %s ", p.AnswerOptions[0].Answer))
    223. 

  223. 		assert.NotNil(err)
    224. 

  224. 	})
    225. 

  225. 	t.Run("empty options", func(t *testing.T) {
    226. 

  226. 		p := testutils.GetPollWithVotes()
    227. 

  227. 

  228. 		err := p.AddAnswerOption("")
    229. 

  229. 		assert.NotNil(err)
    230. 

  230. 	})
    231. 

  231. 	t.Run("empty optinos with spaces", func(t *testing.T) {
    232. 

  232. 		p := testutils.GetPollWithVotes()
    233. 

  233. 

  234. 		err := p.AddAnswerOption("  ")
    235. 

  235. 		assert.NotNil(err)
    236. 

  236. 	})
    237. 

  237. }
    238. 

  238. 

  239. func TestEncodeDecode(t *testing.T) {
    240. 

  240. 	p1 := testutils.GetPollWithVotes()
    241. 

  241. 	p2 := poll.DecodePollFromByte(p1.EncodeToByte())
    242. 

  242. 	assert.Equal(t, p1, p2)
    243. 

  243. }
    244. 

  244. 

  245. func TestDecode(t *testing.T) {
    246. 

  246. 	p := poll.DecodePollFromByte([]byte{})
    247. 

  247. 	assert.Nil(t, p)
    248. 

  248. }
    249. 

  249. 

  250. func TestUpdateVote(t *testing.T) {
    251. 

  251. 	for name, test := range map[string]struct {
    252. 

  252. 		Poll          poll.Poll
    253. 

  253. 		UserID        string
    254. 

  254. 		Index         int
    255. 

  255. 		ExpectedPoll  poll.Poll
    256. 

  256. 		Error         bool
    257. 

  257. 		ReturnMessage bool
    258. 

  258. 	}{
    259. 

  259. 		"Negative Index": {
    260. 

  260. 			Poll: poll.Poll{
    261. 

  261. 				Question: "Question",
    262. 

  262. 				AnswerOptions: []*poll.AnswerOption{
    263. 

  263. 					{Answer: "Answer 1",
    264. 

  264. 						Voter: []string{"a"}},
    265. 

  265. 					{Answer: "Answer 2"},
    266. 

  266. 				},
    267. 

  267. 			},
    268. 

  268. 			UserID: "a",
    269. 

  269. 			Index:  -1,
    270. 

  270. 			ExpectedPoll: poll.Poll{
    271. 

  271. 				Question: "Question",
    272. 

  272. 				AnswerOptions: []*poll.AnswerOption{
    273. 

  273. 					{Answer: "Answer 1",
    274. 

  274. 						Voter: []string{"a"}},
    275. 

  275. 					{Answer: "Answer 2"},
    276. 

  276. 				},
    277. 

  277. 			},
    278. 

  278. 			Error:         true,
    279. 

  279. 			ReturnMessage: false,
    280. 

  280. 		},
    281. 

  281. 		"To high Index": {
    282. 

  282. 			Poll: poll.Poll{
    283. 

  283. 				Question: "Question",
    284. 

  284. 				AnswerOptions: []*poll.AnswerOption{
    285. 

  285. 					{Answer: "Answer 1",
    286. 

  286. 						Voter: []string{"a"}},
    287. 

  287. 					{Answer: "Answer 2"},
    288. 

  288. 				},
    289. 

  289. 			},
    290. 

  290. 			UserID: "a",
    291. 

  291. 			Index:  2,
    292. 

  292. 			ExpectedPoll: poll.Poll{
    293. 

  293. 				Question: "Question",
    294. 

  294. 				AnswerOptions: []*poll.AnswerOption{
    295. 

  295. 					{Answer: "Answer 1",
    296. 

  296. 						Voter: []string{"a"}},
    297. 

  297. 					{Answer: "Answer 2"},
    298. 

  298. 				},
    299. 

  299. 			},
    300. 

  300. 			Error:         true,
    301. 

  301. 			ReturnMessage: false,
    302. 

  302. 		},
    303. 

  303. 		"Invalid userID": {
    304. 

  304. 			Poll: poll.Poll{
    305. 

  305. 				Question: "Question",
    306. 

  306. 				AnswerOptions: []*poll.AnswerOption{
    307. 

  307. 					{Answer: "Answer 1",
    308. 

  308. 						Voter: []string{"a"}},
    309. 

  309. 					{Answer: "Answer 2"},
    310. 

  310. 				},
    311. 

  311. 			},
    312. 

  312. 			UserID: "",
    313. 

  313. 			Index:  1,
    314. 

  314. 			ExpectedPoll: poll.Poll{
    315. 

  315. 				Question: "Question",
    316. 

  316. 				AnswerOptions: []*poll.AnswerOption{
    317. 

  317. 					{Answer: "Answer 1",
    318. 

  318. 						Voter: []string{"a"}},
    319. 

  319. 					{Answer: "Answer 2"},
    320. 

  320. 				},
    321. 

  321. 			},
    322. 

  322. 			Error:         true,
    323. 

  323. 			ReturnMessage: false,
    324. 

  324. 		},
    325. 

  325. 		"Idempotent": {
    326. 

  326. 			Poll: poll.Poll{
    327. 

  327. 				Question: "Question",
    328. 

  328. 				AnswerOptions: []*poll.AnswerOption{
    329. 

  329. 					{Answer: "Answer 1",
    330. 

  330. 						Voter: []string{"a"}},
    331. 

  331. 					{Answer: "Answer 2"},
    332. 

  332. 				},
    333. 

  333. 			},
    334. 

  334. 			UserID: "a",
    335. 

  335. 			Index:  0,
    336. 

  336. 			ExpectedPoll: poll.Poll{
    337. 

  337. 				Question: "Question",
    338. 

  338. 				AnswerOptions: []*poll.AnswerOption{
    339. 

  339. 					{Answer: "Answer 1",
    340. 

  340. 						Voter: []string{"a"}},
    341. 

  341. 					{Answer: "Answer 2"},
    342. 

  342. 				},
    343. 

  343. 			},
    344. 

  344. 			Error:         false,
    345. 

  345. 			ReturnMessage: false,
    346. 

  346. 		},
    347. 

  347. 		"Valid Vote": {
    348. 

  348. 			Poll: poll.Poll{
    349. 

  349. 				Question: "Question",
    350. 

  350. 				AnswerOptions: []*poll.AnswerOption{
    351. 

  351. 					{Answer: "Answer 1",
    352. 

  352. 						Voter: []string{"a"}},
    353. 

  353. 					{Answer: "Answer 2"},
    354. 

  354. 				},
    355. 

  355. 			},
    356. 

  356. 			UserID: "a",
    357. 

  357. 			Index:  1,
    358. 

  358. 			ExpectedPoll: poll.Poll{
    359. 

  359. 				Question: "Question",
    360. 

  360. 				AnswerOptions: []*poll.AnswerOption{
    361. 

  361. 					{Answer: "Answer 1",
    362. 

  362. 						Voter: []string{}},
    363. 

  363. 					{Answer: "Answer 2",
    364. 

  364. 						Voter: []string{"a"}},
    365. 

  365. 				},
    366. 

  366. 			},
    367. 

  367. 			Error:         false,
    368. 

  368. 			ReturnMessage: false,
    369. 

  369. 		},
    370. 

  370. 		"Multi votes setting, first vote": {
    371. 

  371. 			Poll: poll.Poll{
    372. 

  372. 				Question: "Question",
    373. 

  373. 				AnswerOptions: []*poll.AnswerOption{
    374. 

  374. 					{Answer: "Answer 1"},
    375. 

  375. 					{Answer: "Answer 2"},
    376. 

  376. 					{Answer: "Answer 3"},
    377. 

  377. 				},
    378. 

  378. 				Settings: poll.Settings{MaxVotes: 2},
    379. 

  379. 			},
    380. 

  380. 			UserID: "a",
    381. 

  381. 			Index:  0,
    382. 

  382. 			ExpectedPoll: poll.Poll{
    383. 

  383. 				Question: "Question",
    384. 

  384. 				AnswerOptions: []*poll.AnswerOption{
    385. 

  385. 					{Answer: "Answer 1", Voter: []string{"a"}},
    386. 

  386. 					{Answer: "Answer 2"},
    387. 

  387. 					{Answer: "Answer 3"},
    388. 

  388. 				},
    389. 

  389. 				Settings: poll.Settings{MaxVotes: 2},
    390. 

  390. 			},
    391. 

  391. 			Error:         false,
    392. 

  392. 			ReturnMessage: false,
    393. 

  393. 		},
    394. 

  394. 		"Multi votes setting, second vote": {
    395. 

  395. 			Poll: poll.Poll{
    396. 

  396. 				Question: "Question",
    397. 

  397. 				AnswerOptions: []*poll.AnswerOption{
    398. 

  398. 					{Answer: "Answer 1", Voter: []string{"a"}},
    399. 

  399. 					{Answer: "Answer 2"},
    400. 

  400. 					{Answer: "Answer 3"},
    401. 

  401. 				},
    402. 

  402. 				Settings: poll.Settings{MaxVotes: 2},
    403. 

  403. 			},
    404. 

  404. 			UserID: "a",
    405. 

  405. 			Index:  1,
    406. 

  406. 			ExpectedPoll: poll.Poll{
    407. 

  407. 				Question: "Question",
    408. 

  408. 				AnswerOptions: []*poll.AnswerOption{
    409. 

  409. 					{Answer: "Answer 1", Voter: []string{"a"}},
    410. 

  410. 					{Answer: "Answer 2", Voter: []string{"a"}},
    411. 

  411. 					{Answer: "Answer 3"},
    412. 

  412. 				},
    413. 

  413. 				Settings: poll.Settings{MaxVotes: 2},
    414. 

  414. 			},
    415. 

  415. 			Error:         false,
    416. 

  416. 			ReturnMessage: false,
    417. 

  417. 		},
    418. 

  418. 		"Multi votes setting, duplicated vote error": {
    419. 

  419. 			Poll: poll.Poll{
    420. 

  420. 				Question: "Question",
    421. 

  421. 				AnswerOptions: []*poll.AnswerOption{
    422. 

  422. 					{Answer: "Answer 1", Voter: []string{"a"}},
    423. 

  423. 					{Answer: "Answer 2"},
    424. 

  424. 					{Answer: "Answer 3"},
    425. 

  425. 				},
    426. 

  426. 				Settings: poll.Settings{MaxVotes: 2},
    427. 

  427. 			},
    428. 

  428. 			UserID: "a",
    429. 

  429. 			Index:  0,
    430. 

  430. 			ExpectedPoll: poll.Poll{
    431. 

  431. 				Question: "Question",
    432. 

  432. 				AnswerOptions: []*poll.AnswerOption{
    433. 

  433. 					{Answer: "Answer 1", Voter: []string{"a"}},
    434. 

  434. 					{Answer: "Answer 2"},
    435. 

  435. 					{Answer: "Answer 3"},
    436. 

  436. 				},
    437. 

  437. 				Settings: poll.Settings{MaxVotes: 2},
    438. 

  438. 			},
    439. 

  439. 			Error:         false,
    440. 

  440. 			ReturnMessage: true,
    441. 

  441. 		},
    442. 

  442. 		"Multi votes setting, with progress option, duplicated vote error": {
    443. 

  443. 			Poll: poll.Poll{
    444. 

  444. 				Question: "Question",
    445. 

  445. 				AnswerOptions: []*poll.AnswerOption{
    446. 

  446. 					{Answer: "Answer 1", Voter: []string{"a"}},
    447. 

  447. 					{Answer: "Answer 2"},
    448. 

  448. 					{Answer: "Answer 3"},
    449. 

  449. 				},
    450. 

  450. 				Settings: poll.Settings{Progress: true, MaxVotes: 2},
    451. 

  451. 			},
    452. 

  452. 			UserID: "a",
    453. 

  453. 			Index:  0,
    454. 

  454. 			ExpectedPoll: poll.Poll{
    455. 

  455. 				Question: "Question",
    456. 

  456. 				AnswerOptions: []*poll.AnswerOption{
    457. 

  457. 					{Answer: "Answer 1", Voter: []string{"a"}},
    458. 

  458. 					{Answer: "Answer 2"},
    459. 

  459. 					{Answer: "Answer 3"},
    460. 

  460. 				},
    461. 

  461. 				Settings: poll.Settings{Progress: true, MaxVotes: 2},
    462. 

  462. 			},
    463. 

  463. 			Error:         false,
    464. 

  464. 			ReturnMessage: true,
    465. 

  465. 		},
    466. 

  466. 		"Multi votes setting, exceed votes error": {
    467. 

  467. 			Poll: poll.Poll{
    468. 

  468. 				Question: "Question",
    469. 

  469. 				AnswerOptions: []*poll.AnswerOption{
    470. 

  470. 					{Answer: "Answer 1", Voter: []string{"a"}},
    471. 

  471. 					{Answer: "Answer 2", Voter: []string{"a"}},
    472. 

  472. 					{Answer: "Answer 3"},
    473. 

  473. 				},
    474. 

  474. 				Settings: poll.Settings{MaxVotes: 2},
    475. 

  475. 			},
    476. 

  476. 			UserID: "a",
    477. 

  477. 			Index:  2,
    478. 

  478. 			ExpectedPoll: poll.Poll{
    479. 

  479. 				Question: "Question",
    480. 

  480. 				AnswerOptions: []*poll.AnswerOption{
    481. 

  481. 					{Answer: "Answer 1", Voter: []string{"a"}},
    482. 

  482. 					{Answer: "Answer 2", Voter: []string{"a"}},
    483. 

  483. 					{Answer: "Answer 3"},
    484. 

  484. 				},
    485. 

  485. 				Settings: poll.Settings{MaxVotes: 2},
    486. 

  486. 			},
    487. 

  487. 			Error:         false,
    488. 

  488. 			ReturnMessage: true,
    489. 

  489. 		},
    490. 

  490. 		"Multi votes setting, invalid user id error": {
    491. 

  491. 			Poll: poll.Poll{
    492. 

  492. 				Question: "Question",
    493. 

  493. 				AnswerOptions: []*poll.AnswerOption{
    494. 

  494. 					{Answer: "Answer 1", Voter: []string{"a"}},
    495. 

  495. 					{Answer: "Answer 2", Voter: []string{"a"}},
    496. 

  496. 					{Answer: "Answer 3"},
    497. 

  497. 				},
    498. 

  498. 				Settings: poll.Settings{MaxVotes: 2},
    499. 

  499. 			},
    500. 

  500. 			UserID: "",
    501. 

  501. 			Index:  2,
    502. 

  502. 			ExpectedPoll: poll.Poll{
    503. 

  503. 				Question: "Question",
    504. 

  504. 				AnswerOptions: []*poll.AnswerOption{
    505. 

  505. 					{Answer: "Answer 1", Voter: []string{"a"}},
    506. 

  506. 					{Answer: "Answer 2", Voter: []string{"a"}},
    507. 

  507. 					{Answer: "Answer 3"},
    508. 

  508. 				},
    509. 

  509. 				Settings: poll.Settings{MaxVotes: 2},
    510. 

  510. 			},
    511. 

  511. 			Error:         true,
    512. 

  512. 			ReturnMessage: false,
    513. 

  513. 		},
    514. 

  514. 	} {
    515. 

  515. 		t.Run(name, func(t *testing.T) {
    516. 

  516. 			assert := assert.New(t)
    517. 

  517. 

  518. 			msg, err := test.Poll.UpdateVote(test.UserID, test.Index)
    519. 

  519. 

  520. 			if test.Error {
    521. 

  521. 				assert.NotNil(err)
    522. 

  522. 			} else {
    523. 

  523. 				assert.Nil(err)
    524. 

  524. 			}
    525. 

  525. 

  526. 			if test.ReturnMessage {
    527. 

  527. 				assert.NotNil(msg)
    528. 

  528. 			} else {
    529. 

  529. 				assert.Nil(msg)
    530. 

  530. 			}
    531. 

  531. 			assert.Equal(test.ExpectedPoll, test.Poll)
    532. 

  532. 		})
    533. 

  533. 	}
    534. 

  534. }
    535. 

  535. 

  536. func TestResetVotes(t *testing.T) {
    537. 

  537. 	for name, test := range map[string]struct {
    538. 

  538. 		Poll         poll.Poll
    539. 

  539. 		UserID       string
    540. 

  540. 		ExpectedPoll poll.Poll
    541. 

  541. 	}{
    542. 

  542. 		"Reset success, with votes": {
    543. 

  543. 			Poll: poll.Poll{
    544. 

  544. 				ID: testutils.GetPollID(),
    545. 

  545. 				AnswerOptions: []*poll.AnswerOption{
    546. 

  546. 					{Answer: "Answer 1", Voter: []string{"a"}},
    547. 

  547. 					{Answer: "Answer 2", Voter: []string{"a"}},
    548. 

  548. 					{Answer: "Answer 3", Voter: []string{"a"}},
    549. 

  549. 				},
    550. 

  550. 				Settings: poll.Settings{MaxVotes: 3},
    551. 

  551. 			},
    552. 

  552. 			UserID: "a",
    553. 

  553. 			ExpectedPoll: poll.Poll{
    554. 

  554. 				ID: testutils.GetPollID(),
    555. 

  555. 				AnswerOptions: []*poll.AnswerOption{
    556. 

  556. 					{Answer: "Answer 1", Voter: []string{}},
    557. 

  557. 					{Answer: "Answer 2", Voter: []string{}},
    558. 

  558. 					{Answer: "Answer 3", Voter: []string{}},
    559. 

  559. 				},
    560. 

  560. 				Settings: poll.Settings{MaxVotes: 3},
    561. 

  561. 			},
    562. 

  562. 		},
    563. 

  563. 		"Reset success, with no votes": {
    564. 

  564. 			Poll: poll.Poll{
    565. 

  565. 				ID: testutils.GetPollID(),
    566. 

  566. 				AnswerOptions: []*poll.AnswerOption{
    567. 

  567. 					{Answer: "Answer 1", Voter: []string{}},
    568. 

  568. 					{Answer: "Answer 2", Voter: []string{}},
    569. 

  569. 					{Answer: "Answer 3", Voter: []string{}},
    570. 

  570. 				},
    571. 

  571. 				Settings: poll.Settings{MaxVotes: 3},
    572. 

  572. 			},
    573. 

  573. 			UserID: "a",
    574. 

  574. 			ExpectedPoll: poll.Poll{
    575. 

  575. 				ID: testutils.GetPollID(),
    576. 

  576. 				AnswerOptions: []*poll.AnswerOption{
    577. 

  577. 					{Answer: "Answer 1", Voter: []string{}},
    578. 

  578. 					{Answer: "Answer 2", Voter: []string{}},
    579. 

  579. 					{Answer: "Answer 3", Voter: []string{}},
    580. 

  580. 				},
    581. 

  581. 				Settings: poll.Settings{MaxVotes: 3},
    582. 

  582. 			},
    583. 

  583. 		},
    584. 

  584. 		"Reset success, with votes from multi user": {
    585. 

  585. 			Poll: poll.Poll{
    586. 

  586. 				ID: testutils.GetPollID(),
    587. 

  587. 				AnswerOptions: []*poll.AnswerOption{
    588. 

  588. 					{Answer: "Answer 1", Voter: []string{"a", "b"}},
    589. 

  589. 					{Answer: "Answer 2", Voter: []string{"a"}},
    590. 

  590. 					{Answer: "Answer 3", Voter: []string{"1", "a", "z"}},
    591. 

  591. 				},
    592. 

  592. 				Settings: poll.Settings{MaxVotes: 3},
    593. 

  593. 			},
    594. 

  594. 			UserID: "a",
    595. 

  595. 			ExpectedPoll: poll.Poll{
    596. 

  596. 				ID: testutils.GetPollID(),
    597. 

  597. 				AnswerOptions: []*poll.AnswerOption{
    598. 

  598. 					{Answer: "Answer 1", Voter: []string{"b"}},
    599. 

  599. 					{Answer: "Answer 2", Voter: []string{}},
    600. 

  600. 					{Answer: "Answer 3", Voter: []string{"1", "z"}},
    601. 

  601. 				},
    602. 

  602. 				Settings: poll.Settings{MaxVotes: 3},
    603. 

  603. 			},
    604. 

  604. 		},
    605. 

  605. 		"invalid user id": {
    606. 

  606. 			Poll: poll.Poll{
    607. 

  607. 				ID: testutils.GetPollID(),
    608. 

  608. 				AnswerOptions: []*poll.AnswerOption{
    609. 

  609. 					{Answer: "Answer 1", Voter: []string{"a"}},
    610. 

  610. 					{Answer: "Answer 2", Voter: []string{"a"}},
    611. 

  611. 				},
    612. 

  612. 				Settings: poll.Settings{MaxVotes: 3},
    613. 

  613. 			},
    614. 

  614. 			UserID: "",
    615. 

  615. 			ExpectedPoll: poll.Poll{
    616. 

  616. 				ID: testutils.GetPollID(),
    617. 

  617. 				AnswerOptions: []*poll.AnswerOption{
    618. 

  618. 					{Answer: "Answer 1", Voter: []string{"a"}},
    619. 

  619. 					{Answer: "Answer 2", Voter: []string{"a"}},
    620. 

  620. 				},
    621. 

  621. 				Settings: poll.Settings{MaxVotes: 3},
    622. 

  622. 			},
    623. 

  623. 		},
    624. 

  624. 	} {
    625. 

  625. 		t.Run(name, func(t *testing.T) {
    626. 

  626. 			assert := assert.New(t)
    627. 

  627. 

  628. 			test.Poll.ResetVotes(test.UserID)
    629. 

  629. 			assert.Equal(test.ExpectedPoll, test.Poll)
    630. 

  630. 		})
    631. 

  631. 	}
    632. 

  632. }
    633. 

  633. 

  634. func TestGetMetadata(t *testing.T) {
    635. 

  635. 	for name, test := range map[string]struct {
    636. 

  636. 		Poll             poll.Poll
    637. 

  637. 		UserID           string
    638. 

  638. 		Permission       bool
    639. 

  639. 		ExpectedResponse *poll.Metadata
    640. 

  640. 	}{
    641. 

  641. 		"Voted an Answer": {
    642. 

  642. 			Poll: poll.Poll{
    643. 

  643. 				ID: testutils.GetPollID(),
    644. 

  644. 				AnswerOptions: []*poll.AnswerOption{
    645. 

  645. 					{Answer: "Answer 1", Voter: []string{"a"}},
    646. 

  646. 					{Answer: "Answer 2", Voter: []string{"b"}},
    647. 

  647. 					{Answer: "Answer 3", Voter: []string{"b"}},
    648. 

  648. 				},
    649. 

  649. 			},
    650. 

  650. 			UserID:     "a",
    651. 

  651. 			Permission: true,
    652. 

  652. 			ExpectedResponse: &poll.Metadata{
    653. 

  653. 				PollID:          testutils.GetPollID(),
    654. 

  654. 				UserID:          "a",
    655. 

  655. 				AdminPermission: true,
    656. 

  656. 				VotedAnswers:    []string{"Answer 1"},
    657. 

  657. 			},
    658. 

  658. 		},
    659. 

  659. 		"Voted two Answers": {
    660. 

  660. 			Poll: poll.Poll{
    661. 

  661. 				ID: testutils.GetPollID(),
    662. 

  662. 				AnswerOptions: []*poll.AnswerOption{
    663. 

  663. 					{Answer: "Answer 1", Voter: []string{"a"}},
    664. 

  664. 					{Answer: "Answer 2", Voter: []string{"b"}},
    665. 

  665. 					{Answer: "Answer 3", Voter: []string{"b"}},
    666. 

  666. 				},
    667. 

  667. 			},
    668. 

  668. 			UserID:     "b",
    669. 

  669. 			Permission: true,
    670. 

  670. 			ExpectedResponse: &poll.Metadata{
    671. 

  671. 				PollID:          testutils.GetPollID(),
    672. 

  672. 				UserID:          "b",
    673. 

  673. 				AdminPermission: true,
    674. 

  674. 				VotedAnswers:    []string{"Answer 2", "Answer 3"},
    675. 

  675. 			},
    676. 

  676. 		},
    677. 

  677. 		"Voted two Answers, with progress option": {
    678. 

  678. 			Poll: poll.Poll{
    679. 

  679. 				ID: testutils.GetPollID(),
    680. 

  680. 				AnswerOptions: []*poll.AnswerOption{
    681. 

  681. 					{Answer: "Answer 1", Voter: []string{"a"}},
    682. 

  682. 					{Answer: "Answer 2", Voter: []string{"a", "b"}},
    683. 

  683. 					{Answer: "Answer 3", Voter: []string{"b"}},
    684. 

  684. 				},
    685. 

  685. 				Settings: poll.Settings{
    686. 

  686. 					Progress: true,
    687. 

  687. 				},
    688. 

  688. 			},
    689. 

  689. 			UserID:     "b",
    690. 

  690. 			Permission: true,
    691. 

  691. 			ExpectedResponse: &poll.Metadata{
    692. 

  692. 				PollID:          testutils.GetPollID(),
    693. 

  693. 				UserID:          "b",
    694. 

  694. 				AdminPermission: true,
    695. 

  695. 				VotedAnswers:    []string{"Answer 2 (2)", "Answer 3 (1)"},
    696. 

  696. 			},
    697. 

  697. 		},
    698. 

  698. 		"Voted no Answers": {
    699. 

  699. 			Poll: poll.Poll{
    700. 

  700. 				ID: testutils.GetPollID(),
    701. 

  701. 				AnswerOptions: []*poll.AnswerOption{
    702. 

  702. 					{Answer: "Answer 1", Voter: []string{"a"}},
    703. 

  703. 					{Answer: "Answer 2", Voter: []string{"b"}},
    704. 

  704. 					{Answer: "Answer 3", Voter: []string{"b"}},
    705. 

  705. 				},
    706. 

  706. 			},
    707. 

  707. 			UserID:     "c",
    708. 

  708. 			Permission: true,
    709. 

  709. 			ExpectedResponse: &poll.Metadata{
    710. 

  710. 				PollID:          testutils.GetPollID(),
    711. 

  711. 				UserID:          "c",
    712. 

  712. 				AdminPermission: true,
    713. 

  713. 				VotedAnswers:    []string{},
    714. 

  714. 			}},
    715. 

  715. 		"Invalid userID": {
    716. 

  716. 			Poll: poll.Poll{
    717. 

  717. 				ID: testutils.GetPollID(),
    718. 

  718. 				AnswerOptions: []*poll.AnswerOption{
    719. 

  719. 					{Answer: "Answer 1", Voter: []string{"a"}},
    720. 

  720. 					{Answer: "Answer 2", Voter: []string{"b"}},
    721. 

  721. 					{Answer: "Answer 3", Voter: []string{"b"}},
    722. 

  722. 				},
    723. 

  723. 			},
    724. 

  724. 			UserID: "",
    725. 

  725. 			ExpectedResponse: &poll.Metadata{
    726. 

  726. 				PollID:          testutils.GetPollID(),
    727. 

  727. 				UserID:          "",
    728. 

  728. 				AdminPermission: false,
    729. 

  729. 				VotedAnswers:    []string{},
    730. 

  730. 			},
    731. 

  731. 		},
    732. 

  732. 	} {
    733. 

  733. 		t.Run(name, func(t *testing.T) {
    734. 

  734. 			assert := assert.New(t)
    735. 

  735. 

  736. 			metadata := test.Poll.GetMetadata(test.UserID, test.Permission)
    737. 

  737. 			assert.Equal(test.ExpectedResponse, metadata)
    738. 

  738. 		})
    739. 

  739. 	}
    740. 

  740. }
    741. 

  741. 

  742. func TestHasVoted(t *testing.T) {
    743. 

  743. 	p1 := &poll.Poll{Question: "Question",
    744. 

  744. 		AnswerOptions: []*poll.AnswerOption{
    745. 

  745. 			{Answer: "Answer 1", Voter: []string{"a"}},
    746. 

  746. 			{Answer: "Answer 2"},
    747. 

  747. 		},
    748. 

  748. 	}
    749. 

  749. 	assert.True(t, p1.HasVoted("a"))
    750. 

  750. 	assert.False(t, p1.HasVoted("b"))
    751. 

  751. }
    752. 

  752. 

  753. func TestPollCopy(t *testing.T) {
    754. 

  754. 	assert := assert.New(t)
    755. 

  755. 

  756. 	t.Run("no change", func(t *testing.T) {
    757. 

  757. 		p := testutils.GetPoll()
    758. 

  758. 		p2 := p.Copy()
    759. 

  759. 

  760. 		assert.Equal(p, p2)
    761. 

  761. 	})
    762. 

  762. 	t.Run("change Question", func(t *testing.T) {
    763. 

  763. 		p := testutils.GetPoll()
    764. 

  764. 		p2 := p.Copy()
    765. 

  765. 

  766. 		p.Question = "Different question"
    767. 

  767. 		assert.NotEqual(p.Question, p2.Question)
    768. 

  768. 		assert.NotEqual(p, p2)
    769. 

  769. 		assert.Equal(testutils.GetPoll(), p2)
    770. 

  770. 	})
    771. 

  771. 	t.Run("change AnswerOptions", func(t *testing.T) {
    772. 

  772. 		p := testutils.GetPoll()
    773. 

  773. 		p2 := p.Copy()
    774. 

  774. 

  775. 		p.AnswerOptions[0].Answer = "abc"
    776. 

  776. 		assert.NotEqual(p.AnswerOptions[0].Answer, p2.AnswerOptions[0].Answer)
    777. 

  777. 		assert.NotEqual(p, p2)
    778. 

  778. 		assert.Equal(testutils.GetPoll(), p2)
    779. 

  779. 	})
    780. 

  780. 	t.Run("change Voter", func(t *testing.T) {
    781. 

  781. 		p := testutils.GetPollWithVotes()
    782. 

  782. 		p2 := p.Copy()
    783. 

  783. 

  784. 		msg, err := p.UpdateVote("userID1", 0)
    785. 

  785. 		require.Nil(t, msg)
    786. 

  786. 		require.NoError(t, err)
    787. 

  787. 		assert.NotEqual(p, p2)
    788. 

  788. 		assert.Equal(testutils.GetPollWithVotes(), p2)
    789. 

  789. 	})
    790. 

  790. 	t.Run("change Settings", func(t *testing.T) {
    791. 

  791. 		p := testutils.GetPoll()
    792. 

  792. 		p2 := p.Copy()
    793. 

  793. 

  794. 		p.Settings.Progress = true
    795. 

  795. 		assert.NotEqual(p.Settings.Progress, p2.Settings.Progress)
    796. 

  796. 		assert.NotEqual(p, p2)
    797. 

  797. 		assert.Equal(testutils.GetPoll(), p2)
    798. 

  798. 	})
    799. 

  799. }
    800.