-
-
Notifications
You must be signed in to change notification settings - Fork 59
/
conversation.go
57 lines (47 loc) · 1.35 KB
/
conversation.go
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
package fixgen
import (
"encoding/json"
"math/rand"
"strconv"
"time"
"github.com/rusq/slackdump/v2/internal/fixtures"
"github.com/rusq/slackdump/v2/types"
)
func init() {
// Seed the random number generator
rand.Seed(time.Now().UnixNano())
}
// ReSeed reseeds the random number generator
func ReSeed(n int64) {
rand.Seed(n)
}
// randString generates a random string of length n.
func randString(n int) string {
var letter = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
b := make([]rune, n)
for i := range b {
b[i] = letter[rand.Intn(len(letter))]
}
return string(b)
}
// GenerateTestConversation generates a test conversation with a random name.
func GenerateTestConversation(name string, startDate time.Time, endDate time.Time, numMessages int) types.Conversation {
var messages = make([]types.Message, numMessages)
for i := 0; i < numMessages; i++ {
messages[i] = GenerateTestMessage(startDate, endDate)
}
return types.Conversation{
Messages: messages,
Name: name,
ID: randString(9),
}
}
func GenerateTestMessage(startDate, endDate time.Time) types.Message {
var message types.Message
err := json.Unmarshal([]byte(fixtures.TestMessage), &message.Message)
if err != nil {
panic(err)
}
message.Timestamp = strconv.FormatInt(rand.Int63n(endDate.Unix()-startDate.Unix())+startDate.Unix(), 10)
return message
}