def setUp(self):
super(DefaultRelayStateTest, self).setUp()
self.store = MagicMock()
self.client.data_store = self.store
self.drss = DefaultRelayStateList(
client=self.client, properties={'href': 'drss'})
self.organization = Organization(
client=self.client, properties={'name_key': 'NAME KEY'})
class DefaultRelayStateTest(SamlBuildURITest):
def setUp(self):
super(DefaultRelayStateTest, self).setUp()
self.store = MagicMock()
self.client.data_store = self.store
self.drss = DefaultRelayStateList(
client=self.client, properties={'href': 'drss'})
self.organization = Organization(
client=self.client, properties={'name_key': 'NAME KEY'})
def test_default_relay_state_create_empty(self):
self.drss.create()
self.store.create_resource.assert_called_once_with(
'drss', {}, params={})
def test_default_relay_state_create_organization(self):
self.drss.create({'organization': self.organization})
self.store.create_resource.assert_called_once_with(
'drss', {'organization': {'nameKey': 'NAME KEY'}}, params={})
def test_default_relay_state_create_organization_name_key(self):
self.drss.create({'organization': {'name_key': 'ANOTHER NAME KEY'}})
self.store.create_resource.assert_called_once_with(
'drss',
{'organization': {'nameKey': 'ANOTHER NAME KEY'}},
params={})