1 files changed, 77 insertions, 0 deletions
diff --git a/test_nightmare.py b/test_nightmare.py
new file mode 100755
index 0000000..24a6581
--- /dev/null
+++ b/test_nightmare.py
@@ -0,0 +1,77 @@
+#!/usr/bin/env python3
+
+import sys
+import unittest
+import cardbase
+from lxml import html
+
+
+class Test_cardInformationParsing(unittest.TestCase):
+    
+    @classmethod
+    def setUpClass(cls):
+        # fetch shivan dragon info by the card's collector number (281 in M15)
+        # cls.page = html.fromstring(requests.get("http://magiccards.info/m15/en/281.html").text)
+        
+        # but actually, use the pre-fetched file to avoid querying the server too much
+        with open("testcards/nightmare", "r") as file:
+            cls.page = html.fromstring(file.read())
+
+    # Tests
+    def test_correctTitleIsParsed(self):
+        self.assertEqual(cardbase.getTitle(self.page), "Nightmare")
+    
+    def test_correctCostIsParsed(self):
+        self.assertEqual(cardbase.getCost(self.page), "5B")
+        
+    def test_correctColourIsParsed(self):
+        self.assertEqual(cardbase.getColour(self.page), "B")
+        
+    def test_correctTypeIsParsed(self):
+        self.assertEqual(cardbase.getType(self.page), "Creature")
+        
+    def test_correctSubTypeIsParsed(self):
+        self.assertEqual(cardbase.getSubType(self.page), "Nightmare Horse")
+        
+    def test_correctArtistIsParsed(self):
+        self.assertEqual(cardbase.getArtist(self.page), "Vance Kovacs")
+        
+    def test_correctTextIsParsed(self):
+        self.assertEqual(cardbase.getText(self.page), ["Flying (This creature can't be blocked except by creatures with flying or reach.)", "Nightmare's power and toughness are each equal to the number of Swamps you control."])
+        
+    def test_correctFlavourIsParsed(self):
+        self.assertEqual(cardbase.getFlavour(self.page), "The thunder of its hooves beats dreams into despair.")
+        
+    def test_correctRarityIsParsed(self):
+        self.assertEqual(cardbase.getRarity(self.page), "Rare")
+        
+    def test_correctPowerIsParsed(self):
+        self.assertEqual(cardbase.getPower(self.page), "*")
+
+    def test_correctToughnessIsParsed(self):
+        self.assertEqual(cardbase.getToughness(self.page), "*")
+        
+    def test_correctLoyaltyIsParsed(self):
+        self.assertEqual(cardbase.getLoyalty(self.page), "")
+
+class Test_additionalCardData(unittest.TestCase):
+    
+    @classmethod
+    def setUpClass(cls):
+        cls.card = cardbase.fetchCard("m15", "276")
+    
+    def test_cardHasCorrectEdition(self):
+        self.assertEqual(self.card.edition, "m15")
+    
+    def test_cardHasCorrectScan(self):
+        self.assertEqual(self.card.scan, "http://magiccards.info/scans/en/m15/276.jpg")
+
+    def test_cardHasCorrectNumber(self):
+        self.assertEqual(self.card.number, "276")
+        
+def test():
+    unittest.main(exit=False)
+    
+# The entry point
+if __name__ == "__main__":
+    test()