53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
package control
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
"github.com/yukimochi/Activity-Relay/models"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
var err error
|
|
|
|
testConfigPath := "../misc/test/config.yml"
|
|
file, _ := os.Open(testConfigPath)
|
|
defer file.Close()
|
|
|
|
viper.SetConfigType("yaml")
|
|
viper.ReadConfig(file)
|
|
viper.Set("ACTOR_PEM", "../misc/test/testKey.pem")
|
|
viper.BindEnv("REDIS_URL")
|
|
|
|
globalConfig, err = models.NewRelayConfig()
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
err = initialize(globalConfig)
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
relayState = models.NewState(globalConfig.RedisClient(), false)
|
|
relayState.RedisClient.FlushAll().Result()
|
|
|
|
initProxy = emptyProxy
|
|
initProxyE = emptyProxyE
|
|
|
|
code := m.Run()
|
|
os.Exit(code)
|
|
}
|
|
|
|
func emptyProxy(function func(cmd *cobra.Command, args []string), cmd *cobra.Command, args []string) {
|
|
function(cmd, args)
|
|
}
|
|
|
|
func emptyProxyE(function func(cmd *cobra.Command, args []string) error, cmd *cobra.Command, args []string) error {
|
|
return function(cmd, args)
|
|
}
|