← Back to branch summary

~nskaggs/+junk/xenial-test

« back to all changes in this revision

Viewing changes to src/github.com/juju/utils/gomaxprocs_test.go

  • Committer: Nicholas Skaggs
  • Date: 2016-10-24 20:56:05 UTC
  • Revision ID: nicholas.skaggs@canonical.com-20161024205605-z8lta0uvuhtxwzwl
Initi with beta15

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/github.com/juju/utils/gomaxprocs_test.go
 
1
// Copyright 2014 Canonical Ltd.
 
2
// Licensed under the LGPLv3, see LICENCE file for details.
 
3
 
 
4
package utils_test
 
5
 
 
6
import (
 
7
        "os"
 
8
 
 
9
        "github.com/juju/testing"
 
10
        gc "gopkg.in/check.v1"
 
11
 
 
12
        "github.com/juju/utils"
 
13
)
 
14
 
 
15
type gomaxprocsSuite struct {
 
16
        testing.IsolationSuite
 
17
        setmaxprocs    chan int
 
18
        numCPUResponse int
 
19
        setMaxProcs    int
 
20
}
 
21
 
 
22
var _ = gc.Suite(&gomaxprocsSuite{})
 
23
 
 
24
func (s *gomaxprocsSuite) SetUpTest(c *gc.C) {
 
25
        s.IsolationSuite.SetUpTest(c)
 
26
        // always stub out GOMAXPROCS so we don't actually change anything
 
27
        s.numCPUResponse = 2
 
28
        s.setMaxProcs = -1
 
29
        maxProcsFunc := func(n int) int {
 
30
                s.setMaxProcs = n
 
31
                return 1
 
32
        }
 
33
        numCPUFunc := func() int { return s.numCPUResponse }
 
34
        s.PatchValue(utils.GOMAXPROCS, maxProcsFunc)
 
35
        s.PatchValue(utils.NumCPU, numCPUFunc)
 
36
        s.PatchEnvironment("GOMAXPROCS", "")
 
37
}
 
38
 
 
39
func (s *gomaxprocsSuite) TestUseMultipleCPUsDoesNothingWhenGOMAXPROCSSet(c *gc.C) {
 
40
        os.Setenv("GOMAXPROCS", "1")
 
41
        utils.UseMultipleCPUs()
 
42
        c.Check(s.setMaxProcs, gc.Equals, 0)
 
43
}
 
44
 
 
45
func (s *gomaxprocsSuite) TestUseMultipleCPUsWhenEnabled(c *gc.C) {
 
46
        utils.UseMultipleCPUs()
 
47
        c.Check(s.setMaxProcs, gc.Equals, 2)
 
48
        s.numCPUResponse = 4
 
49
        utils.UseMultipleCPUs()
 
50
        c.Check(s.setMaxProcs, gc.Equals, 4)
 
51
}