middleware_spec.rb in grape-datadog-1.2.0

- old
+ new

@@ -1,17 +1,28 @@
 require 'spec_helper'
 
 describe Grape::Datadog::Middleware do
 
+  class VersionedTestAPI < Grape::API
+    version 'v1'
+    prefix  'api'
+
+    get('versioned') { "OK" }
+  end
+
   class TestAPI < Grape::API
     use Grape::Datadog::Middleware,
       hostname: "test.host",
       tags:     ["custom:tag", lambda {|env| "scheme:#{Rack::Request.new(env).scheme}" }]
 
     get 'echo/:key1/:key2' do
       "#{params['key1']} #{params['key2']}"
     end
+
+    namespace :sub do
+      mount VersionedTestAPI
+    end
   end
 
   def app; TestAPI; end
 
   before { $statsd.buffer.clear }
@@ -29,10 +40,22 @@
     get '/echo/1/1234'
     expect(last_response.status).to eq(200)
     expect(last_response.body).to eq('1 1234')
 
     expect($statsd.buffer).to eq([
-      "grape.request:1|c|#custom:tag,scheme:http,host:test.host,method:GET,path:echo.{key1}.{key2},status:200",
-      "grape.request.time:333|ms|#custom:tag,scheme:http,host:test.host,method:GET,path:echo.{key1}.{key2},status:200"
+      "grape.request:1|c|#custom:tag,scheme:http,host:test.host,method:GET,path:/echo/_key1_/_key2_,status:200",
+      "grape.request.time:333|ms|#custom:tag,scheme:http,host:test.host,method:GET,path:/echo/_key1_/_key2_,status:200",
     ])
   end
+
+  it 'should support namespaces and versioning' do
+    get '/api/v1/sub/versioned'
+    expect(last_response.status).to eq(200)
+    expect(last_response.body).to eq('OK')
+
+    expect($statsd.buffer).to eq([
+      "grape.request:1|c|#custom:tag,scheme:http,host:test.host,method:GET,path:/sub/versioned,version:v1,status:200",
+      "grape.request.time:333|ms|#custom:tag,scheme:http,host:test.host,method:GET,path:/sub/versioned,version:v1,status:200",
+    ])
+  end
+
 end