Added tests for source and destination factories.

spec/support/factories.rb

@@ -7,5 +7,6 @@
 
 require_relative 'factories/config/destination_factory'
 require_relative 'factories/config/source_factory'
+require_relative 'factories/destination/one_drive_factory'
 require_relative 'factories/service/one_drive_factory'
 require_relative 'factories/source/url_factory'

spec/support/factories/destination/one_drive_factory.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+require_relative '../../../../lib/destination/one_drive'
+
+FactoryBot.define do
+  factory :destination_one_drive, class: DocumentTransfer::Destination::OneDrive do
+    transient do
+      config { build(:config_destination, type: :onedrive, path: 'rspec-docs') }
+    end
+
+    initialize_with { new(config) }
+  end
+end

spec/unit/document_transfer/destination_spec.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require_relative '../../../lib/destination'
+
+RSpec.describe DocumentTransfer::Destination do
+  describe '.load' do
+    let(:config) { build(:config_destination, type: destination_type) }
+    let(:destination) { build(:destination_one_drive) }
+    let(:destination_type) { :onedrive }
+
+    before do
+      allow(destination.class).to receive(:new).and_return(destination)
+    end
+
+    it 'returns the proper destination' do
+      expect(described_class.load(config)).to eq(destination)
+    end
+
+    context 'when an invalid destination type is provided' do
+      before do
+        # If we try to set an invalid type directly on the config object, it
+        # will raise an error.
+        allow(config).to receive(:type).and_return(:invalid)
+      end
+
+      it 'raises an exception' do
+        expect { described_class.load(config) }.to \
+          raise_error(DocumentTransfer::Destination::InvalidDestinationError)
+      end
+    end
+  end
+end

spec/unit/document_transfer/source_spec.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require_relative '../../../lib/source'
+
+RSpec.describe DocumentTransfer::Source do
+  describe '.load' do
+    let(:config) { build(:config_source, type: source_type) }
+    let(:source) { build(:source_url) }
+    let(:source_type) { :url }
+
+    before do
+      allow(source.class).to receive(:new).and_return(source)
+    end
+
+    it 'returns the proper source' do
+      expect(described_class.load(config)).to eq(source)
+    end
+
+    context 'when an invalid source type is provided' do
+      before do
+        # If we try to set an invalid type directly on the config object, it
+        # will raise an error.
+        allow(config).to receive(:type).and_return(:invalid)
+      end
+
+      it 'raises an exception' do
+        expect { described_class.load(config) }.to \
+          raise_error(DocumentTransfer::Source::InvalidSourceError)
+      end
+    end
+  end
+end