game-repository/main_test.go at master · alarmfox/game-repository · 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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package main

import (
	"database/sql"
	"log"
	"os"
	"testing"

	"github.com/alarmfox/game-repository/model"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
)

func TestCleanup(t *testing.T) {
	if _, ok := os.LookupEnv("SKIP_INTEGRATION"); ok {
		t.Skip()
	}

	postgresAddr := os.Getenv("DB_URI")
	db, err := gorm.Open(postgres.Open(postgresAddr), &gorm.Config{
		SkipDefaultTransaction: true,
	})
	if err != nil {
		log.Fatal(err)
	}
	err = db.AutoMigrate(
		&model.Game{},
		&model.Round{},
		&model.Player{},
		&model.Turn{},
		&model.Metadata{},
		&model.PlayerGame{},
		&model.Robot{})

	if err != nil {
		t.Fatal(err)
	}
	seed(t, db)

	n, err := cleanup(db)

	if err != nil {
		t.Fatal(err)
	}

	if n != 2 {
		t.Fatalf("expected n=2; got n=%d", n)
	}

}

func seed(t *testing.T, db *gorm.DB) {
	t.Helper()

	tmpDir := os.TempDir()
	f1, err := os.CreateTemp(tmpDir, "")
	if err != nil {
		t.Fatal(err)
	}
	defer f1.Close()
	f1.Write([]byte("jfj"))

	f2, err := os.CreateTemp(tmpDir, "")
	if err != nil {
		t.Fatal(err)
	}
	defer f2.Close()

	f2.Write([]byte("jfj"))

	toInsert := []model.Metadata{
		{
			Path:   f1.Name(),
			TurnID: sql.NullInt64{Valid: false},
		},
		{
			Path:   f2.Name(),
			TurnID: sql.NullInt64{Valid: false},
		},
	}

	if err := db.Create(&toInsert).Error; err != nil {
		t.Fatal(err)
	}

	t.Cleanup(func() {
		err := db.
			Session(&gorm.Session{AllowGlobalUpdate: true}).
			Delete(&model.Metadata{}).
			Error

		if err != nil {
			t.Fatal(err)
		}

		os.Remove(f1.Name())
		os.Remove(f2.Name())
	})
}