1 files changed, 31 insertions, 0 deletions
diff --git a/spec/models/data_list_spec.rb b/spec/models/data_list_spec.rb
new file mode 100644
index 00000000000..d2f15386808
--- /dev/null
+++ b/spec/models/data_list_spec.rb
@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe DataList do
+  describe '#to_array' do
+    let(:jira_integration) { create(:jira_integration) }
+    let(:zentao_integration) { create(:zentao_integration) }
+    let(:cases) do
+      [
+        [jira_integration, 'Integrations::JiraTrackerData', 'service_id'],
+        [zentao_integration, 'Integrations::ZentaoTrackerData', 'integration_id']
+      ]
+    end
+
+    def data_list(integration)
+      DataList.new([integration], integration.to_data_fields_hash, integration.data_fields.class).to_array
+    end
+
+    it 'returns current data' do
+      cases.each do |integration, data_fields_class_name, foreign_key|
+        data_fields_klass, columns, values_items = data_list(integration)
+
+        expect(data_fields_klass.to_s).to eq data_fields_class_name
+        expect(columns.last).to eq foreign_key
+        values = values_items.first
+        expect(values.last).to eq integration.id
+      end
+    end
+  end
+end