def test_getSnapshotPath(self):
tm = TMSaveCache({}, conf_filename=self.conf_filename)
tm._getRemoteFileContent = Mock(return_value="""[default]
volume_name = /foo/bar
""")
tm.diskName = 'baz'
assert '/foo/bar/baz' == tm._getSnapshotPath()
def test_updatePDiskIpForNewImageUri(self):
tm = TMSaveCache({}, conf_filename=self.conf_filename)
tm.persistentDiskPublicBaseUrl = 'https://1.2.3.4'
assert 'pdisk:1.2.3.4:987:uuid' == tm._updatePDiskIpForNewImageUri('pdisk:0.0.0.0:987:uuid')
tm.persistentDiskPublicBaseUrl = 'https://1.2.3.4:123'
assert 'pdisk:1.2.3.4:123:uuid' == tm._updatePDiskIpForNewImageUri('pdisk:0.0.0.0:987:uuid')
def testParseArgs(self):
tm = TMSaveCache({TMSaveCache._ARG_SRC_POS: 'foo'},
conf_filename=self.conf_filename)
self.failUnlessRaises(ValueError, tm._parseArgs)
tm = TMSaveCache({TMSaveCache._ARG_SRC_POS: 'foo:bar'},
conf_filename=self.conf_filename)
tm._parseArgs()
self.assertEqual(tm.diskSrcHost, 'foo')
self.assertEqual(tm.diskSrcPath, 'bar')
def testResetImageValidity(self):
tm = TMSaveCache({}, conf_filename=self.conf_filename)
assert 2 == tm._P12_VALIDITY
assert 2 * 24 * 3600 == tm._IMAGE_VALIDITY
tm._update_endroser_cert_and_image_validity_periods()
assert self.IMAGE_VALIDITY_DAYS == tm._P12_VALIDITY
assert self.IMAGE_VALIDITY_DAYS * 24 * 3600 == tm._IMAGE_VALIDITY
def testRetrieveInstanceId(self):
tm = TMSaveCache({},
conf_filename=self.conf_filename)
for path in ['foo', '/a/b', '/1/2']:
tm.diskSrcPath = path
self.failUnlessRaises(ValueError, tm._retrieveInstanceId)
for path in ['/a/1', '/a/1/b']:
tm.diskSrcPath = path
tm._retrieveInstanceId()
self.failUnlessEqual(tm.instanceId, 1)
def xtestSendEmail_Live(self):
"""Remove 'x' from the test name, set correct values for
email_address and smtp_host, and run the test manually.
You should receive email."""
email_address = '<*****@*****.**>'
smtp_host = '<SMTP host>'
tm = TMSaveCache({}, conf_filename=self.conf_filename)
tm.snapshotMarketplaceId = 'ABC'
tm.createImageInfo = {}
tm.createImageInfo['creatorEmail'] = email_address
tm.manifestNotSignedPath = self.conf_filename
tm.configHolder.set('smtp_host', smtp_host)
tm._sendEmailToUser()
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
import os
import sys
sys.path.append('/var/lib/stratuslab/python')
from stratuslab.tm.TMSaveCache import TMSaveCache
if __name__ == '__main__':
try:
tm = TMSaveCache(sys.argv)
tm.run()
except Exception, e:
print >> sys.stderr, 'ERROR MESSAGE --8<------'
print >> sys.stderr, '%s: %s' % (os.path.basename(__file__), e)
print >> sys.stderr, 'ERROR MESSAGE ------>8--'
if TMSaveCache.PRINT_TRACE_ON_ERROR:
raise
sys.exit(1)
def test_getSnapshotPathConfParamNotInFile(self):
tm = TMSaveCache({}, conf_filename=self.conf_filename)
tm._getRemoteFileContent = Mock(return_value="""[default]
foo = bar
""")
self.failUnlessRaises(ConfigurationException, tm._getSnapshotPath)
def testGenerateManifest(self):
manifest_info = ManifestInfo()
manifest_info.parseManifest(self.TEST_MANIFEST)
ManifestDownloader.getManifestInfo = Mock(return_value=manifest_info)
PDISK_ENDPOINT = 'pdisk:0.0.0.0:8445'
TMSaveCache._getAttachedVolumeURIs = Mock(
return_value=[PDISK_ENDPOINT + ':48ac4190-9a11-4a06-8bef-03fd97080eba'])
tm = TMSaveCache({TMSaveCache._ARG_SRC_POS: 'foo:/bar/1'},
conf_filename=self.conf_filename)
tm._parseArgs()
tm._retrieveAttachedVolumeInfo()
tm.createImageInfo = {VmManager.CREATE_IMAGE_KEY_CREATOR_EMAIL: '*****@*****.**',
VmManager.CREATE_IMAGE_KEY_CREATOR_NAME: 'Jay Random',
VmManager.CREATE_IMAGE_KEY_NEWIMAGE_COMMENT: 'test',
VmManager.CREATE_IMAGE_KEY_NEWIMAGE_VERSION: '0.0',
VmManager.CREATE_IMAGE_KEY_NEWIMAGE_MARKETPLACE: 'http://new.markeplace.org'}
tm.imageSha1 = 'ea7d0ddf7af4e2ea431db89639feb7036fb23062'
tm.createdPDiskId = 'foo-bar-baz'
try:
tm._generateP12Cert()
self.failUnless(os.path.exists(tm.p12cert))
tm._generateP12Cert()
tm._retrieveManifestsPath()
tm.pdiskPathNew = tm._buildPDiskPath(tm.createdPDiskId)
tm._buildAndSaveManifest()
self.failUnless(os.path.exists(tm.manifestNotSignedPath))
minfo = ManifestInfo()
minfo.parseManifestFromFile(tm.manifestNotSignedPath)
assert minfo.comment == 'test'
assert minfo.creator == 'Jay Random'
assert minfo.version == '0.0'
assert minfo.sha1 == tm.imageSha1
assert minfo.locations == ['pdisk:' + Util.getHostnamePortFromUri(tm.persistentDiskPublicBaseUrl) + ':foo-bar-baz']
self.failUnless('foo-bar-baz' in str(tm._emailText()))
if not Signator.findJar():
print "Skipping signature sub-test as Signator jar can not be found."
return
tm._signManifest()
self.failUnless(os.path.exists(tm.manifestPath))
finally:
tm._cleanup()
