/wallet/seed api method protected by api key

input-output-hk · Jun 3, 2016 · 9c5391e · 9c5391e
1 parent 47f8766
commit 9c5391e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/scorex-transaction/src/main/scala/scorex/api/http/WalletApiRoute.scala b/scorex-transaction/src/main/scala/scorex/api/http/WalletApiRoute.scala
@@ -22,11 +22,12 @@ case class WalletApiRoute(override val application: Application)(implicit val co
   @ApiOperation(value = "Seed", notes = "Export wallet seed", httpMethod = "GET")
   def seed: Route = {
     path("wallet" / "seed") {
-      getJsonRoute {
-        lazy val seedJs = Json.obj("seed" -> Base58.encode(wallet.seed))
-        walletNotExists(wallet).getOrElse(seedJs)
+      withAuth {
+        getJsonRoute {
+          lazy val seedJs = Json.obj("seed" -> Base58.encode(wallet.seed))
+          walletNotExists(wallet).getOrElse(seedJs)
+        }
       }
-
     }
   }
 

diff --git a/src/test/scala/scorex/lagonaki/integration/api/WalletAPISpecification.scala b/src/test/scala/scorex/lagonaki/integration/api/WalletAPISpecification.scala
@@ -12,7 +12,10 @@ class WalletAPISpecification extends FunSuite with Matchers {
   }
 
   test("/wallet/seed API route") {
-    (GET.request("/wallet/seed") \ "seed").as[String] shouldBe Base58.encode(application.settings.walletSeed.get)
+    GET.incorrectApiKeyTest("/wallet/seed")
+
+    val response = GET.request("/wallet/seed", headers =  Map("api_key" -> "test"))
+    (response \ "seed").as[String] shouldBe Base58.encode(application.settings.walletSeed.get)
   }
 
 }