spec/requestor_spec.rb in launchdarkly-server-sdk-5.6.2 vs spec/requestor_spec.rb in launchdarkly-server-sdk-5.7.0

@@ -2,14 +2,17 @@
 require "spec_helper"
 
 $sdk_key = "secret"
 
 describe LaunchDarkly::Requestor do
-  def with_requestor(base_uri)
-    r = LaunchDarkly::Requestor.new($sdk_key, LaunchDarkly::Config.new(base_uri: base_uri))
-    yield r
-    r.stop
+  def with_requestor(base_uri, opts = {})
+    r = LaunchDarkly::Requestor.new($sdk_key, LaunchDarkly::Config.new({ base_uri: base_uri }.merge(opts)))
+    begin
+      yield r
+    ensure
+      r.stop
+    end
   end
 
   describe "request_all_flags" do
     it "uses expected URI and headers" do
       with_server do |server|
@@ -54,9 +57,22 @@
           expect(server.requests[1].header).to include({ "if-none-match" => [ etag ] })
         end
       end
     end
 
+    it "sends wrapper header if configured" do
+      with_server do |server|
+        with_requestor(server.base_uri.to_s, { wrapper_name: 'MyWrapper', wrapper_version: '1.0' }) do |requestor|
+          server.setup_ok_response("/", "{}")
+          requestor.request_all_data()
+          expect(server.requests.count).to eq 1
+          expect(server.requests[0].header).to include({
+            "x-launchdarkly-wrapper" => [ "MyWrapper/1.0" ]
+          })
+        end
+      end
+    end
+    
     it "can reuse cached data" do
       etag = "xyz"
       expected_data = { flags: { x: { key: "x" } } }
       with_server do |server|
         with_requestor(server.base_uri.to_s) do |requestor|