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'))
