def setUp(self):
super(ElasticSearchManager, self).setUp()
self.event_connection = storage.get_connection(self.url, self.conf)
# prefix each test with unique index name
self.event_connection.index_name = 'events_%s' % uuid.uuid4().hex
# force index on write so data is queryable right away
self.event_connection._refresh_on_write = True
def setUp(self):
super(ElasticSearchManager, self).setUp()
self.connection = storage.get_connection(self.url, self.conf)
# prefix each test with unique index name
inx_uuid = uuidutils.generate_uuid(dashed=False)
self.connection.index_name = 'events_%s' % inx_uuid
# force index on write so data is queryable right away
self.connection._refresh_on_write = True
def setUp(self):
super(ElasticSearchManager, self).setUp()
self.connection = storage.get_connection(
self.url, self.conf)
# prefix each test with unique index name
inx_uuid = uuidutils.generate_uuid(dashed=False)
self.connection.index_name = 'events_%s' % inx_uuid
# force index on write so data is queryable right away
self.connection._refresh_on_write = True
def setUp(self):
super(MongoDbManager, self).setUp()
with warnings.catch_warnings():
warnings.filterwarnings(
action='ignore',
message='.*you must provide a username and password.*')
try:
self.connection = storage.get_connection(self.url, self.conf)
except storage.StorageBadVersion as e:
raise testcase.TestSkipped(six.text_type(e))
def setUp(self):
super(MongoDbManager, self).setUp()
with warnings.catch_warnings():
warnings.filterwarnings(
action='ignore',
message='.*you must provide a username and password.*')
try:
self.connection = storage.get_connection(self.url, self.conf)
except storage.StorageBadVersion as e:
raise testcase.TestSkipped(six.text_type(e))
def main(argv):
if os.getenv("PANKO_TEST_STORAGE_URL", "").startswith("hbase://"):
url = ("%s?table_prefix=%s" %
(os.getenv("PANKO_TEST_STORAGE_URL"),
os.getenv("PANKO_TEST_HBASE_TABLE_PREFIX", "test")))
event_conn = storage.get_connection(url, service.prepare_service())
for arg in argv:
if arg == "--upgrade":
event_conn.upgrade()
if arg == "--clear":
event_conn.clear()
def setUp(self):
super(HBaseManager, self).setUp()
self.connection = storage.get_connection(self.url, self.conf)
# Unique prefix for each test to keep data is distinguished because
# all test data is stored in one table
data_prefix = uuidutils.generate_uuid(dashed=False)
def table(conn, name):
return mocks.MockHBaseTable(name, conn, data_prefix)
# Mock only real HBase connection, MConnection "table" method
# stays origin.
mock.patch('happybase.Connection.table', new=table).start()
# We shouldn't delete data and tables after each test,
# because it last for too long.
# All tests tables will be deleted in setup-test-env.sh
mock.patch("happybase.Connection.disable_table",
new=mock.MagicMock()).start()
mock.patch("happybase.Connection.delete_table",
new=mock.MagicMock()).start()
mock.patch("happybase.Connection.create_table",
new=mock.MagicMock()).start()
def setUp(self):
super(HBaseManager, self).setUp()
self.connection = storage.get_connection(
self.url, self.conf)
# Unique prefix for each test to keep data is distinguished because
# all test data is stored in one table
data_prefix = uuidutils.generate_uuid(dashed=False)
def table(conn, name):
return mocks.MockHBaseTable(name, conn, data_prefix)
# Mock only real HBase connection, MConnection "table" method
# stays origin.
mock.patch('happybase.Connection.table', new=table).start()
# We shouldn't delete data and tables after each test,
# because it last for too long.
# All tests tables will be deleted in setup-test-env.sh
mock.patch("happybase.Connection.disable_table",
new=mock.MagicMock()).start()
mock.patch("happybase.Connection.delete_table",
new=mock.MagicMock()).start()
mock.patch("happybase.Connection.create_table",
new=mock.MagicMock()).start()
def test_get_connection_no_such_engine(self):
try:
storage.get_connection('no-such-engine://localhost', None)
except RuntimeError as err:
self.assertIn('no-such-engine', six.text_type(err))
def test_get_connection(self):
engine = storage.get_connection('log://localhost', None)
self.assertIsInstance(engine, impl_log.Connection)
def setUp(self):
super(SQLManager, self).setUp()
self.connection = storage.get_connection(self.url, self.conf)
def setUp(self):
super(SQLManager, self).setUp()
self.connection = storage.get_connection(self.url, self.conf)
def test_get_connection_no_such_engine(self):
try:
storage.get_connection('no-such-engine://localhost', None)
except RuntimeError as err:
self.assertIn('no-such-engine', six.text_type(err))
def test_get_connection(self):
engine = storage.get_connection('log://localhost', None)
self.assertIsInstance(engine, impl_log.Connection)