Fix FullDPS stat fallback

Author: LocalIdentity

spec/System/TestTradeQuery_spec.lua

@@ -1,8 +1,10 @@
 describe("TradeQuery", function ()
 	local mock_tradeQuery
+	local mock_queryGen
 
 	before_each(function()
 		mock_tradeQuery = new("TradeQuery", { itemsTab = {} })
+		mock_queryGen = new("TradeQueryGenerator", { itemsTab = {} })
 	end)
 
 	describe("ReduceOutput", function()
@@ -21,5 +23,24 @@ describe("TradeQuery", function ()
 			assert.are.equals(260, result.AverageDamage)
 			assert.is_nil(result.Life)
 		end)
+
+		it("keeps fallback DPS stats when FullDPS is selected but not present", function()
+			mock_tradeQuery.statSortSelectionList = { { stat = "FullDPS", weightMult = 1 } }
+
+			local baseOutput = {
+				CombinedDPS = 100,
+				TotalDPS = 100,
+				TotalDotDPS = 0,
+			}
+			local reducedOutput = mock_tradeQuery:ReduceOutput({
+				CombinedDPS = 120,
+				TotalDPS = 120,
+				TotalDotDPS = 0,
+			})
+
+			local result = mock_queryGen.WeightedRatioOutputs(baseOutput, reducedOutput, mock_tradeQuery.statSortSelectionList)
+
+			assert.are.equals(1.2, result)
+		end)
 	end)
 end)

src/Classes/TradeQuery.lua

@@ -780,6 +780,11 @@ function TradeQueryClass:ReduceOutput(output)
 	local smallOutput = {}
 	for _, statTable in ipairs(self.statSortSelectionList) do
 		smallOutput[statTable.stat] = data.powerStatList.GetFromOutput(output, statTable)
+		if statTable.stat == "FullDPS" and not output.FullDPS then
+			smallOutput.TotalDPS = data.powerStatList.GetFromOutput(output, { stat = "TotalDPS" })
+			smallOutput.TotalDotDPS = data.powerStatList.GetFromOutput(output, { stat = "TotalDotDPS" })
+			smallOutput.CombinedDPS = data.powerStatList.GetFromOutput(output, { stat = "CombinedDPS" })
+		end
 	end
 	return smallOutput
 end