From 871ef9806a136cb90e9c911dd98d75a49300ec96 Mon Sep 17 00:00:00 2001
From: Anton <anton@scinote.net>
Date: Fri, 10 May 2024 11:56:15 +0200
Subject: [PATCH] Update results counter logic [SCI-10675]

---
 app/models/my_module.rb                       | 8 ++++++++
 app/views/my_modules/_header_actions.html.erb | 1 +
 2 files changed, 9 insertions(+)

diff --git a/app/models/my_module.rb b/app/models/my_module.rb
index 06223062e..e1eea3c3a 100644
--- a/app/models/my_module.rb
+++ b/app/models/my_module.rb
@@ -154,6 +154,14 @@ class MyModule < ApplicationRecord
     experiment
   end
 
+  def results_count(view_mode = 'active')
+    return results.size if archived_branch?
+
+    return results.archived.size if view_mode == 'archived'
+
+    results.active.size
+  end
+
   def navigable?
     !experiment.archived? && experiment.navigable?
   end
diff --git a/app/views/my_modules/_header_actions.html.erb b/app/views/my_modules/_header_actions.html.erb
index 8e231a1b8..10c66eca8 100644
--- a/app/views/my_modules/_header_actions.html.erb
+++ b/app/views/my_modules/_header_actions.html.erb
@@ -14,6 +14,7 @@
           data-e2e="e2e-BT-taskTabs-results"
       >
         <%= t("nav2.modules.results") %>
+        <%= "[#{@my_module.results_count(params[:view_mode])}]" %>
       </a>
       <a class="p-3 border-b-4 border-transparent hover:no-underline uppercase text-bold capitalize <%= is_module_activities? ? "text-sn-blue" : "text-sn-grey" %>"
           href="<%= activities_my_module_url(@my_module, view_mode: params[:view_mode]) %>"