agent: use testing intention to get valid intentions

This commit is contained in:
Mitchell Hashimoto 2018-03-03 10:12:05 -08:00
parent ab4ea3efb4
commit 70858598e4
No known key found for this signature in database
GPG Key ID: 744E147AA52F5B0A
4 changed files with 26 additions and 20 deletions

View File

@ -99,10 +99,13 @@ func (s *Intention) Apply(
args.Intention.DestinationNS = structs.IntentionDefaultNamespace
}
// Validate
// Validate. We do not validate on delete since it is valid to only
// send an ID in that case.
if args.Op != structs.IntentionOpDelete {
if err := args.Intention.Validate(); err != nil {
return err
}
}
// Commit
resp, err := s.srv.raftApply(structs.IntentionRequestType, args)

View File

@ -173,7 +173,13 @@ func (s *HTTPServer) IntentionSpecificGet(id string, resp http.ResponseWriter, r
return nil, err
}
// TODO: validate length
// This shouldn't happen since the called API documents it shouldn't,
// but we check since the alternative if it happens is a panic.
if len(reply.Intentions) == 0 {
resp.WriteHeader(http.StatusNotFound)
return nil, nil
}
return reply.Intentions[0], nil
}

View File

@ -43,8 +43,10 @@ func TestIntentionsList_values(t *testing.T) {
req := structs.IntentionRequest{
Datacenter: "dc1",
Op: structs.IntentionOpCreate,
Intention: &structs.Intention{SourceName: v},
Intention: structs.TestIntention(t),
}
req.Intention.SourceName = v
var reply string
if err := a.RPC("Intention.Apply", &req, &reply); err != nil {
t.Fatalf("err: %s", err)
@ -93,13 +95,10 @@ func TestIntentionsMatch_basic(t *testing.T) {
ixn := structs.IntentionRequest{
Datacenter: "dc1",
Op: structs.IntentionOpCreate,
Intention: &structs.Intention{
SourceNS: "default",
SourceName: "test",
DestinationNS: v[0],
DestinationName: v[1],
},
Intention: structs.TestIntention(t),
}
ixn.Intention.DestinationNS = v[0]
ixn.Intention.DestinationName = v[1]
// Create
var reply string
@ -198,7 +197,8 @@ func TestIntentionsCreate_good(t *testing.T) {
defer a.Shutdown()
// Make sure an empty list is non-nil.
args := &structs.Intention{SourceName: "foo"}
args := structs.TestIntention(t)
args.SourceName = "foo"
req, _ := http.NewRequest("POST", "/v1/connect/intentions", jsonReader(args))
resp := httptest.NewRecorder()
obj, err := a.srv.IntentionCreate(resp, req)
@ -238,12 +238,7 @@ func TestIntentionsSpecificGet_good(t *testing.T) {
defer a.Shutdown()
// The intention
ixn := &structs.Intention{
SourceNS: structs.IntentionDefaultNamespace,
SourceName: "foo",
DestinationNS: structs.IntentionDefaultNamespace,
DestinationName: "bar",
}
ixn := structs.TestIntention(t)
// Create an intention directly
var reply string
@ -286,7 +281,7 @@ func TestIntentionsSpecificUpdate_good(t *testing.T) {
defer a.Shutdown()
// The intention
ixn := &structs.Intention{SourceName: "foo"}
ixn := structs.TestIntention(t)
// Create an intention directly
var reply string
@ -341,7 +336,8 @@ func TestIntentionsSpecificDelete_good(t *testing.T) {
defer a.Shutdown()
// The intention
ixn := &structs.Intention{SourceName: "foo"}
ixn := structs.TestIntention(t)
ixn.SourceName = "foo"
// Create an intention directly
var reply string

View File

@ -13,5 +13,6 @@ func TestIntention(t testing.T) *Intention {
DestinationName: "db",
Action: IntentionActionAllow,
SourceType: IntentionSourceConsul,
Meta: map[string]string{},
}
}