Cooking some test

Author: nattsw

plugin.rb

@@ -48,16 +48,11 @@ module ::DiscourseTranslator
     scope.can_translate?(object)
   end
 
-  add_to_class :post, :has_translation? do |language|
-    translated = custom_fields[::DiscourseTranslator::TRANSLATED_CUSTOM_FIELD]
-    !(language.blank? || translated.blank? || translated[language].blank?)
-  end
-
   add_to_serializer :post, :cooked, false do
     return super() if !SiteSetting.experimental_topic_translation
     return super() if scope.request.params["show_original"].present?
-    return super() if !object.has_translation?(I18n.locale)
     translated = object.custom_fields[::DiscourseTranslator::TRANSLATED_CUSTOM_FIELD]
+    return super() if (translated.blank? || translated[I18n.locale].blank?)
     translated[I18n.locale]
   end
 end

spec/serializers/post_serializer_spec.rb

@@ -80,4 +80,49 @@
       end
     end
   end
+
+  describe "#cooked" do
+    def serialize_post(user: nil, params: {}, cookies: {})
+      env = {
+        "action_dispatch.request.parameters" => params,
+        "REQUEST_METHOD" => "GET",
+        "rack.input" => {
+        },
+        "HTTP_COOKIE" => cookies.map { |k, v| "#{k}=#{v}" }.join(";"),
+      }
+      request = ActionDispatch::Request.new(env)
+      guardian = Guardian.new(user, request)
+      PostSerializer.new(post, scope: guardian)
+    end
+
+    before { SiteSetting.experimental_topic_translation = true }
+
+    it "returns original cooked when experimental_topic_translation is disabled" do
+      SiteSetting.experimental_topic_translation = false
+      original_cooked = post.cooked
+      expect(serialize_post(user: user).cooked).to eq(original_cooked)
+    end
+
+    it "returns original cooked when show_original param is present" do
+      original_cooked = post.cooked
+      expect(serialize_post(user: user, params: { "show_original" => "true" }).cooked).to eq(
+        original_cooked,
+      )
+    end
+
+    it "returns translated content based on locale" do
+      I18n.locale = "ja"
+      post.custom_fields[DiscourseTranslator::TRANSLATED_CUSTOM_FIELD] = {
+        "ja" => "こんにちは",
+        "es" => "Hola",
+      }
+      expect(serialize_post(user: user).cooked).to eq("こんにちは")
+    end
+
+    it "returns original cooked when plugin is disabled" do
+      SiteSetting.translator_enabled = false
+      original_cooked = post.cooked
+      expect(serialize_post(user: user).cooked).to eq(original_cooked)
+    end
+  end
 end