← Back to branch summary

~nskaggs/+junk/xenial-test

« back to all changes in this revision

Viewing changes to src/github.com/juju/romulus/cmd/listagreements/listagreements_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/romulus/cmd/listagreements/listagreements_test.go
 
1
// Copyright 2016 Canonical Ltd.
 
2
// Licensed under the AGPLv3, see LICENCE file for details.
 
3
 
 
4
package listagreements_test
 
5
 
 
6
import (
 
7
        "errors"
 
8
        "time"
 
9
 
 
10
        "github.com/juju/cmd/cmdtesting"
 
11
        coretesting "github.com/juju/juju/testing"
 
12
        "github.com/juju/terms-client/api/wireformat"
 
13
        jujutesting "github.com/juju/testing"
 
14
        jc "github.com/juju/testing/checkers"
 
15
        gc "gopkg.in/check.v1"
 
16
        "gopkg.in/macaroon-bakery.v1/httpbakery"
 
17
 
 
18
        "github.com/juju/romulus/cmd/listagreements"
 
19
)
 
20
 
 
21
var _ = gc.Suite(&listAgreementsSuite{})
 
22
 
 
23
var testTermsAndConditions = "Test Terms and Conditions"
 
24
 
 
25
type listAgreementsSuite struct {
 
26
        coretesting.FakeJujuXDGDataHomeSuite
 
27
        client *mockClient
 
28
}
 
29
 
 
30
func (s *listAgreementsSuite) SetUpTest(c *gc.C) {
 
31
        s.FakeJujuXDGDataHomeSuite.SetUpTest(c)
 
32
        s.client = &mockClient{}
 
33
 
 
34
        jujutesting.PatchValue(listagreements.NewClient, func(_ *httpbakery.Client) (listagreements.TermsServiceClient, error) {
 
35
                return s.client, nil
 
36
        })
 
37
}
 
38
 
 
39
const (
 
40
        expectedListAgreementsJSONOutput = `[
 
41
    {
 
42
        "user": "test-user",
 
43
        "term": "test-term",
 
44
        "revision": 1,
 
45
        "created-on": "2015-12-25T00:00:00Z"
 
46
    }
 
47
]
 
48
`
 
49
        expectedListAgreementsJSONOutputWithOwner = `[
 
50
    {
 
51
        "user": "test-user",
 
52
        "owner": "owner",
 
53
        "term": "test-term",
 
54
        "revision": 1,
 
55
        "created-on": "2015-12-25T00:00:00Z"
 
56
    }
 
57
]
 
58
`
 
59
)
 
60
 
 
61
func (s *listAgreementsSuite) TestGetUsersAgreements(c *gc.C) {
 
62
        ctx, err := cmdtesting.RunCommand(c, listagreements.NewListAgreementsCommand())
 
63
        c.Assert(err, jc.ErrorIsNil)
 
64
        c.Assert(cmdtesting.Stdout(ctx), gc.Equals, `[]
 
65
`)
 
66
        c.Assert(s.client.called, jc.IsTrue)
 
67
 
 
68
        s.client.setError("well, this is embarassing")
 
69
        ctx, err = cmdtesting.RunCommand(c, listagreements.NewListAgreementsCommand())
 
70
        c.Assert(err, gc.ErrorMatches, "failed to list user agreements: well, this is embarassing")
 
71
        c.Assert(s.client.called, jc.IsTrue)
 
72
 
 
73
        agreements := []wireformat.AgreementResponse{{
 
74
                User:      "test-user",
 
75
                Term:      "test-term",
 
76
                Revision:  1,
 
77
                CreatedOn: time.Date(2015, 12, 25, 0, 0, 0, 0, time.UTC),
 
78
        }}
 
79
        s.client.setAgreements(agreements)
 
80
 
 
81
        ctx, err = cmdtesting.RunCommand(c, listagreements.NewListAgreementsCommand())
 
82
        c.Assert(err, jc.ErrorIsNil)
 
83
        c.Assert(ctx, gc.NotNil)
 
84
        c.Assert(cmdtesting.Stdout(ctx), gc.Equals, expectedListAgreementsJSONOutput)
 
85
        c.Assert(s.client.called, jc.IsTrue)
 
86
 
 
87
        ctx, err = cmdtesting.RunCommand(c, listagreements.NewListAgreementsCommand(), "--format", "yaml")
 
88
        c.Assert(err, jc.ErrorIsNil)
 
89
        c.Assert(ctx, gc.NotNil)
 
90
        c.Assert(cmdtesting.Stdout(ctx), gc.Equals, "- user: test-user\n  term: test-term\n  revision: 1\n  createdon: 2015-12-25T00:00:00Z\n")
 
91
        c.Assert(s.client.called, jc.IsTrue)
 
92
}
 
