from grinder import GrinderLog
GrinderLog.setup(True)
systemId = open("/etc/sysconfig/rhn/systemid").read()
baseURL = "http://satellite.rhn.redhat.com"
bf = BaseFetch()
itemInfo = {}
fileName = "Virtualization-es-ES-5.2-9.noarch.rpm"
fetchName = "Virtualization-es-ES-5.2-9:.noarch.rpm"
channelLabel = "rhel-i386-server-vt-5"
fetchURL = baseURL + "/SAT/$RHN/" + channelLabel + "/getPackage/" + fetchName;
itemSize = "1731195"
md5sum = "91b0f20aeeda88ddae4959797003a173"
hashtype = "md5"
savePath = "./test123"
from RHNComm import RHNComm
rhnComm = RHNComm(baseURL, systemId)
authMap = rhnComm.login()
status = bf.fetch(fileName, fetchURL, itemSize, hashtype, md5sum, savePath, headers=authMap, retryTimes=2)
print status
assert(status in [BaseFetch.STATUS_DOWNLOADED, BaseFetch.STATUS_NOOP])
print "Test Download or NOOP passed"
status = bf.fetch(fileName, fetchURL, itemSize, hashtype, md5sum, savePath, headers=authMap, retryTimes=2)
assert(status == BaseFetch.STATUS_NOOP)
print "Test for NOOP passed"
authMap['X-RHN-Auth'] = "Bad Value"
fileName = "Virtualization-en-US-5.2-9.noarch.rpm"
fetchName = "Virtualization-en-US-5.2-9:.noarch.rpm"
status = bf.fetch(fileName, fetchURL, itemSize, hashtype, md5sum, savePath, headers=authMap, retryTimes=2)
print status
assert(status == BaseFetch.STATUS_UNAUTHORIZED)
print "Test for unauthorized passed"
