LogDog: Support per-namespace expired archival.

appengine/gaeauth/server/internal/authdb/authdb_test.go

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package authdb
 
 import (
         "fmt"
         "testing"
 
         "golang.org/x/net/context"
 
         "github.com/luci/gae/service/datastore"
         "github.com/luci/luci-go/appengine/gaetesting"
         "github.com/luci/luci-go/common/clock"
         "github.com/luci/luci-go/server/auth/service"
         "github.com/luci/luci-go/server/auth/service/protocol"
 
         . "github.com/smartystreets/goconvey/convey"
 )
 
 func TestConfigureAuthService(t *testing.T) {
         Convey("Initial config", t, func() {
                 srv := &fakeAuthService{LatestRev: 123}
                 c := setAuthService(gaetesting.TestingContext(), srv)
 
                 So(ConfigureAuthService(c, "http://base_url", "http://auth-service"), ShouldBeNil)
                 So(srv.Calls, ShouldResemble, []string{
-                        `EnsureSubscription "projects/dev~app/subscriptions/dev-app-server-v1+auth-service" ""`,
+                        `EnsureSubscription "projects/app/subscriptions/dev-app-server-v1+auth-service" ""`,
                 })
 
                 info, err := GetLatestSnapshotInfo(c)
                 So(err, ShouldBeNil)
                 So(info, ShouldResemble, &SnapshotInfo{
                         AuthServiceURL: "http://auth-service",
                         Rev:            123,
                 })
 
                 // Coverage for GetAuthDBSnapshot.
                 _, err = GetAuthDBSnapshot(c, "missing")
                 So(err, ShouldEqual, datastore.ErrNoSuchEntity)
                 snap, err := GetAuthDBSnapshot(c, info.GetSnapshotID())
                 So(err, ShouldBeNil)
                 So(snap, ShouldResemble, &protocol.AuthDB{
                         OauthClientId:     strPtr("client-id-for-rev-123"),
                         OauthClientSecret: strPtr("secret"),
                 })
 
                 // Same config call again triggers resubsciption.
                 srv.Calls = nil
                 So(ConfigureAuthService(c, "http://base_url", "http://auth-service"), ShouldBeNil)
                 So(srv.Calls, ShouldResemble, []string{
-                        `EnsureSubscription "projects/dev~app/subscriptions/dev-app-server-v1+auth-service" ""`,
+                        `EnsureSubscription "projects/app/subscriptions/dev-app-server-v1+auth-service" ""`,
                 })
         })
 
         Convey("Switching cfg", t, func() {
                 srv := &fakeAuthService{LatestRev: 123}
                 c := setAuthService(gaetesting.TestingContext(), srv)
 
                 // Initial config.
                 So(ConfigureAuthService(c, "http://base_url", "http://auth-service-1"), ShouldBeNil)
                 // Change URL of the service.
                 So(ConfigureAuthService(c, "http://base_url", "http://auth-service-2"), ShouldBeNil)
 
                 info, err := GetLatestSnapshotInfo(c)
                 So(err, ShouldBeNil)
                 So(info, ShouldResemble, &SnapshotInfo{
                         AuthServiceURL: "http://auth-service-2",
                         Rev:            123,
                 })
 
                 So(srv.Calls, ShouldResemble, []string{
-                        `EnsureSubscription "projects/dev~app/subscriptions/dev-app-server-v1+auth-service-1" ""`,
-                        `EnsureSubscription "projects/dev~app/subscriptions/dev-app-server-v1+auth-service-2" ""`,
-                        `DeleteSubscription "projects/dev~app/subscriptions/dev-app-server-v1+auth-service-1"`,
+                        `EnsureSubscription "projects/app/subscriptions/dev-app-server-v1+auth-service-1" ""`,
+                        `EnsureSubscription "projects/app/subscriptions/dev-app-server-v1+auth-service-2" ""`,
+                        `DeleteSubscription "projects/app/subscriptions/dev-app-server-v1+auth-service-1"`,
                 })
         })
 
         Convey("Removing cfg", t, func() {
                 srv := &fakeAuthService{LatestRev: 123}
                 c := setAuthService(gaetesting.TestingContext(), srv)
 
                 // Initial config.
                 So(ConfigureAuthService(c, "http://base_url", "http://auth-service-1"), ShouldBeNil)
                 // Remove.
                 So(ConfigureAuthService(c, "http://base_url", ""), ShouldBeNil)
 
                 info, err := GetLatestSnapshotInfo(c)
                 So(err, ShouldBeNil)
                 So(info, ShouldBeNil)
 
                 So(srv.Calls, ShouldResemble, []string{
-                        `EnsureSubscription "projects/dev~app/subscriptions/dev-app-server-v1+auth-service-1" ""`,
-                        `DeleteSubscription "projects/dev~app/subscriptions/dev-app-server-v1+auth-service-1"`,
+                        `EnsureSubscription "projects/app/subscriptions/dev-app-server-v1+auth-service-1" ""`,
+                        `DeleteSubscription "projects/app/subscriptions/dev-app-server-v1+auth-service-1"`,
                 })
         })
 }
 
 func TestSyncAuthDB(t *testing.T) {
         Convey("No new changes", t, func() {
                 srv := &fakeAuthService{LatestRev: 123}
                 c := setAuthService(gaetesting.TestingContext(), srv)
                 So(ConfigureAuthService(c, "http://base_url", "http://auth-service"), ShouldBeNil)
 
@@ skipping 63 matching lines @@
         }
         return &service.Snapshot{
                 AuthDB: &protocol.AuthDB{
                         OauthClientId:     strPtr(fmt.Sprintf("client-id-for-rev-%d", f.LatestRev)),
                         OauthClientSecret: strPtr("secret"),
                 },
                 Rev:     f.LatestRev,
                 Created: clock.Now(c),
         }, nil
 }