From 464cdf9ec765f650fe8976df30b8a65708de1186 Mon Sep 17 00:00:00 2001
From: ruslandoga <67764432+ruslandoga@users.noreply.github.com>
Date: Sun, 10 Nov 2024 20:21:37 +0700
Subject: [PATCH] test package delete

---
 test/hexdocs/search_test.exs | 38 ++++++++++++++++++++++++++++++------
 1 file changed, 32 insertions(+), 6 deletions(-)

diff --git a/test/hexdocs/search_test.exs b/test/hexdocs/search_test.exs
index 71e3992..9a98b52 100644
--- a/test/hexdocs/search_test.exs
+++ b/test/hexdocs/search_test.exs
@@ -21,7 +21,7 @@ defmodule Hexdocs.SearchTest do
     :ok
   end
 
-  defp run_queue(:put, package, version, files) do
+  defp run_upload(package, version, files) do
     tar = Hexdocs.Tar.create(files)
     key = "docs/#{package}-#{version}.tar.gz"
     Hexdocs.Store.put!(:repo_bucket, key, tar)
@@ -29,8 +29,17 @@ defmodule Hexdocs.SearchTest do
     assert_receive {:ack, ^ref, [_], []}
   end
 
-  test "indexes public search_data", %{test: test} do
-    run_queue(:put, _package = test, _version = "1.0.0", [
+  defp run_delete(package, version) do
+    key = "docs/#{package}-#{version}.tar.gz"
+    ref = Broadway.test_message(Hexdocs.Queue, queue_delete_message(key))
+    assert_receive {:ack, ^ref, [_], []}
+  end
+
+  test "indexes public search_data on upload and deindexes it on delete", %{test: test} do
+    package = test
+    version = "1.0.0"
+
+    run_upload(package, version, [
       {"index.html", "contents"},
       {"dist/search_data-0F918FFD.js",
        """
@@ -41,13 +50,15 @@ defmodule Hexdocs.SearchTest do
        """}
     ])
 
+    full_package = "#{package}-#{version}"
+
     assert [
              %{
                "document" => %{
                  "doc" => "example text",
                  "id" => "0",
                  "proglang" => "elixir",
-                 "package" => "test indexes public search_data-1.0.0",
+                 "package" => ^full_package,
                  "ref" => "Example.html",
                  "title" => "Example",
                  "type" => "module"
@@ -58,7 +69,7 @@ defmodule Hexdocs.SearchTest do
                  "doc" => "does example things",
                  "id" => "1",
                  "proglang" => "elixir",
-                 "package" => "test indexes public search_data-1.0.0",
+                 "package" => ^full_package,
                  "ref" => "Example.html#test/4",
                  "title" => "Example.test/4",
                  "type" => "function"
@@ -72,13 +83,17 @@ defmodule Hexdocs.SearchTest do
                  "doc" => "does example things",
                  "id" => "1",
                  "proglang" => "elixir",
-                 "package" => "test indexes public search_data-1.0.0",
+                 "package" => ^full_package,
                  "ref" => "Example.html#test/4",
                  "title" => "Example.test/4",
                  "type" => "function"
                }
              }
            ] = typesense_search(%{"q" => "thing", "query_by" => "doc"})
+
+    run_delete(package, version)
+
+    assert typesense_search(%{"q" => "example", "query_by" => "title"}) == []
   end
 
   defp queue_put_message(key) do
@@ -92,6 +107,17 @@ defmodule Hexdocs.SearchTest do
     })
   end
 
+  defp queue_delete_message(key) do
+    Jason.encode!(%{
+      "Records" => [
+        %{
+          "eventName" => "ObjectRemoved:Delete",
+          "s3" => %{"object" => %{"key" => key}}
+        }
+      ]
+    })
+  end
+
   defp typesense_new_collection do
     collection = Typesense.collection()
     api_key = Typesense.api_key()