The internal drive in a 21.5 inch iMac is not user replaceable. Very questionable about an Apple Store or Certified Service Provider doing this for you, or without huge charge.
You would be better served to use an external USB-C 3.2 Gen 2 (10Mbps) drive. I use Crucial X8 2TB external drives (1050 MB/s) drives as Time Machine drives connected via USB-C to either a Mac's Thunderbolt 3/4 port, or to a USB-C 3.2 Gen 2 rated hub (startech.com) that I then connect to the preceding Thunderbolt 3/4 port. These would be very fast for running macOS externally. These drives comes with a USB-C to USB-C 3.2 Gen 2 cable and a USB-C to USB-A dongle.
Crucial's MX 500 is slower than the second coming compared to other external drive options.
I would use Carbon Copy Cloner (CCC) to clone the Internal drive to the externally mounted, APFS formatted X8, set the macOS Settings > General > Startup Disk to the Crucial X8, and reboot. Recommend a second Crucial X8 as a Time Machine drive to the other TB3 port. This is what I use.