fixtures.BsonFixtures.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of reactivemongo-extensions-bson_2.11 Show documentation

reactivemongo-extensions-bson

The newest version!

// Copyright (C) 2014 Fehmi Can Saglam (@fehmicans) and contributors.
// See the LICENCE.txt file distributed with this work for additional
// information regarding copyright ownership.
//
// 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 reactivemongo.extensions.fixtures

import scala.concurrent.{ ExecutionContext, Future }
import play.api.libs.iteratee.Enumerator
import play.api.libs.json.JsObject
import reactivemongo.bson.BSONDocument
import reactivemongo.api.DefaultDB
import reactivemongo.api.commands.WriteResult
import reactivemongo.api.collections.bson.BSONCollection
import reactivemongo.play.json.BSONFormats
import reactivemongo.extensions.util.Logger

class BsonFixtures(db: => Future[DefaultDB])(implicit ec: ExecutionContext) extends Fixtures[BSONDocument] {
	def map(document: JsObject): BSONDocument =
		BSONFormats.BSONDocumentFormat.reads(document).get

	def bulkInsert(collectionName: String, documents: Stream[BSONDocument]): Future[Int] = db.flatMap(_.collection[BSONCollection](
		collectionName).bulkInsert(documents, ordered = true).map(_.n))

	def removeAll(collectionName: String): Future[WriteResult] =
		db.flatMap(_.collection[BSONCollection](collectionName).
			remove(selector = BSONDocument.empty, firstMatchOnly = false))

	def drop(collectionName: String): Future[Unit] =
		db.flatMap(_.collection[BSONCollection](collectionName).drop())

}

object BsonFixtures {
	def apply(db: Future[DefaultDB])(implicit ec: ExecutionContext): BsonFixtures =
		new BsonFixtures(db)
}