Skip to content

Commit

Permalink
Added banners_test.py
Browse files Browse the repository at this point in the history
Added test file for banners module
  • Loading branch information
Santhosh-Siddhardha authored Jun 6, 2024
1 parent 00ef1d3 commit ad12eb9
Showing 1 changed file with 33 additions and 0 deletions.
33 changes: 33 additions & 0 deletions src/test/banners_test.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
import unittest
from src.scrape_up.banners import Scraper88x31


class TestScraper88x31(unittest.TestCase):

def setUp(self):
"""
Initialize a Scraper88x31 instance before each test method.
"""
self.scraper = Scraper88x31()

def test_get_all(self):
"""
| Methods | Details |
| ------------------ | -------------------------------------------------------- |
| `get_all()` | Returns the list of all available 88x31 banners |
"""
try:
banners = self.scraper.get_all()

# Check if banners is a list of URLs
self.assertIsInstance(banners, list)
for banner in banners:
self.assertIsInstance(banner, str)
self.assertTrue(banner.startswith("https://cyber.dabamos.de/88x31/"))
self.assertTrue(banner.endswith(".gif"))
except:
return None


if __name__ == "__main__":
unittest.main()

0 comments on commit ad12eb9

Please sign in to comment.