from views.web_views.base_web_view import BaseWebView, BaseButton class AssetsButton(BaseButton): def __init__(self, driver): super(AssetsButton, self).__init__(driver) self.locator = self.Locator.webview_selector('Assets') class RequestETHButton(BaseButton): def __init__(self, driver): super(AssetsButton.RequestETHButton, self).__init__(driver) self.locator = self.Locator.webview_selector('Request Ropsten ETH') class RequestSTTButton(BaseButton): def __init__(self, driver): super(AssetsButton.RequestSTTButton, self).__init__(driver) self.locator = self.Locator.webview_selector('Request STT') def navigate(self): from views.send_transaction_view import SendTransactionView return SendTransactionView(self.driver) class TransactionsButton(BaseButton): def __init__(self, driver): super(TransactionsButton, self).__init__(driver) self.locator = self.Locator.text_selector('Transactions') class SignMessageButton(BaseButton): def __init__(self, driver): super(TransactionsButton.SignMessageButton, self).__init__(driver) self.locator = self.Locator.text_selector('Sign message') def navigate(self): from views.send_transaction_view import SendTransactionView return SendTransactionView(self.driver) class DeployContractButton(BaseButton): def __init__(self, driver): super(TransactionsButton.DeployContractButton, self).__init__(driver) self.locator = self.Locator.text_selector('Deploy simple contract') def navigate(self): from views.send_transaction_view import SendTransactionView return SendTransactionView(self.driver) class SendTwoTxOneByOneButton(BaseButton): def __init__(self, driver): super(TransactionsButton.SendTwoTxOneByOneButton, self).__init__(driver) self.locator = self.Locator.text_selector('Send two Txs, one after another, 0.00001 and 0.00002 ETH') def navigate(self): from views.send_transaction_view import SendTransactionView return SendTransactionView(self.driver) class SendTwoTxInBatchButton(BaseButton): def __init__(self, driver): super(TransactionsButton.SendTwoTxInBatchButton, self).__init__(driver) self.locator = self.Locator.text_selector('Send two Txs in batch, 0.00001 and 0.00002 ETH') def navigate(self): from views.send_transaction_view import SendTransactionView return SendTransactionView(self.driver) class TestFiltersButton(BaseButton): def __init__(self, driver): super(TransactionsButton.TestFiltersButton, self).__init__(driver) self.locator = self.Locator.text_selector('Test filters') class StatusAPIButton(BaseButton): def __init__(self, driver): super(StatusAPIButton, self).__init__(driver) self.locator = self.Locator.text_selector('Status API') class RequestContactCodeButton(BaseButton): def __init__(self, driver): super(StatusAPIButton.RequestContactCodeButton, self).__init__(driver) self.locator = self.Locator.text_part_selector('Request contact code') class SendOneTransactionInBatchButton(BaseButton): def __init__(self, driver): super().__init__(driver) self.locator = self.Locator.text_selector('Send one Tx in batch') def navigate(self): from views.send_transaction_view import SendTransactionView return SendTransactionView(self.driver) class StatusTestDAppView(BaseWebView): def __init__(self, driver): super(StatusTestDAppView, self).__init__(driver) self.driver = driver self.assets_button = AssetsButton(self.driver) self.request_eth_button = AssetsButton.RequestETHButton(self.driver) self.request_stt_button = AssetsButton.RequestSTTButton(self.driver) self.transactions_button = TransactionsButton(self.driver) self.sign_message_button = TransactionsButton.SignMessageButton(self.driver) self.deploy_contract_button = TransactionsButton.DeployContractButton(self.driver) self.send_one_tx_in_batch_button = SendOneTransactionInBatchButton(self.driver) self.send_two_tx_one_by_one_button = TransactionsButton.SendTwoTxOneByOneButton(self.driver) self.send_two_tx_in_batch_button = TransactionsButton.SendTwoTxInBatchButton(self.driver) self.test_filters_button = TransactionsButton.TestFiltersButton(self.driver) self.status_api_button = StatusAPIButton(self.driver) self.request_contact_code_button = StatusAPIButton.RequestContactCodeButton(self.driver) def wait_for_d_aap_to_load(self, wait_time=10): self.assets_button.wait_for_visibility_of_element(seconds=wait_time) def faucet_asset(self, asset='eth'): self.wait_for_d_aap_to_load() self.assets_button.click() if asset == 'eth': self.request_eth_button.click() self.element_by_text('Faucet request recieved').wait_for_visibility_of_element() self.ok_button.click() self.element_by_text('Faucet request recieved').wait_for_invisibility_of_element() elif asset == 'stt': send_transaction_view = self.request_stt_button.click() send_transaction_view.sign_transaction()