Add tests for security group migration tasks
Remove unused import from server.py