def setUp(self):
self.clear_cache()
try:
self.hbase_tables_created = True
for hbase_model_class in HBaseModel.__subclasses__():
hbase_model_class.create_table()
except Exception:
self.tearDown()
raise
def get_model_class(cls, model_class_name):
for subclass in HBaseModel.__subclasses__():
if subclass.__name__ == model_class_name:
return subclass
raise Exception('HBaseModel {} not found'.format(model_class_name))
def tearDown(self):
if not self.hbase_tables_created:
return
for hbase_model_class in HBaseModel.__subclasses__():
hbase_model_class.drop_table()
