def setUp(self): super().setUp() with self.connection as conn: conn.execute( "CREATE TABLE test (id INT NOT NULL, num INT, PRIMARY KEY (id));" ) conn.execute("insert into test(id, num) values(1, 1);") conn.execute("insert into test(id, num) values(2, 2);") ensure_empty(join(test.get_test_dir(self.test_name), 'resources'))
def setUp(self): super().setUp() ensure_empty(join(test.get_test_dir(self.test_name), 'resources')) import pika with pika.BlockingConnection(self.connectionParameters) as connection: channel = connection.channel() channel.exchange_declare(exchange=self.config['exchange'], exchange_type='topic') channel.queue_declare(queue=self.config['queue']) channel.queue_bind(exchange=self.config['exchange'], queue=self.config['queue'], routing_key=self.config['routingKey'])
def setUp(self): super().setUp() print(self.conf) with self.connection as conn: cur = conn.cursor() cur.execute( "CREATE TABLE test (id serial PRIMARY KEY, num integer);") cur.execute("insert into test(id, num) values(1, 1);") cur.execute("insert into test(id, num) values(2, 2);") conn.commit() cur.close() ensure_empty(join(test.get_test_dir(self.test_name), 'resources'))
def setUp(self): super().setUp() file_utils.ensure_empty( join(test.get_test_dir(self.test_name), 'private_resource')) self.copy_resource('calculator.py') self.copy_resource('calculator.proto') self.copy_resource('greeter.py') self.copy_resource('greeter.proto') from grpc.tools import command command.build_package_protos(join(test.get_test_dir(self.test_name), 'private_resource'), strict_mode=True)
def setUp(self): super().setUp() conn = pymysql.connect(**self.conf) cur = conn.cursor() cur.execute( "CREATE TABLE test (id INT NOT NULL, num INT, PRIMARY KEY (id));") cur.execute("insert into test(id, num) values(1, 1);") cur.execute("insert into test(id, num) values(2, 2);") conn.commit() cur.close() conn.close() ensure_empty(join(test.get_test_dir(self.test_name), 'resources'))
def setUp(self): ensure_empty(test.get_test_dir(self.test_name))
def setUp(self): Singleton._instances = {} # clean singleton between tests ensure_empty(test.get_test_dir(self.test_name)) ensure_empty(join(test.get_test_dir(self.test_name), 'resources'))
def setUp(self): super().setUp() ensure_empty(join(test.get_test_dir(self.test_name), 'resources'))
def setUp(self): super().setUp() ensure_empty(join(test.get_test_dir(self.test_name), 'resources')) self.populate_file( 'resources/test.py', '''from selenium import webdriver from selenium.webdriver.common.keys import Keys import os driver = webdriver.Firefox() try: driver.get(os.environ['site_url']) assert "Python" in driver.title elem = driver.find_element_by_name("q") elem.clear() elem.send_keys("pycon") elem.send_keys(Keys.RETURN) assert "No results found." not in driver.page_source finally: driver.close()''') self.populate_file( 'resources/google_search.js', '''const {Builder, By, Key, until} = require('selenium-webdriver'); async function basicExample(){ let driver = await new Builder().forBrowser('firefox').build(); try{ await driver.get(process.env.site_url); await driver.findElement(By.name('q')).sendKeys('webdriver', Key.RETURN); await driver.wait(until.titleContains('webdriver'), 1000); await driver.getTitle().then(function(title) { console.log('{\"title\":\"' + title + '\"}') }); driver.quit(); } catch(err) { console.error(err); process.exitCode = 1; driver.quit(); } } basicExample(); ''') self.populate_file( 'resources/MySeleniumTest.java', '''package selenium; import org.openqa.selenium.By; import org.openqa.selenium.WebDriver; import org.openqa.selenium.WebElement; import org.openqa.selenium.firefox.FirefoxDriver; public class MySeleniumTest { public static void main(String[] args) { WebDriver driver = new FirefoxDriver(); try { driver.get(System.getenv("site_url")); WebElement element = driver.findElement(By.name("q")); element.sendKeys("Cheese!"); element.submit(); } finally { driver.quit(); } } } ''')
def setUp(self): super().setUp() ensure_empty(join(test.get_test_dir(self.test_name), 'resources')) self.add_dag('airflow_hello_world.py')
def setUp(self): super().setUp() file_utils.ensure_empty(join(os.getcwd(), TEST_DIR, 'steps'))