From 979ff9e765e3c08501cbd00354a87013853fe796 Mon Sep 17 00:00:00 2001
From: vlad <vlad@drivergrp.com>
Date: Tue, 19 Jul 2016 15:01:30 -0400
Subject: Unit tests for core code and bug fixes

---
 src/main/scala/com/drivergrp/core/app.scala | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

(limited to 'src/main/scala/com/drivergrp/core/app.scala')

diff --git a/src/main/scala/com/drivergrp/core/app.scala b/src/main/scala/com/drivergrp/core/app.scala
index 13663ab..4639fc0 100644
--- a/src/main/scala/com/drivergrp/core/app.scala
+++ b/src/main/scala/com/drivergrp/core/app.scala
@@ -52,14 +52,22 @@ object app {
     }
 
     protected def bindHttp(modules: Seq[Module]): Unit = {
-      import SprayJsonSupport._
-      import DefaultJsonProtocol._
-
       val serviceTypes   = modules.flatMap(_.routeTypes)
       val swaggerService = new Swagger(actorSystem, serviceTypes, config)
       val swaggerRoutes  = swaggerService.routes ~ swaggerService.swaggerUI
+      val versionRt      = versionRoute(version, buildNumber)
+
+      val _ = http.bindAndHandle(
+          route2HandlerFlow(logRequestResult("log")(modules.map(_.route).foldLeft(versionRt ~ swaggerRoutes)(_ ~ _))),
+          interface,
+          port)(materializer)
+    }
 
-      val versionRoute = path("version") {
+    protected def versionRoute(version: String, buildNumber: Int) = {
+      import SprayJsonSupport._
+      import DefaultJsonProtocol._
+
+      path("version") {
         complete(
             Map(
                 "version"     -> version,
@@ -67,12 +75,6 @@ object app {
                 "serverTime"  -> time.currentTime().millis.toString
             ))
       }
-
-      val _ = http.bindAndHandle(
-          route2HandlerFlow(
-              logRequestResult("log")(modules.map(_.route).foldLeft(versionRoute ~ swaggerRoutes)(_ ~ _))),
-          interface,
-          port)(materializer)
     }
 
     /**
-- 
cgit v1.2.3