fix: create fk model in benchmark script

[betodealmeida]

SUMMARY

When identifying models, the benchmark migration script should also take into consideration models that are referenced in foreign keys.

For example, running the script on 301362411006_add_execution_id_to_report_execution_.py would identify a single model, for the report_execution_log table. But that table has a foreign key to report_schedule.id. On a particular run the script was failing because report_schedule was empty.

I changed the script to also populate the models referenced in foreign keys. Ideally we'd need only 1 entity for each of those models, but for now the script generates multiple entities of them as well. I'll clean it up in a follow up PR.

BEFORE/AFTER SCREENSHOTS OR ANIMATED GIF

N/A

TESTING INSTRUCTIONS

Make sure you have no reports scheduled. Run:

% python scripts/benchmark_migration.py superset/migrations/versions/301362411006_add_execution_id_to_report_execution_.py

It should work:

...
Identifying models used in the migration:
- ab_user (1 rows in table ab_user)
- dashboards (12 rows in table dashboards)
- slices (114 rows in table slices)
- dbs (2 rows in table dbs)
- report_schedule (0 rows in table report_schedule)
- report_execution_log (0 rows in table report_execution_log)
...
Results:

Current: 0.20 s
10+: 0.37 s
100+: 0.19 s
1000+: 0.20 s

ADDITIONAL INFORMATION

• Has associated issue:
• Changes UI
• Includes DB Migration (follow approval process in SIP-59)
  • Migration is atomic, supports rollback & is backwards-compatible
  • Confirm DB migration upgrade and downgrade tested
  • Runtime estimates and downtime expectations provided
• Introduces new feature or API
• Removes existing feature or API

[codecov]

Codecov Report

Merging #15804 (83ed3d2) into master (cbd3780) will decrease coverage by 0.00%.
The diff coverage is n/a.

@@            Coverage Diff             @@
##           master   #15804      +/-   ##
==========================================
- Coverage   76.91%   76.90%   -0.01%     
==========================================
  Files         983      983              
  Lines       51643    51643              
  Branches     6991     6991              
==========================================
- Hits        39720    39717       -3     
- Misses      11699    11702       +3     
  Partials      224      224              

Flag	Coverage Δ
javascript	71.81% <ø> (-0.02%)	⬇️
mysql	81.54% <ø> (ø)
postgres	81.57% <ø> (ø)
python	81.66% <ø> (ø)
sqlite	81.18% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
superset-frontend/src/components/Select/Select.tsx	72.25% <0.00%> (-1.74%)	⬇️

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update cbd3780...83ed3d2. Read the comment docs.