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)