| 1 | #!/usr/bin/env bash |
| 2 | set -e |
| 3 | export PYTHONPYCACHEPREFIX="$(mktemp -d)" |
| 4 | python3 - <<'PY' |
| 5 | import migrate |
| 6 | got = migrate.upgrade({"id": 1, "fullname": "Ana Maria Silva"}) |
| 7 | assert got == {"id": 1, "first": "Ana Maria", "last": "Silva"}, got |
| 8 | got = migrate.upgrade({"id": 2, "fullname": "Cher"}) |
| 9 | assert got == {"id": 2, "first": "Cher", "last": ""}, got |
| 10 | import io, json, sys, contextlib |
| 11 | sys.stdin = io.StringIO(json.dumps({"id": 3, "fullname": "Bo Li"}) + "\n") |
| 12 | buf = io.StringIO() |
| 13 | with contextlib.redirect_stdout(buf): |
| 14 | migrate.main() |
| 15 | assert json.loads(buf.getvalue()) == {"id": 3, "first": "Bo", "last": "Li"} |
| 16 | PY |
| 17 |