test: add tests to get capacity colours

lucasmonstrox · Feb 6, 2024 · bcff856 · bcff856
1 parent 79d877b
commit bcff856
Show file tree

Hide file tree

Showing 7 changed files with 24 additions and 4 deletions.
diff --git a/...getCapacity/images/capacityWith318218.png → ...images/capacityWith318218AndGrayColor.png b/...getCapacity/images/capacityWith318218.png → ...images/capacityWith318218AndGrayColor.png
diff --git a/tests/skills/getCapacity/images/capacityWith318218AndGreenColor.png b/tests/skills/getCapacity/images/capacityWith318218AndGreenColor.png
diff --git a/tests/skills/getCapacity/images/capacityWith318218AndOrangeColor.png b/tests/skills/getCapacity/images/capacityWith318218AndOrangeColor.png
diff --git a/...ls/getCapacity/images/capacityWith327.png → ...ty/images/capacityWith327AndGrayColor.png b/...ls/getCapacity/images/capacityWith327.png → ...ty/images/capacityWith327AndGrayColor.png
diff --git a/tests/skills/getCapacity/images/capacityWith327AndGreenColor.png b/tests/skills/getCapacity/images/capacityWith327AndGreenColor.png
diff --git a/tests/skills/getCapacity/images/capacityWith327AndOrangeColor.png b/tests/skills/getCapacity/images/capacityWith327AndOrangeColor.png
diff --git a/tests/skills/getCapacity/test_getCapacity.py b/tests/skills/getCapacity/test_getCapacity.py
@@ -21,11 +21,31 @@ def test_should_return_None_when_capacity_label_is_not_visible():
     assert skills.getCapacity() is None
 
 
-def test_should_return_327():
-    skills = loadSkillsByImage('capacityWith327.png')
+def test_should_return_327_when_gray_color():
+    skills = loadSkillsByImage('capacityWith327AndGrayColor.png')
     assert skills.getCapacity() == 327
 
 
-def test_should_return_318_218():
-    skills = loadSkillsByImage('capacityWith318218.png')
+def test_should_return_327_when_green_color():
+    skills = loadSkillsByImage('capacityWith327AndGreenColor.png')
+    assert skills.getCapacity() == 327
+
+
+def test_should_return_327_when_orange_color():
+    skills = loadSkillsByImage('capacityWith327AndOrangeColor.png')
+    assert skills.getCapacity() == 327
+
+
+def test_should_return_318_218_when_gray_color():
+    skills = loadSkillsByImage('capacityWith318218AndGrayColor.png')
+    assert skills.getCapacity() == 318_218
+
+
+def test_should_return_318_218_when_green_color():
+    skills = loadSkillsByImage('capacityWith318218AndGreenColor.png')
+    assert skills.getCapacity() == 318_218
+
+
+def test_should_return_318_218_when_orange_color():
+    skills = loadSkillsByImage('capacityWith318218AndOrangeColor.png')
     assert skills.getCapacity() == 318_218