-
Notifications
You must be signed in to change notification settings - Fork 0
/
mock.go
51 lines (47 loc) · 1.34 KB
/
mock.go
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
// Package mock provides mock objects and setup for Deis tests.
package mock
import (
"testing"
"time"
"github.com/deis/deis/tests/dockercli"
"github.com/deis/deis/tests/etcdutils"
"github.com/deis/deis/tests/utils"
)
// RunMockDatabase starts a mock postgresql database for testing.
func RunMockDatabase(t *testing.T, uid string, etcdPort string, dbPort string) {
var err error
cli, stdout, stdoutPipe := dockercli.GetNewClient()
done := make(chan bool, 1)
dbImage := "paintedfox/postgresql:latest"
ipaddr := utils.GetHostIPAddress()
done <- true
go func() {
<-done
err = dockercli.RunContainer(cli,
"--name", "deis-test-database-"+uid,
"--rm",
"-p", dbPort+":5432",
"-e", "PUBLISH="+dbPort,
"-e", "HOST="+ipaddr,
"-e", "USER=deis",
"-e", "DB=deis",
"-e", "PASS=deis",
dbImage)
}()
time.Sleep(1000 * time.Millisecond)
dockercli.PrintToStdout(t, stdout, stdoutPipe, "Starting")
setkeys := []string{
"/deis/database/user",
"/deis/database/password",
"/deis/database/name",
}
setdir := []string{}
dbhandler := etcdutils.InitetcdValues(setdir, setkeys, etcdPort)
etcdutils.Publishvalues(t, dbhandler)
etcdutils.SetEtcdValues(t,
[]string{"/deis/database/host", "/deis/database/port", "/deis/database/engine"},
[]string{ipaddr, dbPort, "postgresql_psycopg2"}, dbhandler.C)
if err != nil {
t.Fatal(err)
}
}