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
|
package svc // import "github.com/getwtxt/getwtxt/svc"
import (
"bytes"
"fmt"
"log"
"net"
"os"
"strings"
"sync"
"testing"
"github.com/getwtxt/registry"
"github.com/spf13/viper"
)
var (
testport string
initTestOnce sync.Once
initDBOnce sync.Once
)
func initTestConf() {
initTestOnce.Do(func() {
logToNull()
testConfig()
tmpls = initTemplates()
pingAssets()
confObj.Mu.RLock()
defer confObj.Mu.RUnlock()
testport = fmt.Sprintf(":%v", confObj.Port)
})
}
func initTestDB() {
initDBOnce.Do(func() {
initDatabase()
})
}
func logToNull() {
hush, err := os.Open("/dev/null")
if err != nil {
log.Printf("%v\n", err)
}
log.SetOutput(hush)
reqLog = log.New(hush, "", log.LstdFlags)
}
func testConfig() {
viper.SetConfigName("getwtxt")
viper.SetConfigType("yml")
viper.AddConfigPath("../")
viper.SetDefault("BehindProxy", true)
viper.SetDefault("UseTLS", false)
viper.SetDefault("TLSCert", "/etc/ssl/getwtxt.pem")
viper.SetDefault("TLSKey", "/etc/ssl/private/getwtxt.pem")
viper.SetDefault("ListenPort", 9001)
viper.SetDefault("DatabasePath", "getwtxt.db")
viper.SetDefault("AssetsDirectory", "assets")
viper.SetDefault("DatabaseType", "leveldb")
viper.SetDefault("ReCacheInterval", "9m")
viper.SetDefault("DatabasePushInterval", "4m")
viper.SetDefault("Instance.SiteName", "getwtxt")
viper.SetDefault("Instance.OwnerName", "Anonymous Microblogger")
viper.SetDefault("Instance.Email", "nobody@knows")
viper.SetDefault("Instance.URL", "https://twtxt.example.com")
viper.SetDefault("Instance.Description", "A fast, resilient twtxt registry server written in Go!")
confObj.Mu.Lock()
defer confObj.Mu.Unlock()
confObj.Port = viper.GetInt("ListenPort")
confObj.AssetsDir = "../" + viper.GetString("AssetsDirectory")
confObj.DBType = strings.ToLower(viper.GetString("DatabaseType"))
confObj.DBPath = viper.GetString("DatabasePath")
confObj.CacheInterval = viper.GetDuration("StatusFetchInterval")
confObj.DBInterval = viper.GetDuration("DatabasePushInterval")
confObj.Instance.Vers = Vers
confObj.Instance.Name = viper.GetString("Instance.SiteName")
confObj.Instance.URL = viper.GetString("Instance.URL")
confObj.Instance.Owner = viper.GetString("Instance.OwnerName")
confObj.Instance.Mail = viper.GetString("Instance.Email")
confObj.Instance.Desc = viper.GetString("Instance.Description")
}
// Creates a fresh mock registry, with a single
// user and their statuses, for testing.
func mockRegistry() {
twtxtCache = registry.NewIndex()
statuses, _, _ := registry.GetTwtxt("https://gbmor.dev/twtxt.txt")
parsed, _ := registry.ParseUserTwtxt(statuses, "gbmor", "https://gbmor.dev/twtxt.txt")
_ = twtxtCache.AddUser("gbmor", "https://gbmor.dev/twtxt.txt", "1", net.ParseIP("127.0.0.1"), parsed)
}
// Empties the mock registry's user of statuses
// for functions that test status modifications
func killStatuses() {
twtxtCache.Mu.Lock()
user := twtxtCache.Users["https://gbmor.dev/twtxt.txt"]
user.Mu.Lock()
user.Status = registry.NewTimeMap()
user.RLen = "0"
twtxtCache.Users["https://gbmor.dev/twtxt.txt"] = user
user.Mu.Unlock()
twtxtCache.Mu.Unlock()
}
func Test_errLog(t *testing.T) {
t.Run("Log to Buffer", func(t *testing.T) {
b := []byte{}
buf := bytes.NewBuffer(b)
log.SetOutput(buf)
err := fmt.Errorf("test error")
errLog("", err)
if !strings.Contains(buf.String(), "test error") {
t.Errorf("Output Incorrect: %#v\n", buf.String())
}
})
}
|