2 * Copyright 2011-2012 GRNET S.A. All rights reserved.
4 * Redistribution and use in source and binary forms, with or
5 * without modification, are permitted provided that the following
8 * 1. Redistributions of source code must retain the above
9 * copyright notice, this list of conditions and the following
12 * 2. Redistributions in binary form must reproduce the above
13 * copyright notice, this list of conditions and the following
14 * disclaimer in the documentation and/or other materials
15 * provided with the distribution.
17 * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
18 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
20 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
21 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
24 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
25 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
27 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28 * POSSIBILITY OF SUCH DAMAGE.
30 * The views and conclusions contained in the software and
31 * documentation are those of the authors and should not be
32 * interpreted as representing official policies, either expressed
33 * or implied, of GRNET S.A.
36 package gr.grnet.aquarium.store.mongodb
38 import org.junit.Assert._
39 import org.junit.Assume._
40 import gr.grnet.aquarium.Configurator._
41 import gr.grnet.aquarium.util.{RandomEventGenerator, TestMethods}
42 import gr.grnet.aquarium.logic.events.ResourceEvent
43 import collection.mutable.ArrayBuffer
44 import org.junit.{After, Test, Before}
45 import gr.grnet.aquarium.{StoreConfigurator, LogicTestsAssumptions}
46 import gr.grnet.aquarium.store.memory.MemStore
49 * @author Georgios Gousios <gousiosg@gmail.com>
51 class EventStoreTest extends TestMethods
52 with RandomEventGenerator with StoreConfigurator {
55 def testStoreEvent() = {
56 assumeTrue(LogicTestsAssumptions.EnableStoreTests)
58 val event = nextResourceEvent()
59 val store = config.resourceEventStore
60 val result = store.storeResourceEvent(event)
66 def testFindEventById(): Unit = {
67 assumeTrue(LogicTestsAssumptions.EnableStoreTests)
69 val event = nextResourceEvent()
70 val store = config.resourceEventStore
72 val result1 = store.storeResourceEvent(event)
73 assert(result1.isJust)
75 val result2 = store.findResourceEventById(event.id)
76 assertNotNone(result2)
80 def testfindEventsByUserId(): Unit = {
81 assumeTrue(LogicTestsAssumptions.EnableStoreTests)
82 val events = new ArrayBuffer[ResourceEvent]()
83 val store = config.resourceEventStore
87 val e = nextResourceEvent
89 store.storeResourceEvent(e)
92 val mostUsedId = events
96 .foldLeft(("",0))((acc, kv) => if (kv._2 > acc._2) kv else acc)._1
98 val result = store.findResourceEventsByUserId(mostUsedId)(None)
99 assertEquals(events.filter(p => p.userID.equals(mostUsedId)).size, result.size)
103 def testMultipleMongos = {
104 assumeTrue(LogicTestsAssumptions.EnableStoreTests)
107 //assertEquals(a.Connection.mongo.get.hashCode(), b.Connection.mongo.get.hashCode())
112 if (isInMemStore) return
116 val col = a.mongo.getDB(
117 MasterConfigurator.get(Keys.persistence_db)
118 ).getCollection(MongoDBStore.RESOURCE_EVENTS_COLLECTION)
121 //while (res.hasNext)
122 // col.remove(res.next)
125 private lazy val config = configurator
126 private val isInMemStore = config.resourceEventStore.isInstanceOf[MemStore]
127 private def getMongo = config.resourceEventStore.asInstanceOf[MongoDBStore]