fix: prism card code with test [upload]

ShinyColorsWiki · Nov 19, 2023 · cfc9e57 · cfc9e57
1 parent 4a16bbc
commit cfc9e57
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/spec/PrismCardCode_spec.lua b/spec/PrismCardCode_spec.lua
@@ -13,6 +13,10 @@ describe("PrismCardCode", function()
     describe("Produce", function()
         local code = c("1328001")
 
+        it("Type check", function()
+            assert.is_equal(PrismCardCode.getTypeFromCode(code), "Produce")
+        end)
+
         it("Rarity check", function()
             assert.is_equal(PrismCardCode.getRarityFromCode(code), "3*")
         end)
@@ -29,6 +33,10 @@ describe("PrismCardCode", function()
     describe("Support", function()
         local code = c("2390001")
 
+        it("Type check", function()
+            assert.is_equal(PrismCardCode.getTypeFromCode(code), "Support")
+        end)
+
         it("Rarity check", function()
             assert.is_equal(PrismCardCode.getRarityFromCode(code), "SSR")
         end)

diff --git a/src/PrismCardCode.lua b/src/PrismCardCode.lua
@@ -121,16 +121,20 @@ function p.getInfoFromCode(code)
     return p.getSupportInfo(raw_data)
 end
 
-function p.getRarityFromCode(code)
+function p.getTypeFromCode(code)
     return p.getInfoFromCode(code)[1]
 end
 
-function p.getCharacterFromCode(code)
+function p.getRarityFromCode(code)
     return p.getInfoFromCode(code)[2]
 end
 
-function p.getCardNumberFromCode(code)
+function p.getCharacterFromCode(code)
     return p.getInfoFromCode(code)[3]
 end
 
+function p.getCardNumberFromCode(code)
+    return p.getInfoFromCode(code)[4]
+end
+
 return p