Fix retry functionality failing of Unifi

[Kane610]

Description:

Fix controller not being stored when setup fails and sequentially fails the retry functionality

Same fix as #17877, but fixes tests as well.

Checklist:

• The code change is tested and works locally.
• Local tests pass with tox. Your PR cannot be merged unless tests pass

[houndci-bot]

Some files could not be reviewed due to errors:

Traceback (most recent call last):

Traceback (most recent call last):
  File "/home/linters/.local/bin/flake8", line 7, in 
    from flake8.main.cli import main
ModuleNotFoundError: No module named 'flake8'

[Kane610]

@balloob this one would be good to get into .2