Skip to content

Commit 22d24c3

Browse files
committed
Update generated content when using "seleniumbase mkdir [DIR_NAME]"
1 parent bf7f553 commit 22d24c3

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

console_scripts/sb_mkdir.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,8 @@ def main():
180180
data.append("class BingTests(BaseCase):")
181181
data.append("")
182182
data.append(" def test_bing(self):")
183-
data.append(" self.open('https://www.bing.com/')")
183+
data.append(" self.open('https://bing.com')")
184+
data.append(" self.assert_text('Bing', Page.logo_box)")
184185
data.append(" self.update_text(Page.search_box, 'github')")
185186
data.append(" self.assert_element('li[query=\"github\"]')")
186187
data.append(" self.click(Page.search_button)")
@@ -196,6 +197,7 @@ def main():
196197

197198
data = []
198199
data.append("class Page(object):")
200+
data.append(" logo_box = '#sbox div[class*=logo]'")
199201
data.append(" search_box = 'input.b_searchbox'")
200202
data.append(" search_button = 'input[name=\"go\"]'")
201203
data.append(" search_results = '#b_results'")
@@ -213,7 +215,15 @@ def main():
213215
data.append("class GoogleTests(BaseCase):")
214216
data.append("")
215217
data.append(" def test_google_dot_com(self):")
216-
data.append(" self.open('https://www.google.com')")
218+
data.append(" self.open('https://google.com')")
219+
data.append(" try:")
220+
data.append(" # Remove the Privacy Checkup box if present.")
221+
data.append(" self.assert_text('Privacy Checkup', "
222+
"HomePage.dialog_box, timeout=3)")
223+
data.append(" self.click('link=NO, THANKS')")
224+
data.append(" except Exception:")
225+
data.append(" # Google may have removed it. Continue test.")
226+
data.append(" pass")
217227
data.append(" self.assert_element(HomePage.search_button)")
218228
data.append(
219229
" self.assert_element(HomePage.feeling_lucky_button)")
@@ -234,6 +244,7 @@ def main():
234244

235245
data = []
236246
data.append("class HomePage(object):")
247+
data.append(" dialog_box = '[role=\"dialog\"] div'")
237248
data.append(" search_box = 'input[title=\"Search\"]'")
238249
data.append(" search_button = 'input[value=\"Google Search\"]'")
239250
data.append(

0 commit comments

Comments
 (0)