state/apiserver: Relation ops for Uniter API

state/apiserver/uniter/uniter_test.go

 // Copyright 2013 Canonical Ltd.
 // Licensed under the AGPLv3, see LICENCE file for details.
 
 package uniter_test
 
 import (
         stdtesting "testing"
 
         gc "launchpad.net/gocheck"
 
@@ skipping 770 matching lines @@
 func (s *uniterSuite) TestCurrentEnvironUUID(c *gc.C) {
         env, err := s.State.Environment()
         c.Assert(err, gc.IsNil)
 
         result, err := s.uniter.CurrentEnvironUUID()
         c.Assert(err, gc.IsNil)
         c.Assert(result, gc.DeepEquals, params.StringResult{Result: env.UUID()})
 }
 
 func (s *uniterSuite) TestRelation(c *gc.C) {
-        s.addRelatedService(c, "wordpress", "logging", s.wordpressUnit)
-        rel, err := s.State.Relation(0)
-        c.Assert(err, gc.IsNil)
-        relEps := rel.Endpoints()
-        eps := make([]params.Endpoint, len(relEps))
-        for i, ep := range relEps {
-                eps[i] = params.Endpoint{
-                        ServiceName: ep.ServiceName,
-                        Relation:    ep.Relation,
-                }
-        }
-
-        args := params.Entities{Entities: []params.Entity{
-                {Tag: "relation-42"},
-                {Tag: "relation-0"},
-                {Tag: "relation-blah"},
-                {Tag: "service-foo"},
-                {Tag: "foo"},
-                {Tag: "unit-wordpress-0"},
+        eps, err := s.State.InferEndpoints([]string{"wordpress", "mysql"})
+        c.Assert(err, gc.IsNil)
+        rel, err := s.State.AddRelation(eps...)
+        c.Assert(err, gc.IsNil)
+        c.Assert(rel.Id(), gc.Equals, 0)
+        wpEp, err := rel.Endpoint("wordpress")
+        c.Assert(err, gc.IsNil)
+
+        args := params.Relations{Relations: []params.Relation{
+                {Relation: "relation-42", Unit: "unit-foo-0"},
+                {Relation: "relation-0", Unit: "unit-wordpress-0"},
+                {Relation: "relation-0", Unit: "unit-mysql-0"},
+                {Relation: "relation-0", Unit: "unit-foo-0"},
+                {Relation: "relation-blah", Unit: "unit-wordpress-0"},
+                {Relation: "service-foo", Unit: "user-admin"},
+                {Relation: "foo", Unit: "bar"},
+                {Relation: "unit-wordpress-0", Unit: "relation-0"},
         }}
         result, err := s.uniter.Relation(args)
         c.Assert(err, gc.IsNil)
+        c.Assert(result.Results, gc.HasLen, len(args.Relations))
         c.Assert(result, gc.DeepEquals, params.RelationResults{
                 Results: []params.RelationResult{
                         {Error: apiservertesting.ErrUnauthorized},
-                        {Id: rel.Id(), Key: rel.String(), Endpoints: eps},
-                        {Error: apiservertesting.ErrUnauthorized},
-                        {Error: apiservertesting.ErrUnauthorized},
-                        {Error: apiservertesting.ErrUnauthorized},
-                        {Error: apiservertesting.ErrUnauthorized},
-                },
-        })
-}
+                        {
+                                Id:  rel.Id(),
+                                Key: rel.String(),
+                                Endpoint: params.Endpoint{
+                                        ServiceName: wpEp.ServiceName,
+                                        Relation:    wpEp.Relation,
+                                },
+                        },
+                        {Error: apiservertesting.ErrUnauthorized},
+                        {Error: apiservertesting.ErrUnauthorized},
+                        {Error: apiservertesting.ErrUnauthorized},
+                        {Error: apiservertesting.ErrUnauthorized},
+                        {Error: apiservertesting.ErrUnauthorized},
+                        {Error: apiservertesting.ErrUnauthorized},
+                },
+        })
+}