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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
// Copyright 2013 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.
package uniter_test
import (
jc "github.com/juju/testing/checkers"
gc "gopkg.in/check.v1"
"gopkg.in/juju/names.v2"
apitesting "github.com/juju/juju/api/testing"
"github.com/juju/juju/apiserver/params"
"github.com/juju/juju/network"
)
type stateSuite struct {
uniterSuite
*apitesting.APIAddresserTests
*apitesting.ModelWatcherTests
}
var _ = gc.Suite(&stateSuite{})
func (s *stateSuite) SetUpTest(c *gc.C) {
s.uniterSuite.SetUpTest(c)
s.APIAddresserTests = apitesting.NewAPIAddresserTests(s.uniter, s.BackingState)
s.ModelWatcherTests = apitesting.NewModelWatcherTests(s.uniter, s.BackingState, apitesting.NoSecrets)
}
func (s *stateSuite) TestProviderType(c *gc.C) {
cfg, err := s.State.ModelConfig()
c.Assert(err, jc.ErrorIsNil)
providerType, err := s.uniter.ProviderType()
c.Assert(err, jc.ErrorIsNil)
c.Assert(providerType, gc.DeepEquals, cfg.Type())
}
func (s *stateSuite) TestAllMachinePorts(c *gc.C) {
// Verify no ports are opened yet on the machine or unit.
machinePorts, err := s.wordpressMachine.AllPorts()
c.Assert(err, jc.ErrorIsNil)
c.Assert(machinePorts, gc.HasLen, 0)
unitPorts, err := s.wordpressUnit.OpenedPorts()
c.Assert(err, jc.ErrorIsNil)
c.Assert(unitPorts, gc.HasLen, 0)
// Add another wordpress unit on the same machine.
wordpressUnit1, err := s.wordpressService.AddUnit()
c.Assert(err, jc.ErrorIsNil)
err = wordpressUnit1.AssignToMachine(s.wordpressMachine)
c.Assert(err, jc.ErrorIsNil)
// Open some ports on both units.
err = s.wordpressUnit.OpenPorts("tcp", 100, 200)
c.Assert(err, jc.ErrorIsNil)
err = s.wordpressUnit.OpenPorts("udp", 10, 20)
c.Assert(err, jc.ErrorIsNil)
err = wordpressUnit1.OpenPorts("tcp", 201, 250)
c.Assert(err, jc.ErrorIsNil)
err = wordpressUnit1.OpenPorts("udp", 1, 8)
c.Assert(err, jc.ErrorIsNil)
portsMap, err := s.uniter.AllMachinePorts(s.wordpressMachine.Tag().(names.MachineTag))
c.Assert(err, jc.ErrorIsNil)
c.Assert(portsMap, jc.DeepEquals, map[network.PortRange]params.RelationUnit{
network.PortRange{100, 200, "tcp"}: {Unit: s.wordpressUnit.Tag().String()},
network.PortRange{10, 20, "udp"}: {Unit: s.wordpressUnit.Tag().String()},
network.PortRange{201, 250, "tcp"}: {Unit: wordpressUnit1.Tag().String()},
network.PortRange{1, 8, "udp"}: {Unit: wordpressUnit1.Tag().String()},
})
}
|