Подготовить skeleton лабораторной 5 по мессенджеру

tests/smoke.py

@@ -0,0 +1,82 @@
+#!/usr/bin/env python3
+"""One complete public acceptance scenario, cumulative by lab number."""
+import base64
+import hashlib
+import uuid
+from client import Client, LAB, arguments, check, run, timestamp
+
+
+def main():
+    args = arguments('Messenger public smoke; creates synthetic users and messages')
+    c = Client(args.base_url,args.ca_file)
+    alice,bob,eve,chat = c.bootstrap()
+    path = '/api/v1/chats/'+chat['id']
+    c.call('POST','/api/v1/users',409,body=alice['registration'])
+    c.call('GET','/api/v1/users/me',401)
+    c.call('GET',path,404,actor=eve)
+    me = c.call('GET','/api/v1/users/me',actor=alice,schema='User')['json']
+    check(me==alice['user'],'current user mismatch')
+    listed = c.call('GET','/api/v1/chats?limit=100',actor=bob,schema='ChatPage')['json']
+    check(any(x['id']==chat['id'] for x in listed['items']),'member cannot list chat')
+    hidden = c.call('GET','/api/v1/chats?limit=100',actor=eve,schema='ChatPage')['json']
+    check(all(x['id']!=chat['id'] for x in hidden['items']),'chat leaks to outsider')
+    key = uuid.uuid4().hex
+    first = c.message(alice,chat,'Привет, мир 👋',key=key)['json']
+    second = c.message(bob,chat,'Сообщение Боба')['json']
+    check(first['sender_id']==alice['id'] and second['sender_id']==bob['id'],'sender was not derived from actor')
+    check(first['chat_id']==second['chat_id']==chat['id'],'wrong chat in message')
+    check(first['text']=='Привет, мир 👋' and second['text']=='Сообщение Боба','message text changed')
+    page1 = c.call('GET',path+'/messages?limit=1',actor=bob,schema='MessagePage')['json']
+    check(len(page1['items'])==1 and page1['next_cursor'],'first page must have one item and cursor')
+    from urllib.parse import quote
+    page2 = c.call('GET',path+'/messages?limit=1&cursor='+quote(page1['next_cursor'],safe=''),actor=alice,schema='MessagePage')['json']
+    check(len(page2['items'])==1 and page2['next_cursor'] is None,'second page must finish history')
+    ordered = sorted([first,second],key=lambda x:(timestamp(x['created_at']),uuid.UUID(x['id']).int))
+    check(page1['items']+page2['items']==ordered,'pagination lost, duplicated or reordered a message')
+    h = {'Idempotency-Key':uuid.uuid4().hex} if LAB>=3 else {}
+    c.call('POST',path+'/messages',400,actor=alice,body={'text':''},headers=h)
+    c.call('POST',path+'/messages',404,actor=eve,body={'text':'чужое'},headers=h)
+    c.call('GET',path+'/messages?limit=0',400,actor=alice)
+    if LAB>=2:
+        bad = base64.b64encode((alice['registration']['username']+':wrong-password').encode()).decode()
+        c.call('POST','/api/v1/auth/sessions',401,headers={'Authorization':'Basic '+bad})
+        good = base64.b64encode((alice['registration']['username']+':'+alice['registration']['password']).encode()).decode()
+        c.call('GET','/api/v1/users/me',401,headers={'Authorization':'Basic '+good})
+        c.call('GET','/api/v1/users/me',401,headers={'X-User-Id':alice['id']})
+        me = c.call('GET','/api/v1/users/me',actor=alice,headers={'X-User-Id':eve['id']},schema='User')['json']
+        check(me['id']==alice['id'],'X-User-Id overrode bearer identity')
+    if LAB>=3:
+        replay = c.message(alice,chat,first['text'],key=key)['json']
+        check(replay==first,'idempotency replay differs')
+        c.call('POST',path+'/messages',409,actor=alice,body={'text':'другой body'},headers={'Idempotency-Key':key})
+        c.call('POST',path+'/messages',400,actor=alice,body={'text':'без ключа'})
+    if LAB>=4:
+        att,content = c.attachment(alice,chat)
+        # A queued attachment must already be linkable to its own chat.
+        attached = c.message(alice,chat,'',attachments=[att['id']])['json']
+        check(attached['attachment_ids']==[att['id']],'message lost attachment')
+        ready = c.wait_ready(bob,att) if LAB>=5 else att
+        check(ready['error_code'] is None,'ready attachment has an error')
+        download = c.call('GET','/api/v1/attachments/'+att['id']+'/content',actor=bob)
+        check(download['bytes']==content,'download differs from original')
+        check(download['headers'].get_content_type()=='image/png','original content type mismatch')
+        for suffix in ('','/content'):
+            c.call('GET','/api/v1/attachments/'+att['id']+suffix,404,actor=eve)
+        if LAB>=5:
+            variants = [x for x in ready['variants'] if x['name']=='thumbnail']
+            check(len(variants)==1,'expected exactly one thumbnail')
+            v = variants[0]
+            check((v['width'],v['height'])==(256,160),'thumbnail metadata has wrong dimensions')
+            thumb = c.call('GET','/api/v1/attachments/'+att['id']+'/content?variant=thumbnail',actor=bob)
+            check(thumb['headers'].get_content_type()=='image/jpeg','thumbnail content type mismatch')
+            check(len(thumb['bytes'])==v['size_bytes'] and hashlib.sha256(thumb['bytes']).hexdigest()==v['sha256'],'thumbnail checksum mismatch')
+            from image_probe import jpeg_size
+            check(jpeg_size(thumb['bytes'])==(256,160),'actual JPEG dimensions mismatch')
+    if LAB>=2:
+        c.call('DELETE','/api/v1/auth/sessions/current',204,actor=alice)
+        c.call('GET','/api/v1/users/me',401,actor=alice)
+        c.call('DELETE','/api/v1/auth/sessions/current',401,actor=alice)
+    print('PASS: lab%d public API scenario; infrastructure and grading evidence remain separate.' % LAB)
+
+if __name__=='__main__':
+    run(main)