moths/token/code/newCode_test.go at main · Mobilpadde/moths · 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
package code_test

import (
	"testing"
	"time"

	"github.com/Mobilpadde/moths/v6/token/code"
	"github.com/Mobilpadde/moths/v6/token/emojies"
)

func TestNewCode(t *testing.T) {
	amount := 6
	token := "000000"
	correct := "😸😼😹😺😸😻"

	code, err := code.NewCode(token, amount, emojies.CATS, time.Time{}, time.Time{})
	if err != nil {
		t.Error("Expected to not return an error when creating new code:", err)
	}

	if code.Token() != token {
		t.Error("Expected token to be", token, "not", code.Token())
	}

	if code.String() != correct {
		t.Error("Expected code to be", correct, "not", code.String())
	}
}

func TestCreatedAt(t *testing.T) {
	amount := 6
	token := "000000"
	createdAt := time.Date(0, 0, 0, 0, 0, 0, 0, time.UTC)

	code, err := code.NewCode(token, amount, emojies.CATS, createdAt, time.Time{})
	if err != nil {
		t.Error("Expected to not return an error when creating new code:", err)
	}

	if code.CreatedAt().Compare(createdAt) != 0 {
		t.Error("Expected to be created at", createdAt, "not", code.CreatedAt())
	}
}

func TestExpiresAt(t *testing.T) {
	amount := 6
	token := "000000"
	expiresAt := time.Date(0, 0, 0, 0, 0, 0, 0, time.UTC)

	code, err := code.NewCode(token, amount, emojies.CATS, time.Time{}, expiresAt)
	if err != nil {
		t.Error("Expected to not return an error when creating new code:", err)
	}

	if code.ExpiresAt().Compare(expiresAt) != 0 {
		t.Error("Expected to expire at", expiresAt, "not", code.ExpiresAt())
	}
}