From 8e2f6625524b0f7ac83eb2549be70644bfbedde2 Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Sat, 4 Apr 2026 18:48:46 +0800
Subject: [PATCH] bugfix

---
 pricing/init.py    | 2 ++
 pricing/pricing.py | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/pricing/init.py b/pricing/init.py
index 3580fd9..b68f2bc 100644
--- a/pricing/init.py
+++ b/pricing/init.py
@@ -1,5 +1,6 @@
 from pricing.pricing import (
 	PricingProgram,
+	test_pricing,
 	get_pricing_program,
 	get_pricing_program_timing
 )
@@ -13,3 +14,4 @@ def load_pricing():
 	env.pricing_program_charging = PricingProgram.charging
 	env.load_pricing_data = PricingProgram.load_pricing_data
 	env.get_pricing_program = PricingProgram.get_pricing_program
+	env.test_pricing = test_pricing
diff --git a/pricing/pricing.py b/pricing/pricing.py
index bcde8ce..572679a 100644
--- a/pricing/pricing.py
+++ b/pricing/pricing.py
@@ -414,6 +414,8 @@ async def get_pricing_program_timeing(pptid):
 async def test_pricing(pptid, data):
 	ppt = get_pricing_program_timeing(pptid)
 	prices = PricingProgram.get_pricing_from_ymalstr(data, ppt.pricing_data)
+	if prices is None:
+		return None
 	amount = 0
 	for p in prices:
 		amount += p.amount