Add Chartkick donut chart to budgets index and update tests

Author: sarrietav-dev

Gemfile

@@ -88,3 +88,5 @@ gem "view_component", "~> 3.21"
 gem "requestjs-rails", "~> 0.0.13"
 
 gem "pagy", "~> 9.3"
+
+gem "chartkick"

Gemfile.lock

@@ -102,6 +102,7 @@ GEM
       rack-test (>= 0.6.3)
       regexp_parser (>= 1.5, < 3.0)
       xpath (~> 3.2)
+    chartkick (5.2.0)
     concurrent-ruby (1.3.5)
     connection_pool (2.5.3)
     crass (1.0.6)
@@ -436,6 +437,7 @@ DEPENDENCIES
   bootsnap
   brakeman
   capybara
+  chartkick
   debug
   erb-formatter
   erb_lint

app/controllers/budgets_controller.rb

@@ -6,6 +6,15 @@ class BudgetsController < ApplicationController
   # GET /budgets or /budgets.json
   def index
     @budgets = Budget.includes(category: [:transactions]).all
+
+    transaction_sums = Transaction.group(:category_id).sum(:amount)
+    category_ids = transaction_sums.keys.compact
+    categories = Category.where(id: category_ids).pluck(:id, :name).to_h
+
+    @donut_data = transaction_sums.each_with_object({}) do |(category_id, amount), hash|
+      label = categories[category_id] || "Unknown"
+      hash[label] = amount.abs
+    end
   end
 
   # GET /budgets/1 or /budgets/1.json

app/javascript/application.js

@@ -1,4 +1,6 @@
 // Configure your import map in config/importmap.rb. Read more: https://github.com/rails/importmap-rails
-import "@hotwired/turbo-rails"
-import "controllers"
-import "@rails/request.js"
+import "@hotwired/turbo-rails";
+import "controllers";
+import "@rails/request.js";
+import "chartkick";
+import "Chart.bundle";

app/views/budgets/index.html.erb

@@ -20,9 +20,15 @@
       id="notice"
     ><%= notice %></p>
   <% end %>
-  
+
   <article class="bg-white rounded-lg px-5 py-7 space-y-5">
-    
+    <%= pie_chart @donut_data,
+    donut: true,
+    legend: false,
+    prefix: "$",
+    library: {
+      cutout: "70%",
+    } %>
   </article>
 
   <div>

app/views/categories/show.html.erb

@@ -0,0 +1,8 @@
+<p>
+  <strong>Name:</strong>
+  <%= @category.name %>
+</p>
+
+<%= link_to "Edit", edit_category_path(@category) %>
+|
+<%= link_to "Back", categories_path %>

config/importmap.rb

@@ -7,3 +7,5 @@
 pin "@hotwired/stimulus", to: "stimulus.min.js"
 pin "@hotwired/stimulus-loading", to: "stimulus-loading.js"
 pin_all_from "app/javascript/controllers", under: "controllers"
+pin "chartkick", to: "chartkick.js"
+pin "Chart.bundle", to: "Chart.bundle.js"

test/controllers/bills_controller_test.rb

@@ -3,8 +3,12 @@
 require "test_helper"
 
 class BillsControllerTest < ActionDispatch::IntegrationTest
+  setup do
+    sign_in_as users(:one)
+  end
+
   test "should get index" do
-    get bills_index_url
+    get bills_url
     assert_response :success
   end
 end

test/controllers/budgets_controller_test.rb

@@ -5,6 +5,7 @@
 class BudgetsControllerTest < ActionDispatch::IntegrationTest
   setup do
     @budget = budgets(:one)
+    sign_in_as users(:one)
   end
 
   test "should get index" do
@@ -20,10 +21,10 @@ class BudgetsControllerTest < ActionDispatch::IntegrationTest
   test "should create budget" do
     assert_difference("Budget.count") do
       post budgets_url,
-        params: {budget: {category: @budget.category, maximum: @budget.maximum, theme: @budget.theme}}
+        params: {budget: {category_id: @budget.category_id, maximum: @budget.maximum, theme: @budget.theme}}
     end
 
-    assert_redirected_to budget_url(Budget.last)
+    assert_redirected_to budgets_path
   end
 
   test "should show budget" do
@@ -38,15 +39,15 @@ class BudgetsControllerTest < ActionDispatch::IntegrationTest
 
   test "should update budget" do
     patch budget_url(@budget),
-      params: {budget: {category: @budget.category, maximum: @budget.maximum, theme: @budget.theme}}
-    assert_redirected_to budget_url(@budget)
+      params: {budget: {category_id: @budget.category_id, maximum: @budget.maximum, theme: @budget.theme}}
+    assert_redirected_to budgets_path
   end
 
   test "should destroy budget" do
     assert_difference("Budget.count", -1) do
       delete budget_url(@budget)
     end
 
-    assert_redirected_to budgets_url
+    assert_redirected_to budgets_path
   end
 end

test/controllers/categories_controller_test.rb

@@ -1,33 +1,53 @@
+# frozen_string_literal: true
+
 require "test_helper"
 
 class CategoriesControllerTest < ActionDispatch::IntegrationTest
+  setup do
+    @category = categories(:one)
+    sign_in_as users(:one)
+  end
+
   test "should get index" do
-    get categories_index_url
+    get categories_url
     assert_response :success
   end
 
   test "should get new" do
-    get categories_new_url
+    get new_category_url
     assert_response :success
   end
 
-  test "should get create" do
-    get categories_create_url
+  test "should create category" do
+    assert_difference("Category.count") do
+      post categories_url, params: {category: {name: "New Category"}}
+    end
+
+    assert_redirected_to categories_url
+  end
+
+  test "should show category" do
+    get category_url(@category)
     assert_response :success
   end
 
   test "should get edit" do
-    get categories_edit_url
+    get edit_category_url(@category)
     assert_response :success
   end
 
-  test "should get update" do
-    get categories_update_url
-    assert_response :success
+  test "should update category" do
+    patch category_url(@category), params: {category: {name: "Updated Category"}}
+    assert_redirected_to categories_url
   end
 
-  test "should get destroy" do
-    get categories_destroy_url
-    assert_response :success
+  test "should destroy category" do
+    # Create a category that is not associated with any budget or transaction
+    category_to_destroy = Category.create!(name: "To be destroyed")
+    assert_difference("Category.count", -1) do
+      delete category_url(category_to_destroy)
+    end
+
+    assert_redirected_to categories_url
   end
 end