214 lines
6.3 KiB
Java
214 lines
6.3 KiB
Java
// Copyright (C) 2013 The Android Open Source Project
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package com.google.gerrit.acceptance;
|
|
|
|
import com.google.gerrit.common.TimeUtil;
|
|
import com.google.gerrit.reviewdb.server.ReviewDb;
|
|
import com.google.gerrit.server.CurrentUser;
|
|
import com.google.gerrit.server.RequestCleanup;
|
|
import com.google.gerrit.server.config.RequestScopedReviewDbProvider;
|
|
import com.google.gerrit.server.util.RequestContext;
|
|
import com.google.gerrit.server.util.ThreadLocalRequestContext;
|
|
import com.google.gerrit.server.util.ThreadLocalRequestScopePropagator;
|
|
import com.google.gerrit.testutil.DisabledReviewDb;
|
|
import com.google.gwtorm.server.SchemaFactory;
|
|
import com.google.inject.Inject;
|
|
import com.google.inject.Key;
|
|
import com.google.inject.OutOfScopeException;
|
|
import com.google.inject.Provider;
|
|
import com.google.inject.Scope;
|
|
import com.google.inject.util.Providers;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
/** Guice scopes for state during an Acceptance Test connection. */
|
|
public class AcceptanceTestRequestScope {
|
|
private static final Key<RequestCleanup> RC_KEY = Key.get(RequestCleanup.class);
|
|
|
|
private static final Key<RequestScopedReviewDbProvider> DB_KEY =
|
|
Key.get(RequestScopedReviewDbProvider.class);
|
|
|
|
public static class Context implements RequestContext {
|
|
private final RequestCleanup cleanup = new RequestCleanup();
|
|
private final Map<Key<?>, Object> map = new HashMap<>();
|
|
private final SchemaFactory<ReviewDb> schemaFactory;
|
|
private final SshSession session;
|
|
private final CurrentUser user;
|
|
|
|
final long created;
|
|
volatile long started;
|
|
volatile long finished;
|
|
|
|
private Context(SchemaFactory<ReviewDb> sf, SshSession s, CurrentUser u, long at) {
|
|
schemaFactory = sf;
|
|
session = s;
|
|
user = u;
|
|
created = started = finished = at;
|
|
map.put(RC_KEY, cleanup);
|
|
map.put(DB_KEY, new RequestScopedReviewDbProvider(schemaFactory, Providers.of(cleanup)));
|
|
}
|
|
|
|
private Context(Context p, SshSession s, CurrentUser c) {
|
|
this(p.schemaFactory, s, c, p.created);
|
|
started = p.started;
|
|
finished = p.finished;
|
|
}
|
|
|
|
SshSession getSession() {
|
|
return session;
|
|
}
|
|
|
|
@Override
|
|
public CurrentUser getUser() {
|
|
if (user == null) {
|
|
throw new IllegalStateException("user == null, forgot to set it?");
|
|
}
|
|
return user;
|
|
}
|
|
|
|
@Override
|
|
public Provider<ReviewDb> getReviewDbProvider() {
|
|
return (RequestScopedReviewDbProvider) map.get(DB_KEY);
|
|
}
|
|
|
|
synchronized <T> T get(Key<T> key, Provider<T> creator) {
|
|
@SuppressWarnings("unchecked")
|
|
T t = (T) map.get(key);
|
|
if (t == null) {
|
|
t = creator.get();
|
|
map.put(key, t);
|
|
}
|
|
return t;
|
|
}
|
|
}
|
|
|
|
static class ContextProvider implements Provider<Context> {
|
|
@Override
|
|
public Context get() {
|
|
return requireContext();
|
|
}
|
|
}
|
|
|
|
static class SshSessionProvider implements Provider<SshSession> {
|
|
@Override
|
|
public SshSession get() {
|
|
return requireContext().getSession();
|
|
}
|
|
}
|
|
|
|
static class Propagator extends ThreadLocalRequestScopePropagator<Context> {
|
|
private final AcceptanceTestRequestScope atrScope;
|
|
|
|
@Inject
|
|
Propagator(
|
|
AcceptanceTestRequestScope atrScope,
|
|
ThreadLocalRequestContext local,
|
|
Provider<RequestScopedReviewDbProvider> dbProviderProvider) {
|
|
super(REQUEST, current, local, dbProviderProvider);
|
|
this.atrScope = atrScope;
|
|
}
|
|
|
|
@Override
|
|
protected Context continuingContext(Context ctx) {
|
|
// The cleanup is not chained, since the RequestScopePropagator executors
|
|
// the Context's cleanup when finished executing.
|
|
return atrScope.newContinuingContext(ctx);
|
|
}
|
|
}
|
|
|
|
private static final ThreadLocal<Context> current = new ThreadLocal<>();
|
|
|
|
private static Context requireContext() {
|
|
final Context ctx = current.get();
|
|
if (ctx == null) {
|
|
throw new OutOfScopeException("Not in command/request");
|
|
}
|
|
return ctx;
|
|
}
|
|
|
|
private final ThreadLocalRequestContext local;
|
|
|
|
@Inject
|
|
AcceptanceTestRequestScope(ThreadLocalRequestContext local) {
|
|
this.local = local;
|
|
}
|
|
|
|
public Context newContext(SchemaFactory<ReviewDb> sf, SshSession s, CurrentUser user) {
|
|
return new Context(sf, s, user, TimeUtil.nowMs());
|
|
}
|
|
|
|
private Context newContinuingContext(Context ctx) {
|
|
return new Context(ctx, ctx.getSession(), ctx.getUser());
|
|
}
|
|
|
|
public Context set(Context ctx) {
|
|
Context old = current.get();
|
|
current.set(ctx);
|
|
local.setContext(ctx);
|
|
return old;
|
|
}
|
|
|
|
public Context get() {
|
|
return current.get();
|
|
}
|
|
|
|
public Context disableDb() {
|
|
Context old = current.get();
|
|
SchemaFactory<ReviewDb> sf =
|
|
new SchemaFactory<ReviewDb>() {
|
|
@Override
|
|
public ReviewDb open() {
|
|
return new DisabledReviewDb();
|
|
}
|
|
};
|
|
Context ctx = new Context(sf, old.session, old.user, old.created);
|
|
|
|
current.set(ctx);
|
|
local.setContext(ctx);
|
|
return old;
|
|
}
|
|
|
|
public Context reopenDb() {
|
|
// Setting a new context with the same fields is enough to get the ReviewDb
|
|
// provider to reopen the database.
|
|
Context old = current.get();
|
|
return set(new Context(old.schemaFactory, old.session, old.user, old.created));
|
|
}
|
|
|
|
/** Returns exactly one instance per command executed. */
|
|
static final Scope REQUEST =
|
|
new Scope() {
|
|
@Override
|
|
public <T> Provider<T> scope(Key<T> key, Provider<T> creator) {
|
|
return new Provider<T>() {
|
|
@Override
|
|
public T get() {
|
|
return requireContext().get(key, creator);
|
|
}
|
|
|
|
@Override
|
|
public String toString() {
|
|
return String.format("%s[%s]", creator, REQUEST);
|
|
}
|
|
};
|
|
}
|
|
|
|
@Override
|
|
public String toString() {
|
|
return "Acceptance Test Scope.REQUEST";
|
|
}
|
|
};
|
|
}
|