added a test case showing the use of the array accessor style access …

…to headers/properties
koolapp · May 12, 2012 · 6a4fa0d · 6a4fa0d
1 parent 730def6
commit 6a4fa0d
Show file tree

Hide file tree

Showing 9 changed files with 67 additions and 6 deletions.
diff --git a/koolapp-camel/src/main/kotlin/org/koolapp/camel/Exchanges.kt b/koolapp-camel/src/main/kotlin/org/koolapp/camel/Exchanges.kt
@@ -39,3 +39,13 @@ var Exchange.out: Message
  */
 inline fun Exchange.bodyString(nullValue: String = ""): String = body<String>(javaClass<String>) ?: nullValue
 
+/**
+ * Provides array style access to properties on the exchange
+ */
+inline fun Exchange.get(propertyName: String): Any? = getProperty(propertyName)
+
+/**
+ * Provides array style access to properties on the exchange
+ */
+inline fun Exchange.set(propertyName: String, value: Any?): Unit = setProperty(propertyName, value)
+
diff --git a/koolapp-camel/src/main/kotlin/org/koolapp/camel/Messages.kt b/koolapp-camel/src/main/kotlin/org/koolapp/camel/Messages.kt
@@ -2,9 +2,28 @@ package org.koolapp.camel
 
 import org.apache.camel.Exchange
 import org.apache.camel.Message
+import java.util.Map
 
 var Message.body: Any?
     get() = getBody()
     set(value) {
         setBody(value)
     }
+
+var Message.headers: Map<String?, Any?>
+    get() = getHeaders().orEmpty()
+    set(value) {
+        setHeaders(value)
+    }
+
+
+/**
+ * Provides array style access to headers on the message
+ */
+inline fun Message.get(headerName: String): Any? = getHeader(headerName)
+
+/**
+ * Provides array style access to headers on the message
+ */
+inline fun Message.set(headerName: String, value: Any?): Unit = setHeader(headerName, value)
+
diff --git a/koolapp-camel/src/test/kotlin/test/koolapp/camel/ChoiceTest.kt b/koolapp-camel/src/test/kotlin/test/koolapp/camel/ChoiceTest.kt
@@ -40,7 +40,7 @@ class ChoiceTest {
                 result.assertIsSatisfied()
                 println("$result has messages:")
                 for (exchange in result.getReceivedExchanges()) {
-                    println("    ${exchange?.getIn()}")
+                    println("    ${exchange?.input}")
                 }
                 println()
             }

diff --git a/koolapp-camel/src/test/kotlin/test/koolapp/camel/FilterTest.kt b/koolapp-camel/src/test/kotlin/test/koolapp/camel/FilterTest.kt
@@ -26,7 +26,7 @@ class FilterTest {
             result.assertIsSatisfied()
 
             for (exchange in result.getReceivedExchanges()) {
-                println("Found message ${exchange?.getIn()}")
+                println("Found message ${exchange?.input}")
             }
         }
     }

diff --git a/koolapp-camel/src/test/kotlin/test/koolapp/camel/FilterThenTest.kt b/koolapp-camel/src/test/kotlin/test/koolapp/camel/FilterThenTest.kt
@@ -27,7 +27,7 @@ class FilterThenTest {
             result.assertIsSatisfied()
 
             for (exchange in result.getReceivedExchanges()) {
-                println("Found message ${exchange?.getIn()}")
+                println("Found message ${exchange?.input}")
             }
         }
     }

diff --git a/koolapp-camel/src/test/kotlin/test/koolapp/camel/ProcessTest.kt b/koolapp-camel/src/test/kotlin/test/koolapp/camel/ProcessTest.kt
@@ -22,7 +22,7 @@ class ProcessTest {
             result.assertIsSatisfied()
 
             for (exchange in result.getReceivedExchanges()) {
-                println("Found message ${exchange?.getIn()}")
+                println("Found message ${exchange?.input}")
             }
         }
     }

diff --git a/koolapp-camel/src/test/kotlin/test/koolapp/camel/RouteBuilderTest.kt b/koolapp-camel/src/test/kotlin/test/koolapp/camel/RouteBuilderTest.kt
@@ -27,7 +27,7 @@ class RouteBuilderTest {
             result.assertIsSatisfied()
 
             for (exchange in result.getReceivedExchanges()) {
-                println("Found message ${exchange?.getIn()}")
+                println("Found message ${exchange?.input}")
             }
         }
     }

diff --git a/koolapp-camel/src/test/kotlin/test/koolapp/camel/TransformAndSetHeaderTest.kt b/koolapp-camel/src/test/kotlin/test/koolapp/camel/TransformAndSetHeaderTest.kt
@@ -0,0 +1,32 @@
+package test.koolapp.camel
+
+import org.koolapp.camel.*
+
+import org.junit.Test as test
+import org.apache.camel.component.mock.MockEndpoint
+
+class TransformAndSetHeaderTest {
+    test fun createRoute() {
+        camel {
+            val result = mockEndpoint("mock:result")
+            routes {
+                from("seda:foo") {
+                    transform {
+                        out["foo"] = "bar"
+                        "Hello ${bodyString()}"
+                    }.sendTo(result)
+                }
+            }
+            result.expectedBodiesReceived("Hello world!")
+
+            val producer = producerTemplate()
+            producer.sendBody("seda:foo", "world!")
+
+            result.assertIsSatisfied()
+
+            for (exchange in result.getReceivedExchanges()) {
+                println("Found message ${exchange?.input} with headers ${exchange?.input?.headers}")
+            }
+        }
+    }
+}
diff --git a/koolapp-camel/src/test/kotlin/test/koolapp/camel/TransformTest.kt b/koolapp-camel/src/test/kotlin/test/koolapp/camel/TransformTest.kt
@@ -22,7 +22,7 @@ class TransformTest {
             result.assertIsSatisfied()
 
             for (exchange in result.getReceivedExchanges()) {
-                println("Found message ${exchange?.getIn()}")
+                println("Found message ${exchange?.input}")
             }
         }
     }