class TestCompanyModel(unittest.TestCase):
def setUp(self):
self.company = CompanyModel()
def test_add_department(self):
self.company.add_user('Nisse', 'Donald Duck')
assert True
def test_get_head_count_for(self):
#self.company.add_user('Nisse', 'Donald Duck')
self.company.get_headcount_for('Donald Duck')
assert True
def step_impl(context):
model = getattr(context, "model", None)
if not model:
context.model = CompanyModel()
for row in context.table:
context.model.add_user(name=row["name"], department=row["department"])
assert True
def setUp(self):
self.company = CompanyModel()