def testRecordFiltering(self):
now = datetime.datetime.utcnow()
recordsData = [ (str(uuid.uuid4()), now - datetime.timedelta(days=10-i)) for i in range(10) ]
Record = db.models.Record
for recUuid, createdTime in recordsData:
resp = self._newRecord(uuid=recUuid,
created_time=createdTime.isoformat())
self.assertEqual(resp.status_code, 200)
self.assertEquals(resp.json['uuid'], recUuid)
rec = self.conn.query(Record).filter_by(uuid=recUuid).one()
self.assertEquals(rec.created_time, createdTime)
# Set updated_time
rec.updated_time = rec.created_time + datetime.timedelta(minutes=5)
self.conn.add(rec)
self.conn.commit()
req = resp.request
req.headers['Authorization'] = 'Basic %s' % base64.b64encode(
'{username}:{password}'.format(username=self.Username,
password=self.Password))
# Make sure the record got persisted correctly
allRecordsResp = records.records_view(req)
self.assertEquals(allRecordsResp['count'], 10)
allRecords = allRecordsResp['records']
self.assertEqual(
[(x['uuid'], x['created_time'], x['updated_time']) for x in allRecords],
[(x[0], x[1].isoformat(), (x[1] + datetime.timedelta(minutes=5)).isoformat()) for x in recordsData])
# Now build query
# Make sure we accept timezone specs too
uTimeStamp = (now - datetime.timedelta(days=1)).isoformat() + "%2B00:00"
nreq = self._req(req.url + '?filter=ge(updated_time,"%s")' % uTimeStamp,
headers=req.headers)
resp = self.app.invoke_subrequest(nreq, use_tweens=True)
self.assertEquals(resp.status_code, 200)
self.assertEquals(resp.json['count'], 1)
url = 'http://localhost/registration/v1/records?start=0&limit=100&filter=ge(updated_time,"{updatedTime}")'.format(updatedTime=uTimeStamp)
expectedLinks = [
('self', url),
('first', url),
]
self.assertEquals(
[ (x['rel'], x['href']) for x in resp.json['links'] ],
expectedLinks)
# Make sure we don't explode on a bad time spec
nreq = self._req(req.url + '?filter=ge(updated_time,"AAAA")',
headers=req.headers)
resp = self.app.invoke_subrequest(nreq, use_tweens=True)
self.assertEquals(resp.status_code, 400)
# APPENG-3387
# update_time is 5 minutes after create_time, and we want to bracket
# one minute before and after, so 4 and 6.
t0 = (recordsData[6][1] + datetime.timedelta(minutes=4)).isoformat() + "%2B00:00"
t1 = (recordsData[8][1] + datetime.timedelta(minutes=6)).isoformat() + "%2B00:00"
url = 'http://localhost/registration/v1/records?start=0&limit=100&filter=and(ge(updated_time,"{0}"),le(updated_time,"{1}"))'.format(t0, t1)
nreq = self._req(url, headers=req.headers)
resp = self.app.invoke_subrequest(nreq, use_tweens=True)
expectedLinks = [
('self', url),
('first', url),
]
self.assertEquals(
[ (x['rel'], x['href']) for x in resp.json['links'] ],
expectedLinks)
self.assertEquals(resp.json['count'], 3)
self.assertEqual(
[x['uuid'] for x in resp.json['records']],
[recordsData[6][0], recordsData[7][0], recordsData[8][0]])
def testRecordCreate(self):
# No entitlement
resp = self._newRecord(entitlements=None)
self.assertEquals(resp.status_code, 401)
logEntries = self._getLoggingCalls()
self.assertEquals(len(logEntries), 5)
self.assertEquals(logEntries[3], ('upsrv.views.records', "Missing auth header `%s' from %s", ('X-Conary-Entitlement', '10.11.12.13')))
self._resetLoggingCalls()
resp = self._newRecord(
producers={
'system-information' : self.DefaultProducers['system-information']})
self.assertEquals(resp.status_code, 400)
logEntries = self._getLoggingCalls()
self.assertEquals(len(logEntries), 4)
self.assertEquals(logEntries[1], ('upsrv.views.records', 'Missing system model from %s', ('10.11.12.13',)))
self._resetLoggingCalls()
# Correct record
resp = self._newRecord()
self.assertEquals(resp.status_code, 200)
self.assertEquals(resp.json_body['entitlement_valid'], True)
self.assertEquals(resp.json_body['entitlements_json'],
'[["a", "aaa"], ["*", "bbb"]]')
now = datetime.datetime.utcnow()
# Decode the json coming back
record = json.loads(resp.body)
self.assertEquals(record['uuid'], self.DefaultUuid)
self.assertEquals(record['created_time'],
self.DefaultCreatedTime + "+00:00")
self.assertEquals(record['client_address'], '10.11.12.13')
req = resp.request
allRecords = records.records_view(req)
self.assertEquals(allRecords.status_code, 403)
# Let's add an auth header, with a bad username
req.headers['Authorization'] = 'Basic %s' % base64.b64encode(
'{username}:{password}'.format(username='******',
password=self.Password))
allRecords = records.records_view(req)
self.assertEquals(allRecords.status_code, 403)
# Let's add an auth header, with a good username and a bad password
req.headers['Authorization'] = 'Basic %s' % base64.b64encode(
'{username}:{password}'.format(username=self.Username,
password='******'))
allRecords = records.records_view(req)
self.assertEquals(allRecords.status_code, 403)
# Let's add an auth header, with a good username/password
req.headers['Authorization'] = 'Basic %s' % base64.b64encode(
'{username}:{password}'.format(username=self.Username,
password=self.Password))
# Make sure the record got persisted correctly
allRecords = records.records_view(req)['records']
self.assertEquals([ x['entitlement_valid'] for x in allRecords ],
[ True ])
self.assertEquals([ x['entitlements_json'] for x in allRecords ],
[ '[["a", "aaa"], ["*", "bbb"]]' ])
rec = allRecords[0]
self.assertEquals(rec['uuid'], self.DefaultUuid)
# Make sure updated_time got set by the server
rectime = datetime.datetime.strptime(rec['updated_time'],
"%Y-%m-%dT%H:%M:%S.%f")
delta = now - rectime
totalSeconds = 86400 * delta.days + delta.seconds + delta.microseconds / 1e6
self.assertTrue(0 <= totalSeconds)
self.assertTrue(totalSeconds < 2)
self.assertEquals(req.getConaryClient._mock.popCall(),
((), (('entitlements', ['aaa', 'bbb']),)))
self.assertEquals(req.getConaryClient._mock.calls, [])
# Remove all records
self._resetLoggingCalls()
self._resetRecords()
# Same deal, but make findTroves raise an exception
_calls = []
_exc = records.repoerrors.TroveNotFound('blah')
def fakeFindTroves(label, troves, *args, **kwargs):
_calls.append((label, troves, args, kwargs))
raise _exc
req._conaryClient.repos._mock.set(findTroves=fakeFindTroves)
resp = self.app.invoke_subrequest(req, use_tweens=True)
self.assertEquals(resp.status_code, 200)
self.assertEquals(resp.json_body['entitlement_valid'], False)
self.assertEquals(resp.json_body['entitlements_json'],
'[["a", "aaa"], ["*", "bbb"]]')
logEntries = self._getLoggingCalls()
self.assertEquals(len(logEntries), 5)
self.assertEquals(logEntries[2],
('upsrv.views.records', '%s: bad entitlements %s for system model %s: %s',
('10.11.12.13', [('a', 'aaa'), ('*', 'bbb')],
self.DefaultProducers['conary-system-model']['data'],
_exc)))
self.assertEquals(len(_calls), 1)
self.assertEquals([str(x) for x in _calls[0][1]],
['foo=cny.tv@ns:1/1-2-3', 'group-university-appliance=university.cny.sas.com@sas:university-3p-staging/1-2-3[~!xen is: x86(i486,i586,i686) x86_64]'])
# Make sure the record got persisted correctly
allRecords = records.records_view(req)['records']
self.assertEquals([ x['entitlement_valid'] for x in allRecords ],
[ False ])
self.assertEquals([ x['entitlements_json'] for x in allRecords ],
[ '[["a", "aaa"], ["*", "bbb"]]' ])
# Remove all records
self._resetLoggingCalls()
self._resetRecords()
# Same deal, but with a 1M provider payload
content = '0123456789abcdef' * 64 * 1024
resp = self._newRecord(
producers={
'system-information' : content})
self.assertEquals(resp.status_code, 413)
logEntries = self._getLoggingCalls()
self.assertEquals(len(logEntries), 4)
self.assertEquals(logEntries[2],
('upsrv.views.records', 'Request too large from %s: %s bytes',
('10.11.12.13', 1048797)))
