From 519fe290e3f6bef6bfe97f8869a54a7cafb55d8f Mon Sep 17 00:00:00 2001
From: Vinoth Kannan <vinothkannan@vinkas.com>
Date: Mon, 14 Oct 2019 12:57:33 +0530
Subject: [PATCH] SPEC: 'lookup_upload_urls' method should use cdn url if
 available.

e4fe864c0b7d4d5d1f0cd604829eca81b71bb4ec
---
 spec/components/pretty_text/helpers_spec.rb | 22 +++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 spec/components/pretty_text/helpers_spec.rb

diff --git a/spec/components/pretty_text/helpers_spec.rb b/spec/components/pretty_text/helpers_spec.rb
new file mode 100644
index 00000000000..1f5c882cd1f
--- /dev/null
+++ b/spec/components/pretty_text/helpers_spec.rb
@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require 'rails_helper'
+
+describe PrettyText::Helpers do
+
+  context "lookup_upload_urls" do
+    let(:upload) { Fabricate(:upload) }
+
+    it "should return cdn url if available" do
+      short_url = upload.short_url
+      result = PrettyText::Helpers.lookup_upload_urls([short_url])
+      expect(result[short_url][:url]).to eq(upload.url)
+
+      set_cdn_url "https://awesome.com"
+
+      result = PrettyText::Helpers.lookup_upload_urls([short_url])
+      expect(result[short_url][:url]).to eq("https://awesome.com#{upload.url}")
+    end
+  end
+
+end