from unittest import TestCase
from threading import Thread
from http.client import HTTPConnection
+from json import loads as json_loads
from urllib.parse import urlencode
from uuid import uuid4
from os import remove as remove_file
f'/process?id={id_}')
return form_data
- @staticmethod
- def blank_history_keys_in(d: dict[str, object]) -> None:
- """Re-write "history" object keys to bracketed integer strings."""
- def walk_tree(d: Any) -> Any:
- if isinstance(d, dict):
- if 'history' in d.keys():
- vals = d['history'].values()
- history = {}
- for i, val in enumerate(vals):
- history[f'[{i}]'] = val
- d['history'] = history
- for k in list(d.keys()):
- walk_tree(d[k])
- elif isinstance(d, list):
- d[:] = [walk_tree(i) for i in d]
- return d
- walk_tree(d)
+ def check_json_get(self, path: str, expected: dict[str, object]) -> None:
+ """Compare JSON on GET path with expected.
+
+ To simplify comparison of VersionedAttribute histories, transforms
+ timestamp keys of VersionedAttribute history keys into integers
+ counting chronologically forward from 0.
+ """
+ def rewrite_history_keys_in(item: Any) -> Any:
+ if isinstance(item, dict):
+ if '_versioned' in item.keys():
+ for k in item['_versioned']:
+ vals = item['_versioned'][k].values()
+ history = {}
+ for i, val in enumerate(vals):
+ history[i] = val
+ item['_versioned'][k] = history
+ for k in list(item.keys()):
+ rewrite_history_keys_in(item[k])
+ elif isinstance(item, list):
+ item[:] = [rewrite_history_keys_in(i) for i in item]
+ return item
+ self.conn.request('GET', path)
+ response = self.conn.getresponse()
+ self.assertEqual(response.status, 200)
+ retrieved = json_loads(response.read().decode())
+ rewrite_history_keys_in(retrieved)
+ self.assertEqual(expected, retrieved)