/
suite_test.go
53 lines (43 loc) · 1.22 KB
/
suite_test.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
52
53
// Copyright 2012 tsuru authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package router
import (
"testing"
"github.com/tsuru/config"
"github.com/tsuru/tsuru/db"
"github.com/tsuru/tsuru/db/dbtest"
"github.com/tsuru/tsuru/servicemanager"
_ "github.com/tsuru/tsuru/storage/mongodb"
check "gopkg.in/check.v1"
)
func Test(t *testing.T) { check.TestingT(t) }
type S struct {
conn *db.Storage
routers map[string]routerFactory
}
var _ = check.Suite(&S{})
func (s *S) SetUpSuite(c *check.C) {
config.Set("log:disable-syslog", true)
config.Set("database:url", "127.0.0.1:27017?maxPoolSize=100")
config.Set("database:name", "router_tests")
var err error
s.conn, err = db.Conn()
c.Assert(err, check.IsNil)
servicemanager.DynamicRouter, err = DynamicRouterService()
c.Assert(err, check.IsNil)
}
func (s *S) SetUpTest(c *check.C) {
s.routers = make(map[string]routerFactory)
for k, v := range routers {
s.routers[k] = v
}
dbtest.ClearAllCollections(s.conn.Apps().Database)
}
func (s *S) TearDownTest(c *check.C) {
routers = s.routers
}
func (s *S) TearDownSuite(c *check.C) {
dbtest.ClearAllCollections(s.conn.Apps().Database)
s.conn.Close()
}