/**
 * Copyright 2010 Google Inc.
 *
 * 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 org.waveprotocol.box.server.persistence.mongodb;

import com.mongodb.DB;

import org.waveprotocol.box.server.persistence.AttachmentStore;
import org.waveprotocol.box.server.persistence.AttachmentStoreTestBase;

/**
 * A wrapper around AttachmentStoreBase which tests the mongo-based attachment
 * store.
 */
public class AttachmentStoreTest extends AttachmentStoreTestBase {
  private static final String TEST_DATABASE = "AttachmentTest";

  private final MongoDbStore store;
  private final DB database;

  /**
   * Initializes the MongoDB version of a {@link AttachmentStoreTestBase}.
   */
  public AttachmentStoreTest() throws Exception {
    MongoDbProvider mongoDbProvider = new MongoDbProvider();
    this.database = mongoDbProvider.getDatabaseForName(TEST_DATABASE);
    store = new MongoDbStore(database);
  }

  @Override
  protected void tearDown() throws Exception {
    super.tearDown();
    database.dropDatabase();
  }
  
  @Override
  protected AttachmentStore newAttachmentStore() {
    database.dropDatabase();
    return store;
  }
}