1 files changed, 102 insertions, 0 deletions
diff --git a/spec/models/performance_monitoring/prometheus_dashboard_spec.rb b/spec/models/performance_monitoring/prometheus_dashboard_spec.rb
new file mode 100644
index 00000000000..e6fc03a0fb6
--- /dev/null
+++ b/spec/models/performance_monitoring/prometheus_dashboard_spec.rb
@@ -0,0 +1,102 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe PerformanceMonitoring::PrometheusDashboard do
+  let(:json_content) do
+    {
+      "dashboard" => "Dashboard Title",
+      "templating" => {
+        "variables" => {
+          "variable1" => %w(value1 value2 value3)
+        }
+      },
+      "panel_groups" => [{
+        "group" => "Group Title",
+        "panels" => [{
+          "type" => "area-chart",
+          "title" => "Chart Title",
+          "y_label" => "Y-Axis",
+          "metrics" => [{
+            "id" => "metric_of_ages",
+            "unit" => "count",
+            "label" => "Metric of Ages",
+            "query_range" => "http_requests_total"
+          }]
+        }]
+      }]
+    }
+  end
+
+  describe '.from_json' do
+    subject { described_class.from_json(json_content) }
+
+    it 'creates a PrometheusDashboard object' do
+      expect(subject).to be_a PerformanceMonitoring::PrometheusDashboard
+      expect(subject.dashboard).to eq(json_content['dashboard'])
+      expect(subject.panel_groups).to all(be_a PerformanceMonitoring::PrometheusPanelGroup)
+    end
+
+    describe 'validations' do
+      context 'when dashboard is missing' do
+        before do
+          json_content['dashboard'] = nil
+        end
+
+        subject { described_class.from_json(json_content) }
+
+        it { expect { subject }.to raise_error(ActiveModel::ValidationError) }
+      end
+
+      context 'when panel groups are missing' do
+        before do
+          json_content['panel_groups'] = []
+        end
+
+        subject { described_class.from_json(json_content) }
+
+        it { expect { subject }.to raise_error(ActiveModel::ValidationError) }
+      end
+    end
+  end
+
+  describe '.find_for' do
+    let(:project) { build_stubbed(:project) }
+    let(:user) { build_stubbed(:user) }
+    let(:environment) { build_stubbed(:environment) }
+    let(:path) { ::Metrics::Dashboard::SystemDashboardService::DASHBOARD_PATH }
+
+    context 'dashboard has been found' do
+      it 'uses dashboard finder to find and load dashboard data and returns dashboard instance', :aggregate_failures do
+        expect(Gitlab::Metrics::Dashboard::Finder).to receive(:find).with(project, user, environment: environment, dashboard_path: path).and_return(status: :success, dashboard: json_content)
+
+        dashboard_instance = described_class.find_for(project: project, user: user, path: path, options: { environment: environment })
+
+        expect(dashboard_instance).to be_instance_of described_class
+        expect(dashboard_instance.environment).to be environment
+        expect(dashboard_instance.path).to be path
+      end
+    end
+
+    context 'dashboard has NOT been found' do
+      it 'returns nil' do
+        allow(Gitlab::Metrics::Dashboard::Finder).to receive(:find).and_return(status: :error)
+
+        dashboard_instance = described_class.find_for(project: project, user: user, path: path, options: { environment: environment })
+
+        expect(dashboard_instance).to be_nil
+      end
+    end
+  end
+
+  describe '#to_yaml' do
+    subject { prometheus_dashboard.to_yaml }
+
+    let(:prometheus_dashboard) { described_class.from_json(json_content) }
+    let(:expected_yaml) do
+      "---\npanel_groups:\n- panels:\n  - metrics:\n    - id: metric_of_ages\n      unit: count\n      label: Metric of Ages\n      query: \n      query_range: http_requests_total\n    type: area-chart\n    title: Chart Title\n    y_label: Y-Axis\n    weight: \n  group: Group Title\n  priority: \ndashboard: Dashboard Title\n"
+    end
+
+    it { is_expected.to eq(expected_yaml) }
+  end
+end