openstack/deb-python-cassandra-driver
Code Issues Proposed changes

Reverted test base to original

Browse Source
This commit is contained in:
timmartin19
2014-10-16 12:19:54 -04:00
parent 6e4dc3afdf
commit de18d56296
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
cqlengine/tests/base.py
View File

@@ -2,9 +2,9 @@ from unittest import TestCase
import os
import sys
import six
from cqlengine import connection
from cqlengine.connection import get_session
CASSANDRA_VERSION = 20 #int(os.environ['CASSANDRA_VERSION'])
CASSANDRA_VERSION = int(os.environ['CASSANDRA_VERSION'])
class BaseCassEngTestCase(TestCase):
@@ -13,11 +13,9 @@ class BaseCassEngTestCase(TestCase):
# super(BaseCassEngTestCase, cls).setUpClass()
session = None
@classmethod
def setUpClass(cls):
connection.setup(['192.168.56.103'], 'test')
cls.session = connection.get_session()
super(BaseCassEngTestCase, cls).setUpClass()
def setUp(self):
self.session = get_session()
super(BaseCassEngTestCase, self).setUp()
def assertHasAttr(self, obj, attr):
self.assertTrue(hasattr(obj, attr),