93
 
 
94
func (s *listAgreementsSuite) TestGetUsersAgreementsWithTermOwner(c *gc.C) {
 
95
        ctx, err := cmdtesting.RunCommand(c, listagreements.NewListAgreementsCommand())
 
96
        c.Assert(err, jc.ErrorIsNil)
 
97
        c.Assert(cmdtesting.Stdout(ctx), gc.Equals, `[]
 
98
`)
 
99
        c.Assert(s.client.called, jc.IsTrue)
 
100
 
 
101
        s.client.setError("well, this is embarassing")
 
102
        ctx, err = cmdtesting.RunCommand(c, listagreements.NewListAgreementsCommand())
 
103
        c.Assert(err, gc.ErrorMatches, "failed to list user agreements: well, this is embarassing")
 
104
        c.Assert(s.client.called, jc.IsTrue)
 
105
 
 
106
        agreements := []wireformat.AgreementResponse{{
 
107
                User:      "test-user",
 
108
                Owner:     "owner",
 
109
                Term:      "test-term",
 
110
                Revision:  1,
 
111
                CreatedOn: time.Date(2015, 12, 25, 0, 0, 0, 0, time.UTC),
 
112
        }}
 
113
        s.client.setAgreements(agreements)
 
114
 
 
115
        ctx, err = cmdtesting.RunCommand(c, listagreements.NewListAgreementsCommand())
 
116
        c.Assert(err, jc.ErrorIsNil)
 
117
        c.Assert(ctx, gc.NotNil)
 
118
        c.Assert(cmdtesting.Stdout(ctx), gc.Equals, expectedListAgreementsJSONOutputWithOwner)
 
119
        c.Assert(s.client.called, jc.IsTrue)
 
120
 
 
121
        ctx, err = cmdtesting.RunCommand(c, listagreements.NewListAgreementsCommand(), "--format", "yaml")
 
122
        c.Assert(err, jc.ErrorIsNil)
 
123
        c.Assert(ctx, gc.NotNil)
 
124
        c.Assert(cmdtesting.Stdout(ctx), gc.Equals, "- user: test-user\n  owner: owner\n  term: test-term\n  revision: 1\n  createdon: 2015-12-25T00:00:00Z\n")
 
125
        c.Assert(s.client.called, jc.IsTrue)
 
126
}
 
127
 
 
128
type mockClient struct {
 
129
        called bool
 
130
 
 
131
        agreements []wireformat.AgreementResponse
 
132
        err        string
 
133
}
 
134
 
 
135
func (c *mockClient) setAgreements(agreements []wireformat.AgreementResponse) {
 
136
        c.agreements = agreements
 
137
        c.called = false
 
138
        c.err = ""
 
139
}
 
140
 
 
141
func (c *mockClient) setError(err string) {
 
142
        c.err = err
 
143
        c.called = false
 
144
        c.agreements = nil
 
145
}
 
146
 
 
147
func (c *mockClient) GetUsersAgreements() ([]wireformat.AgreementResponse, error) {
 
148
        c.called = true
 
149
        if c.err != "" {
 
150
                return nil, errors.New(c.err)
 
151
        }
 
152
        return c.agreements, nil
 
153
}