def test_list_with_not_found(self):
class NotFound(Exception):
status_code = 404
neut = resources.NeutronSecurityGroup()
neut.user = mock.MagicMock()
neut._resource = "security_group"
neut.tenant_uuid = "user_tenant"
neut.user.neutron().list_security_groups.side_effect = NotFound()
expected_result = []
self.assertEqual(expected_result, list(neut.list()))
neut.user.neutron().list_security_groups.assert_called_once_with(
tenant_id=neut.tenant_uuid)
def test_list(self, admin, admin_required):
sg_list = [{"tenant_id": "user_tenant", "name": "default"},
{"tenant_id": "user_tenant", "name": "foo_sg"}]
neut = resources.NeutronSecurityGroup()
neut.user = mock.MagicMock()
neut._resource = "security_group"
neut.tenant_uuid = "user_tenant"
neut.user.neutron().list_security_groups.return_value = {
"security_groups": sg_list
}
expected_result = [sg_list[1]]
self.assertEqual(expected_result, list(neut.list()))
neut.user.neutron().list_security_groups.assert_called_once_with(
tenant_id=neut.tenant_uuid)