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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
package main
import (
"context"
"fmt"
"os"
"path/filepath"
"time"
cfgpkg "github.com/mediocregopher/blog.mediocregopher.com/srv/cfg"
"github.com/mediocregopher/blog.mediocregopher.com/srv/post"
"github.com/mediocregopher/mediocre-go-lib/v2/mctx"
"github.com/mediocregopher/mediocre-go-lib/v2/mlog"
"gopkg.in/yaml.v3"
)
type testData struct {
PublishedPosts []post.Post `yaml:"published_posts"`
Assets map[string]string
}
func loadTestData(path string) (testData, error) {
f, err := os.Open(path)
if err != nil {
return testData{}, fmt.Errorf("opening file: %w", err)
}
defer f.Close()
var res testData
if err := yaml.NewDecoder(f).Decode(&res); err != nil {
return testData{}, fmt.Errorf("decoding file contents: %w", err)
}
return res, nil
}
func main() {
ctx := context.Background()
cfg := cfgpkg.NewBlogCfg(cfgpkg.Params{})
var dataDir cfgpkg.DataDir
dataDir.SetupCfg(cfg)
defer dataDir.Close()
ctx = mctx.WithAnnotator(ctx, &dataDir)
testDataPath := cfg.String("test-data-path", "./test-data.yml", "File containing the data to be loaded in")
// initialization
err := cfg.Init(ctx)
logger := mlog.NewLogger(nil)
defer logger.Close()
logger.Info(ctx, "process started")
defer logger.Info(ctx, "process exiting")
if err != nil {
logger.Fatal(ctx, "initializing", err)
}
testDataDir := filepath.Dir(*testDataPath)
testData, err := loadTestData(*testDataPath)
if err != nil {
logger.Fatal(ctx, "loading test data", err)
}
postDB, err := post.NewSQLDB(dataDir)
if err != nil {
logger.Fatal(ctx, "initializing post sql db", err)
}
defer postDB.Close()
{
postStore := post.NewStore(postDB)
now := time.Now().Truncate(time.Hour)
for _, post := range testData.PublishedPosts {
ctx := mctx.Annotate(ctx,
"postID", post.ID,
"now", now)
if _, err := postStore.Set(post, now); err != nil {
logger.Fatal(ctx, "setting post", err)
}
logger.Info(ctx, "set post")
now = now.Add(-1 * time.Hour)
}
}
{
assetStore := post.NewAssetStore(postDB)
setAsset := func(assetID, assetPath string) error {
assetFullPath := filepath.Join(testDataDir, assetPath)
f, err := os.Open(assetFullPath)
if err != nil {
return fmt.Errorf("opening %q for reading: %w", assetFullPath, err)
}
defer f.Close()
return assetStore.Set(assetID, f)
}
for assetID, assetPath := range testData.Assets {
ctx := mctx.Annotate(ctx,
"assetID", assetID,
"assetPath", assetPath)
if err := setAsset(assetID, assetPath); err != nil {
logger.Fatal(ctx, "setting asset", err)
}
logger.Info(ctx, "set asset")
}
}
}
|
