telebot/giveaway_test.go at v4 · tucnak/telebot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package telebot

import (
	"encoding/json"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestGiveawayPrizeStarCount(t *testing.T) {
	jsonData := `{
		"chats": [],
		"winners_selection_date": 1234567890,
		"winner_count": 5,
		"prize_star_count": 100
	}`

	var g Giveaway
	err := json.Unmarshal([]byte(jsonData), &g)
	assert.NoError(t, err)
	assert.Equal(t, 5, g.WinnerCount)
	assert.Equal(t, 100, g.PrizeStarCount)
}

func TestGiveawayCreatedPrizeStarCount(t *testing.T) {
	jsonData := `{
		"prize_star_count": 50
	}`

	var gc GiveawayCreated
	err := json.Unmarshal([]byte(jsonData), &gc)
	assert.NoError(t, err)
	assert.Equal(t, 50, gc.PrizeStarCount)
}

func TestGiveawayWinnersPrizeStarCount(t *testing.T) {
	jsonData := `{
		"chat": {
			"id": 123,
			"type": "private"
		},
		"giveaway_message_id": 456,
		"winners_selection_date": 1234567890,
		"winner_count": 3,
		"winners": [],
		"prize_star_count": 150
	}`

	var gw GiveawayWinners
	err := json.Unmarshal([]byte(jsonData), &gw)
	assert.NoError(t, err)
	assert.Equal(t, 3, gw.WinnerCount)
	assert.Equal(t, 150, gw.PrizeStarCount)
}

func TestGiveawayCompletedIsStarGiveaway(t *testing.T) {
	jsonData := `{
		"winner_count": 2,
		"is_star_giveaway": true
	}`

	var gc GiveawayCompleted
	err := json.Unmarshal([]byte(jsonData), &gc)
	assert.NoError(t, err)
	assert.Equal(t, 2, gc.WinnerCount)
	assert.True(t, gc.IsStarGiveaway)
}

func TestBoostSourcePrizeStarCount(t *testing.T) {
	jsonData := `{
		"source": "giveaway",
		"giveaway_message_id": 789,
		"prize_star_count": 200
	}`

	var bs BoostSource
	err := json.Unmarshal([]byte(jsonData), &bs)
	assert.NoError(t, err)
	assert.Equal(t, BoostGiveaway, bs.Source)
	assert.Equal(t, 789, bs.GiveawayMessageID)
	assert.Equal(t, 200, bs.PrizeStarCount)
}