From e0d79d10485a11d620ac35c9dcc414089dea48b2 Mon Sep 17 00:00:00 2001
From: "J. Brisbin" <jon@jbrisbin.com>
Date: Thu, 9 Dec 2010 12:53:29 -0600
Subject: [PATCH] Setting default QosParameters for tests.

---
 .../data/keyvalue/riak/core/AbstractRiakTemplate.java     | 8 ++++++++
 .../springframework/data/RiakKeyValueTemplateTests.xml    | 7 ++++++-
 .../org/springframework/data/RiakTemplateTests.xml        | 7 ++++++-
 3 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/spring-data-riak/src/main/java/org/springframework/data/keyvalue/riak/core/AbstractRiakTemplate.java b/spring-data-riak/src/main/java/org/springframework/data/keyvalue/riak/core/AbstractRiakTemplate.java
index abf45a9fd..3ed61af18 100644
--- a/spring-data-riak/src/main/java/org/springframework/data/keyvalue/riak/core/AbstractRiakTemplate.java
+++ b/spring-data-riak/src/main/java/org/springframework/data/keyvalue/riak/core/AbstractRiakTemplate.java
@@ -168,6 +168,14 @@ public abstract class AbstractRiakTemplate extends RestGatewaySupport implements
     this.useCache = useCache;
   }
 
+  public QosParameters getDefaultQosParameters() {
+    return defaultQosParameters;
+  }
+
+  public void setDefaultQosParameters(QosParameters defaultQosParameters) {
+    this.defaultQosParameters = defaultQosParameters;
+  }
+
   /**
    * Extract the prefix from the URI for use in creating links.
    *
diff --git a/spring-data-riak/src/test/resources/org/springframework/data/RiakKeyValueTemplateTests.xml b/spring-data-riak/src/test/resources/org/springframework/data/RiakKeyValueTemplateTests.xml
index 9238a146e..43aafe1d5 100644
--- a/spring-data-riak/src/test/resources/org/springframework/data/RiakKeyValueTemplateTests.xml
+++ b/spring-data-riak/src/test/resources/org/springframework/data/RiakKeyValueTemplateTests.xml
@@ -1,11 +1,16 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <beans xmlns="http://www.springframework.org/schema/beans"
+       xmlns:p="http://www.springframework.org/schema/p"
        xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
        xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd">
 
   <import resource="classpath:/META-INF/spring/app-context.xml"/>
 
+  <bean id="qos" class="org.springframework.data.keyvalue.riak.core.RiakQosParameters"
+        p:durableWriteThreshold="all"
+        p:writeThreshold="all"/>
   <bean id="riakTemplate"
-        class="org.springframework.data.keyvalue.riak.core.RiakKeyValueTemplate"/>
+        class="org.springframework.data.keyvalue.riak.core.RiakKeyValueTemplate"
+        p:defaultQosParameters-ref="qos"/>
 
 </beans>
diff --git a/spring-data-riak/src/test/resources/org/springframework/data/RiakTemplateTests.xml b/spring-data-riak/src/test/resources/org/springframework/data/RiakTemplateTests.xml
index 9163cdbaa..e8ebcfb28 100644
--- a/spring-data-riak/src/test/resources/org/springframework/data/RiakTemplateTests.xml
+++ b/spring-data-riak/src/test/resources/org/springframework/data/RiakTemplateTests.xml
@@ -18,12 +18,17 @@
   -->
 
 <beans xmlns="http://www.springframework.org/schema/beans"
+       xmlns:p="http://www.springframework.org/schema/p"
        xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
        xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd">
 
   <import resource="classpath:/META-INF/spring/app-context.xml"/>
 
+  <bean id="qos" class="org.springframework.data.keyvalue.riak.core.RiakQosParameters"
+        p:durableWriteThreshold="all"
+        p:writeThreshold="all"/>
   <bean id="riakTemplate"
-        class="org.springframework.data.keyvalue.riak.core.RiakTemplate"/>
+        class="org.springframework.data.keyvalue.riak.core.RiakTemplate"
+        p:defaultQosParameters-ref="qos"/>
 
 </beans>