Add mdm::payload test and run tests under rails 7.1

Author: dwelch-r7

db/migrate/20110317144932_add_session_table.rb

@@ -1,7 +1,11 @@
 class AddSessionTable < ActiveRecord::Migration[4.2]
 
 	class Event < ApplicationRecord
-		serialize :info
+		if ActiveRecord::VERSION::MAJOR >= 7 && ActiveRecord::VERSION::MINOR >= 1
+			serialize :info, coder: YAML
+		else
+			serialize :info, YAML
+		end
 	end
 
 	class SessionEvent < ApplicationRecord
@@ -10,7 +14,11 @@ class SessionEvent < ApplicationRecord
 
 	class Session < ApplicationRecord
 		has_many :events, :class_name => 'AddSessionTable::SessionEvent'
-		serialize :datastore
+		if ActiveRecord::VERSION::MAJOR >= 7 && ActiveRecord::VERSION::MINOR >= 1
+			serialize :datastore, coder: YAML
+		else
+			serialize :datastore, YAML
+		end
 	end
 
 	def self.up

db/migrate/20110422000000_convert_binary.rb

@@ -3,11 +3,19 @@ class ConvertBinary < ActiveRecord::Migration[4.2]
 
 
 	class WebPage < ApplicationRecord
-		serialize :headers
+		if ActiveRecord::VERSION::MAJOR >= 7 && ActiveRecord::VERSION::MINOR >= 1
+			serialize :headers, coder: YAML
+		else
+			serialize :headers, YAML
+		end
 	end
 
 	class WebVuln < ApplicationRecord
-		serialize :params
+		if ActiveRecord::VERSION::MAJOR >= 7 && ActiveRecord::VERSION::MINOR >= 1
+			serialize :params, coder: YAML
+		else
+			serialize :params, YAML
+		end
 	end
 
 	def bfilter(str)

spec/app/models/mdm/payload_spec.rb

@@ -0,0 +1,45 @@
+RSpec.describe Mdm::Payload, type: :model do
+  it_should_behave_like 'Metasploit::Concern.run'
+
+  context 'factory' do
+    it 'should be valid' do
+      mdm_payload = FactoryBot.build(:mdm_payload)
+      expect(mdm_payload).to be_valid
+    end
+  end
+
+  context 'database' do
+
+    context 'timestamps'do
+      it { is_expected.to have_db_column(:created_at).of_type(:datetime) }
+      it { is_expected.to have_db_column(:updated_at).of_type(:datetime) }
+    end
+
+    context 'columns' do
+      it { is_expected.to have_db_column(:name).of_type(:string) }
+      it { is_expected.to have_db_column(:uuid).of_type(:string) }
+      it { is_expected.to have_db_column(:uuid_mask).of_type(:integer) }
+      it { is_expected.to have_db_column(:timestamp).of_type(:integer) }
+      it { is_expected.to have_db_column(:arch).of_type(:string) }
+      it { is_expected.to have_db_column(:platform).of_type(:string) }
+      it { is_expected.to have_db_column(:urls).of_type(:string) }
+      it { is_expected.to have_db_column(:description).of_type(:string) }
+      it { is_expected.to have_db_column(:raw_payload).of_type(:string) }
+      it { is_expected.to have_db_column(:raw_payload_hash).of_type(:string) }
+      it { is_expected.to have_db_column(:build_status).of_type(:string) }
+      it { is_expected.to have_db_column(:build_opts).of_type(:string) }
+    end
+  end
+
+  context '#destroy' do
+    it 'should successfully destroy the object' do
+      payload = FactoryBot.build(:mdm_payload)
+      expect {
+        payload.destroy
+      }.to_not raise_error
+      expect {
+        payload.reload
+      }.to raise_error(ActiveRecord::RecordNotFound)
+    end
+  end
+end

spec/factories/mdm/payloads.rb

@@ -0,0 +1,7 @@
+FactoryBot.define do
+  factory :mdm_payload, :aliases => [:payload], :class => Mdm::Payload do
+    #
+    # Associations
+    #
+  end
+end