Add tests for uri and headers

Author: steffansluis

spec/openai/client/http_spec.rb

@@ -184,4 +184,58 @@
       it { expect(parsed).to eq([{ "prompt" => ":)" }, { "prompt" => ":(" }]) }
     end
   end
+
+  describe ".uri" do
+    let(:path) { "/chat" }
+    let(:uri) { OpenAI::Client.send(:uri, path: path) }
+
+    it { expect(uri).to eq("https://api.openai.com/v1/chat") }
+
+    describe "with Azure" do
+      before do
+        OpenAI.configuration.uri_base = "https://custom-domain.openai.azure.com/openai/deployments/gpt-35-turbo"
+        OpenAI.configuration.api_type = :azure
+      end
+
+      after do
+        OpenAI.configuration.uri_base = "https://api.openai.com/"
+        OpenAI.configuration.api_type = nil
+      end
+
+      let(:path) { "/chat" }
+      let(:uri) { OpenAI::Client.send(:uri, path: path) }
+
+      it { expect(uri).to eq("https://custom-domain.openai.azure.com/openai/deployments/gpt-35-turbo/chat?api-version=v1") }
+    end
+  end
+
+  describe ".headers" do
+    before do
+      OpenAI.configuration.api_type = :nil
+    end
+
+    let(:headers) { OpenAI::Client.send(:headers) }
+
+    it {
+      expect(headers).to eq({ "Authorization" => "Bearer #{OpenAI.configuration.access_token}",
+                              "Content-Type" => "application/json", "OpenAI-Organization" => nil })
+    }
+
+    describe "with Azure" do
+      before do
+        OpenAI.configuration.api_type = :azure
+      end
+
+      after do
+        OpenAI.configuration.api_type = nil
+      end
+
+      let(:headers) { OpenAI::Client.send(:headers) }
+
+      it {
+        expect(headers).to eq({ "Content-Type" => "application/json",
+                                "api-key" => OpenAI.configuration.access_token })
+      }
+    end
+  end
 end