diff --git a/app/assets/javascripts/admin/services/variant_unit_manager.js.coffee b/app/assets/javascripts/admin/services/variant_unit_manager.js.coffee index dbdac44808..eec6130bad 100644 --- a/app/assets/javascripts/admin/services/variant_unit_manager.js.coffee +++ b/app/assets/javascripts/admin/services/variant_unit_manager.js.coffee @@ -35,7 +35,4 @@ angular.module("admin.products").factory "VariantUnitManager", -> @unitNames[unitType][scale] @unitScales: (unitType) -> - unitScales = - 'weight': [1.0, 1000.0, 1000000.0] - 'volume': [0.001, 1.0, 1000.0] - unitScales[unitType] + (parseFloat(scale) for scale in Object.keys(@unitNames[unitType])).sort() diff --git a/spec/javascripts/unit/admin/services/variant_unit_manager_spec.js.coffee b/spec/javascripts/unit/admin/services/variant_unit_manager_spec.js.coffee index 7f9ced4be0..21830cae65 100644 --- a/spec/javascripts/unit/admin/services/variant_unit_manager_spec.js.coffee +++ b/spec/javascripts/unit/admin/services/variant_unit_manager_spec.js.coffee @@ -26,3 +26,10 @@ describe "VariantUnitManager", -> expect(VariantUnitManager.getUnitName(0.001, "volume")).toEqual "mL" expect(VariantUnitManager.getUnitName(1, "volume")).toEqual "L" expect(VariantUnitManager.getUnitName(1000, "volume")).toEqual "kL" + + describe "unitScales", -> + it "returns a set of scales for unit type weight", -> + expect(VariantUnitManager.unitScales('weight')).toEqual [1.0, 1000.0, 1000000.0] + + it "returns a set of scales for unit type volume", -> + expect(VariantUnitManager.unitScales('volume')).toEqual [0.001, 1.0, 1000.0]