blob: eb7cebbf3e267c2a12f9056fa40a2e51661ef903 [file] [log] [blame]
import json
import os
class Suite(object):
def __init__(self, suitename):
filename = "{0}.suite".format(suitename)
filepath = os.path.join(os.path.dirname(__file__), filename)
with open(filepath) as suitefile:
_suite = json.load(suitefile)
self.urls = _suite['urls']
self.name = suitename
self.attempts = 2
self.max_attempts = 2
@classmethod
def get_available_suites(cls):
suitesdir = os.path.dirname(__file__)
suiteslist = [os.path.splitext(f)[0].lower() for f in os.listdir(suitesdir) if f.endswith('.suite')]
if not suiteslist:
raise Exception('Cant find any .suite files in directory %s' % suitesdir)
return suiteslist