def tearDown(self):
from ddbmock.database.db import dynamodb
from ddbmock import clean_boto_patch
# Do a full database wipe
dynamodb.hard_reset()
# Remove the patch from Boto code (if any)
clean_boto_patch()
def tearDown(self):
from ddbmock.database.db import dynamodb
from ddbmock import clean_boto_patch
# Do a full database wipe
dynamodb.hard_reset()
# Remove the patch from Boto code (if any)
clean_boto_patch()
def test_connect_boto_patch_network(self):
from ddbmock import connect_boto_network, clean_boto_patch
clean_boto_patch()
db = connect_boto_network(HOST, PORT)
self.assertEqual('ddbmock', db.layer1.region.name)
self.assertEqual(ENDPOINT, db.layer1.region.endpoint)
self.assertEqual(ENDPOINT, db.layer1.host)
self.assertEqual(PORT, db.layer1.port)
self.assertFalse(db.layer1.is_secure)
def test_connect_boto_patch_network(self):
try:
socket.gethostbyname(HOST)
except Exception:
self.skipTest("Can't connect to test host: %s" % HOST)
from ddbmock import connect_boto_network, clean_boto_patch
clean_boto_patch()
db = connect_boto_network(HOST, PORT)
self.assertEqual('ddbmock', db.layer1.region.name)
self.assertEqual(ENDPOINT, db.layer1.region.endpoint)
self.assertEqual(ENDPOINT, db.layer1.host)
self.assertEqual(PORT, db.layer1.port)
self.assertFalse(db.layer1.is_secure)
def tearDown(self):
from ddbmock.database.db import dynamodb
from ddbmock import clean_boto_patch
dynamodb.hard_reset()
clean_boto_patch()
import sys
# Just, skip this unless "--no-skip"
if '--no-skip' in sys.argv:
import os, imp, boto, unittest
from ddbmock import connect_boto_patch, clean_boto_patch
# Load boto dir
boto_dir = os.path.realpath(boto.__file__+'/../../')
# Plug patch
connect_boto_patch()
# Import boto test modules, pretending they are belonging to us. see hack in tests.unit.__init__
layer1_file, _, desc1 = imp.find_module('test_layer1', [boto_dir+'/tests/integration/dynamodb'])
layer2_file, _, desc2 = imp.find_module('test_layer2', [boto_dir+'/tests/integration/dynamodb'])
layer1_module = imp.load_module('test_layer1', layer1_file, boto_dir, desc1)
layer2_module = imp.load_module('test_layer2', layer2_file, boto_dir, desc2)
# Finally, import the tests class, that's enough to let the magic happen !
DynamoDBLayer1Test = layer1_module.DynamoDBLayer1Test
DynamoDBLayer2Test = layer2_module.DynamoDBLayer2Test
# clean up
layer1_file.close()
layer2_file.close()
clean_boto_patch()
else:
print "Boto Integration tests are slow hence disabled by default, enable them with '--no-skip'"
def tearDown(self):
from ddbmock import clean_boto_patch
clean_boto_patch()
def tearDown(self):
from ddbmock import clean_boto_patch
clean_boto_patch()