def test_set_appserver_active(self, mocks): """ Test set_appserver_active() """ instance = OpenEdXInstanceFactory(internal_lms_domain='test.activate.opencraft.co.uk', use_ephemeral_databases=True) appserver_id = instance.spawn_appserver() instance.set_appserver_active(appserver_id) instance.refresh_from_db() self.assertEqual(instance.active_appserver.pk, appserver_id) self.assertEqual(mocks.mock_load_balancer_run_playbook.call_count, 2) instance.set_appserver_inactive() instance.refresh_from_db() self.assertIsNone(instance.active_appserver) self.assertEqual(mocks.mock_load_balancer_run_playbook.call_count, 3)