~juju-qa/ubuntu/xenial/juju/2.0-rc2

« back to all changes in this revision

Viewing changes to src/github.com/juju/txn/prune_test.go

  • Committer: Nicholas Skaggs
  • Date: 2016-09-30 14:39:30 UTC
  • mfrom: (1.8.1)
  • Revision ID: nicholas.skaggs@canonical.com-20160930143930-vwwhrefh6ftckccy
import upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
7
7
        "time"
8
8
 
9
9
        "github.com/juju/loggo"
10
 
        jujutesting "github.com/juju/testing"
 
10
        "github.com/juju/testing"
11
11
        jc "github.com/juju/testing/checkers"
12
12
        gc "gopkg.in/check.v1"
13
13
        "gopkg.in/mgo.v2"
18
18
)
19
19
 
20
20
type PruneSuite struct {
21
 
        jujutesting.MgoSuite
 
21
        testing.IsolationSuite
 
22
        testing.MgoSuite
22
23
        db     *mgo.Database
23
24
        txns   *mgo.Collection
24
25
        runner *txn.Runner
26
27
 
27
28
var _ = gc.Suite(&PruneSuite{})
28
29
 
 
30
func (s *PruneSuite) SetUpSuite(c *gc.C) {
 
31
        s.IsolationSuite.SetUpSuite(c)
 
32
        s.MgoSuite.SetUpSuite(c)
 
33
}
 
34
 
 
35
func (s *PruneSuite) TearDownSuite(c *gc.C) {
 
36
        txn.SetChaos(txn.Chaos{})
 
37
        s.MgoSuite.TearDownSuite(c)
 
38
        s.IsolationSuite.TearDownSuite(c)
 
39
}
 
40
 
29
41
func (s *PruneSuite) SetUpTest(c *gc.C) {
 
42
        s.IsolationSuite.SetUpTest(c)
30
43
        s.MgoSuite.SetUpTest(c)
31
44
        txn.SetChaos(txn.Chaos{})
32
45
 
35
48
        s.runner = txn.NewRunner(s.txns)
36
49
}
37
50
 
38
 
func (s *PruneSuite) TearDownSuite(c *gc.C) {
39
 
        txn.SetChaos(txn.Chaos{})
40
 
        s.MgoSuite.TearDownSuite(c)
 
51
func (s *PruneSuite) TearDownTest(c *gc.C) {
 
52
        s.MgoSuite.TearDownTest(c)
 
53
        s.IsolationSuite.TearDownTest(c)
41
54
}
42
55
 
43
56
func (s *PruneSuite) maybePrune(c *gc.C, pruneFactor float32) {
385
398
        c.Assert(err, jc.ErrorIsNil)
386
399
 
387
400
        var tw loggo.TestWriter
388
 
        c.Assert(loggo.RegisterWriter("test", &tw, loggo.WARNING), gc.IsNil)
 
401
        c.Assert(loggo.RegisterWriter("test", loggo.NewMinimumLevelWriter(&tw, loggo.WARNING)), gc.IsNil)
389
402
        defer loggo.RemoveWriter("test")
390
403
 
391
404
        // Pruning should occur when "last" pointer is broken.