def test_add_shareholder_with_existing_number(self):
"""
add shareholder with existing number
"""
op = OperatorGenerator().generate()
self.add_subscription(op.company)
user = UserGenerator().generate()
shareholder = ShareholderGenerator().generate(company=op.company)
user.email = None
try:
start = page.StartPage(self.selenium, self.live_server_url,
op.user)
# wait for list
start.wait_until_visible((By.CSS_SELECTOR, '#shareholder_list'))
start.is_properly_displayed()
self.assertEqual(
start.has_shareholder_count(),
Shareholder.objects.filter(company=op.company).count())
start.click_open_add_shareholder()
start.wait_until_modal_opened('addShareholder')
start.add_shareholder(user, number=shareholder.number)
start.click_save_add_shareholder()
start.close_modal('addShareholder')
time.sleep(2)
self.assertEqual(
Shareholder.objects.filter(company=op.company,
number=shareholder.number).count(),
1)
except Exception, e:
self._handle_exception(e)
def test_add_shareholder_without_email(self):
"""
add shareholder without email
"""
op = OperatorGenerator().generate()
self.add_subscription(op.company)
CompanyShareholderGenerator().generate(company=op.company)
user = UserGenerator().generate()
user.email = None
try:
start = page.StartPage(self.selenium, self.live_server_url,
op.user)
# wait for list
start.wait_until_visible((By.CSS_SELECTOR, '#shareholder_list'))
start.is_properly_displayed()
self.assertEqual(
start.has_shareholder_count(),
Shareholder.objects.filter(company=op.company).count())
start.click_open_add_shareholder()
start.wait_until_modal_opened('addShareholder')
start.add_shareholder(user)
start.click_save_add_shareholder()
start.close_modal('addShareholder')
time.sleep(5)
# wait for list entry
xpath = ('//div[@id="shareholder_list"]//span[text()="{}"]'
u''.format(Shareholder.objects.last().get_full_name()))
start.wait_until_visible((By.XPATH, xpath))
self.assertEqual(
start.has_shareholder_count(),
Shareholder.objects.filter(company=op.company).count())
self.assertEqual(
user.first_name,
Shareholder.objects.filter(
company=op.company).last().user.first_name)
except Exception, e:
self._handle_exception(e)
