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
73
74
|
/*
Copyright 2013-2014 Canonical Ltd.
This program is free software: you can redistribute it and/or modify it
under the terms of the GNU General Public License version 3, as published
by the Free Software Foundation.
This program is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranties of
MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR
PURPOSE. See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License along
with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package server
import (
"bytes"
"io/ioutil"
"os"
"path/filepath"
"time"
. "launchpad.net/gocheck"
"launchpad.net/ubuntu-push/config"
helpers "launchpad.net/ubuntu-push/testing"
)
type configSuite struct{}
var _ = Suite(&configSuite{})
func (s *configSuite) TestDevicesParsedConfig(c *C) {
buf := bytes.NewBufferString(`{
"ping_interval": "5m",
"exchange_timeout": "10s",
"session_queue_size": 10,
"broker_queue_size": 100,
"addr": "127.0.0.1:9999",
"key_pem_file": "key.key",
"cert_pem_file": "cert.cert"
}`)
cfg := &DevicesParsedConfig{}
err := config.ReadConfig(buf, cfg)
c.Assert(err, IsNil)
c.Check(cfg.PingInterval(), Equals, 5*time.Minute)
c.Check(cfg.ExchangeTimeout(), Equals, 10*time.Second)
c.Check(cfg.BrokerQueueSize(), Equals, uint(100))
c.Check(cfg.SessionQueueSize(), Equals, uint(10))
c.Check(cfg.Addr(), Equals, "127.0.0.1:9999")
}
func (s *configSuite) TestTLSParsedConfigLoadPEMs(c *C) {
tmpDir := c.MkDir()
cfg := &TLSParsedConfig{
ParsedKeyPEMFile: "key.key",
ParsedCertPEMFile: "cert.cert",
}
err := cfg.LoadPEMs(tmpDir)
c.Check(err, ErrorMatches, "reading key_pem_file:.*no such file.*")
err = ioutil.WriteFile(filepath.Join(tmpDir, "key.key"), helpers.TestKeyPEMBlock, os.ModePerm)
c.Assert(err, IsNil)
err = cfg.LoadPEMs(tmpDir)
c.Check(err, ErrorMatches, "reading cert_pem_file:.*no such file.*")
err = ioutil.WriteFile(filepath.Join(tmpDir, "cert.cert"), helpers.TestCertPEMBlock, os.ModePerm)
c.Assert(err, IsNil)
err = cfg.LoadPEMs(tmpDir)
c.Assert(err, IsNil)
tlsCfg := cfg.TLSServerConfig()
c.Check(tlsCfg.Certificates, HasLen, 1)
}
|