Simplify memory package interface.

go/src/infra/gae/libs/wrapper/memory/taskqueue.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 memory
 
 import (
         "fmt"
         "infra/gae/libs/wrapper"
         "math/rand"
         "net/http"
         "regexp"
         "time"
 
         "golang.org/x/net/context"
 
         "appengine"
         "appengine/taskqueue"
         "appengine_internal"
         dbpb "appengine_internal/datastore"
         pb "appengine_internal/taskqueue"
 )
 
 /////////////////////////////// public functions ///////////////////////////////
 
-// UseTQ adds a wrapper.TaskQueue implementation to context, accessible
-// by wrapper.GetTQ(c)
-func UseTQ(c context.Context) context.Context {
+func useTQ(c context.Context) context.Context {
         return wrapper.SetTQFactory(c, func(ic context.Context) wrapper.TaskQueue {
                 tqd := cur(ic).Get(memContextTQIdx)
                 var ret interface {
                         wrapper.TQTestable
                         wrapper.TaskQueue
                 }
                 switch x := tqd.(type) {
                 case *taskQueueData:
                         ret = &taskqueueImpl{
                                 wrapper.DummyTQ(),
@@ skipping 261 matching lines @@
 func dupQueue(q wrapper.QueueData) wrapper.QueueData {
         r := make(wrapper.QueueData, len(q))
         for k, q := range q {
                 r[k] = make(map[string]*taskqueue.Task, len(q))
                 for tn, t := range q {
                         r[k][tn] = dupTask(t)
                 }
         }
         return r
 }