(FM-7691) create and pass through context when connecting a transport

lib/puppet/resource_api/transport.rb

@@ -1,8 +1,4 @@
-# rubocop:disable Style/Documentation
-module Puppet; end
-module Puppet::ResourceApi; end
-module Puppet::ResourceApi::Transport; end
-# rubocop:enable Style/Documentation
+require 'puppet/resource_api/puppet_context'
 
 # Remote target transport API
 module Puppet::ResourceApi::Transport
@@ -23,7 +19,8 @@ def connect(name, connection_info)
     validate(name, connection_info)
     require "puppet/transport/#{name}"
     class_name = name.split('_').map { |e| e.capitalize }.join
-    Puppet::Transport.const_get(class_name).new(connection_info)
+    context = Puppet::ResourceApi::PuppetContext.new(@transports[name])
+    Puppet::Transport.const_get(class_name).new(context, connection_info)
   end
   module_function :connect # rubocop:disable Style/AccessModifierDeclarations
 

spec/puppet/resource_api/transport_spec.rb

@@ -111,6 +111,8 @@
     context 'when the transport file does exist' do
       context 'with an incorrectly defined transport' do
         it 'throws a NameError' do
+          described_class.register(schema)
+
           expect(described_class).to receive(:validate).with(name, host: 'example.com')
           expect(described_class).to receive(:require).with('puppet/transport/test_target')
           expect { described_class.connect(name, host: 'example.com') }.to raise_error NameError,
@@ -120,13 +122,17 @@
 
       context 'with a correctly defined transport' do
         let(:test_target) { double('Puppet::Transport::TestTarget') } # rubocop:disable RSpec/VerifiedDoubles
+        let(:context) { instance_double(Puppet::ResourceApi::PuppetContext, 'context') }
 
         it 'loads initiates the class successfully' do
+          described_class.register(schema)
+
           expect(described_class).to receive(:require).with('puppet/transport/test_target')
           expect(described_class).to receive(:validate).with(name, host: 'example.com')
+          expect(Puppet::ResourceApi::PuppetContext).to receive(:new).with(kind_of(Puppet::ResourceApi::TransportSchemaDef)).and_return(context)
 
           stub_const('Puppet::Transport::TestTarget', test_target)
-          expect(test_target).to receive(:new).with(host: 'example.com')
+          expect(test_target).to receive(:new).with(context, host: 'example.com')
 
           described_class.connect(name, host: 'example.com')
         end