alias ssh-test="ssh test-server.example.com"
alias ssh-prod="ssh prod-server.example.com"
There is another, much better, and more straightforward solution to this problem.
With SSH you can configure different options for each machine you connect to.
Host test
Hostname test-server.example.com
Host prod
Hostname prod-server.example.com
$ ssh test
Host dev test
Hostname test-server.example.com
IdentityFile ~/.ssh/devop_ed25519
User devop
Host preview stage qa
Hostname stage-server.example.com
IdentityFile ~/.ssh/devop_ed25519
User devop
Host prod production
Hostname prod-server.example.com
IdentityFile ~/.ssh/admin_ed25519
User admin
ssh-keygen -t ed25519 -f ~/.ssh/admin_ed25519
ssh-copy-id -i ~/.ssh/admin_ed25519.pub prod