mirror of
https://github.com/we-promise/sure.git
synced 2026-04-07 14:31:25 +00:00
e1ff6d46ee
* Make categories global This solves us A LOT of cash flow and budgeting problems. * Update schema.rb * Update auto_categorizer.rb * Update income_statement.rb * FIX budget sub-categories * FIX sub-categories and tests * Add 2 step migration
96 lines
3.0 KiB
Ruby
96 lines
3.0 KiB
Ruby
require "test_helper"
|
|
|
|
class CategoryImportTest < ActiveSupport::TestCase
|
|
setup do
|
|
@family = families(:dylan_family)
|
|
@csv = <<~CSV
|
|
name,color,parent_category,icon
|
|
Food & Drink,#f97316,,carrot
|
|
Groceries,#407706,Food & Drink,shopping-basket
|
|
Salary,#22c55e,,briefcase
|
|
CSV
|
|
end
|
|
|
|
test "imports categories from Sure export" do
|
|
import = @family.imports.create!(type: "CategoryImport", raw_file_str: @csv, col_sep: ",")
|
|
import.generate_rows_from_csv
|
|
assert_equal 3, import.rows.count
|
|
|
|
tracked_categories = Category.where(family: @family, name: [ "Food & Drink", "Groceries", "Salary" ])
|
|
|
|
assert_difference -> { tracked_categories.count }, 2 do
|
|
import.send(:import!)
|
|
end
|
|
|
|
food = Category.find_by!(family: @family, name: "Food & Drink")
|
|
groceries = Category.find_by!(family: @family, name: "Groceries")
|
|
salary = Category.find_by!(family: @family, name: "Salary")
|
|
|
|
assert_equal "carrot", food.lucide_icon
|
|
assert_equal food, groceries.parent
|
|
assert_equal "shopping-basket", groceries.lucide_icon
|
|
assert_equal "briefcase", salary.lucide_icon
|
|
end
|
|
|
|
test "imports subcategories even when parent row comes later" do
|
|
csv = <<~CSV
|
|
name,color,parent_category,icon
|
|
Utilities,#407706,Household,plug
|
|
Household,#f97316,,house
|
|
CSV
|
|
|
|
import = @family.imports.create!(type: "CategoryImport", raw_file_str: csv, col_sep: ",")
|
|
import.generate_rows_from_csv
|
|
|
|
import.send(:import!)
|
|
|
|
household = Category.find_by!(family: @family, name: "Household")
|
|
utilities = Category.find_by!(family: @family, name: "Utilities")
|
|
|
|
assert_equal household, utilities.parent
|
|
assert_equal "#f97316", household.color
|
|
end
|
|
|
|
test "updates categories when duplicate rows are provided" do
|
|
csv = <<~CSV
|
|
name,color,parent_category,icon
|
|
Snacks,#aaaaaa,,cookie
|
|
Snacks,#bbbbbb,,pizza
|
|
CSV
|
|
|
|
import = @family.imports.create!(type: "CategoryImport", raw_file_str: csv, col_sep: ",")
|
|
import.generate_rows_from_csv
|
|
|
|
import.send(:import!)
|
|
|
|
snacks = Category.find_by!(family: @family, name: "Snacks")
|
|
assert_equal "#bbbbbb", snacks.color
|
|
assert_equal "pizza", snacks.lucide_icon
|
|
end
|
|
|
|
test "accepts required headers with an asterisk suffix" do
|
|
csv = <<~CSV
|
|
name*,color,parent_category,icon
|
|
Food & Drink,#f97316,,carrot
|
|
CSV
|
|
|
|
import = @family.imports.create!(type: "CategoryImport", raw_file_str: csv, col_sep: ",")
|
|
import.generate_rows_from_csv
|
|
|
|
assert_equal 1, import.rows.count
|
|
assert_equal "Food & Drink", import.rows.first.name
|
|
end
|
|
|
|
test "fails fast when required headers are missing" do
|
|
csv = <<~CSV
|
|
title,color,parent_category,icon
|
|
Food & Drink,#f97316,,carrot
|
|
CSV
|
|
|
|
import = @family.imports.create!(type: "CategoryImport", raw_file_str: csv, col_sep: ",")
|
|
|
|
error = assert_raises(ActiveRecord::RecordInvalid) { import.generate_rows_from_csv }
|
|
assert_includes error.message, "Missing required columns: name"
|
|
end
|
|
end
|