forked from juju/gomaasapi
/
file_test.go
116 lines (102 loc) · 3.76 KB
/
file_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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
// Copyright 2016 Canonical Ltd.
// Licensed under the LGPLv3, see LICENCE file for details.
package gomaasapi
import (
"net/http"
"github.com/juju/testing"
jc "github.com/juju/testing/checkers"
"github.com/juju/version"
gc "gopkg.in/check.v1"
)
type fileSuite struct {
testing.CleanupSuite
}
var _ = gc.Suite(&fileSuite{})
func (*fileSuite) TestReadFilesBadSchema(c *gc.C) {
_, err := readFiles(twoDotOh, "wat?")
c.Check(err, jc.Satisfies, IsDeserializationError)
c.Assert(err.Error(), gc.Equals, `file base schema check failed: expected list, got string("wat?")`)
}
func (*fileSuite) TestReadFiles(c *gc.C) {
files, err := readFiles(twoDotOh, parseJSON(c, filesResponse))
c.Assert(err, jc.ErrorIsNil)
c.Assert(files, gc.HasLen, 2)
file := files[0]
c.Assert(file.Filename(), gc.Equals, "test")
}
func (*fileSuite) TestLowVersion(c *gc.C) {
_, err := readFiles(version.MustParse("1.9.0"), parseJSON(c, filesResponse))
c.Assert(err, jc.Satisfies, IsUnsupportedVersionError)
}
func (*fileSuite) TestHighVersion(c *gc.C) {
files, err := readFiles(version.MustParse("2.1.9"), parseJSON(c, filesResponse))
c.Assert(err, jc.ErrorIsNil)
c.Assert(files, gc.HasLen, 2)
}
func (s *fileSuite) TestReadAllFromGetFile(c *gc.C) {
// When get file is used, the response includes the body of the file
// base64 encoded, so ReadAll just decodes it.
server, controller := createTestServerController(c, s)
server.AddGetResponse("/api/2.0/files/testing/", http.StatusOK, fileResponse)
file, err := controller.GetFile("testing")
c.Assert(err, jc.ErrorIsNil)
content, err := file.ReadAll()
c.Assert(err, jc.ErrorIsNil)
c.Assert(string(content), gc.Equals, "this is a test\n")
}
func (s *fileSuite) TestReadAllFromFiles(c *gc.C) {
// When get file is used, the response includes the body of the file
// base64 encoded, so ReadAll just decodes it.
server, controller := createTestServerController(c, s)
server.AddGetResponse("/api/2.0/files/", http.StatusOK, filesResponse)
server.AddGetResponse("/api/2.0/files/?filename=test&op=get", http.StatusOK, "some content\n")
files, err := controller.Files("")
c.Assert(err, jc.ErrorIsNil)
file := files[0]
content, err := file.ReadAll()
c.Assert(err, jc.ErrorIsNil)
c.Assert(string(content), gc.Equals, "some content\n")
}
func (s *fileSuite) TestDeleteMissing(c *gc.C) {
// If we get a file, but someone else deletes it first, we get a ...
server, controller := createTestServerController(c, s)
server.AddGetResponse("/api/2.0/files/testing/", http.StatusOK, fileResponse)
file, err := controller.GetFile("testing")
c.Assert(err, jc.ErrorIsNil)
err = file.Delete()
c.Assert(err, jc.Satisfies, IsNoMatchError)
}
func (s *fileSuite) TestDelete(c *gc.C) {
// If we get a file, but someone else deletes it first, we get a ...
server, controller := createTestServerController(c, s)
server.AddGetResponse("/api/2.0/files/testing/", http.StatusOK, fileResponse)
server.AddDeleteResponse("/api/2.0/files/testing/", http.StatusOK, "")
file, err := controller.GetFile("testing")
c.Assert(err, jc.ErrorIsNil)
err = file.Delete()
c.Assert(err, jc.Satisfies, IsNoMatchError)
}
var (
fileResponse = `
{
"resource_uri": "/MAAS/api/2.0/files/testing/",
"content": "dGhpcyBpcyBhIHRlc3QK",
"anon_resource_uri": "/MAAS/api/2.0/files/?op=get_by_key&key=88e64b76-fb82-11e5-932f-52540051bf22",
"filename": "testing"
}
`
filesResponse = `
[
{
"resource_uri": "/MAAS/api/2.0/files/test/",
"anon_resource_uri": "/MAAS/api/2.0/files/?op=get_by_key&key=3afba564-fb7d-11e5-932f-52540051bf22",
"filename": "test"
},
{
"resource_uri": "/MAAS/api/2.0/files/test-file.txt/",
"anon_resource_uri": "/MAAS/api/2.0/files/?op=get_by_key&key=69913e62-fad2-11e5-932f-52540051bf22",
"filename": "test-file.txt"
}
]
`
)