def test_set_relationship(self):
resource = create_resource({"type": "fake"})
resource.relationship(
"messages",
[
create_resource({
"type": "message",
"id": "messageOne"
}),
create_resource({
"type": "message",
"id": "messageTwo"
}),
],
)
self.assertEqual(
resource.get_json_relationships(),
{
"messages": {
"data": [
{
"id": "messageOne",
"type": "message"
},
{
"id": "messageTwo",
"type": "message"
},
]
}
},
)
def test_to_json(self):
resource = ApiResource({
"type": "fake",
"id": "FakeID",
})
resource.set_relationship(
"thing",
create_resource({
"type": "things",
"id": "thingID",
}),
)
self.assertEqual(
json.dumps(resource.to_json()),
json.dumps({
"type": "fake",
"attributes": {},
"id": "FakeID",
"relationships": {
"thing": {
"data": {
"type": "things",
"id": "thingID"
}
}
},
}),
)
def test_to_json_changed_attributes(self):
resource = ApiResource({
"type": "fake",
"id": "FakeID",
})
self.assertEqual({}, resource.to_json_changed_attributes())
resource.attribute("test", "Hello World")
resource.set_relationship(
"thing",
create_resource({
"type": "things",
"id": "thingID",
}),
)
self.assertEqual(
json.dumps({
"type": "fake",
"attributes": {
"test": "Hello World"
},
"id": "FakeID"
}),
json.dumps(resource.to_json_changed_attributes()),
)
def test_get_relationship_create(self):
resource = create_resource({
"type": "fake",
})
relationship = resource.get_relationship("something")
self.assertIsInstance(relationship, Relationship)
def test_related(self):
resource = create_resource({
"type": "fake",
})
relation = resource.related("something")
self.assertIsInstance(relation, Relation)
def test_init_relationship(self):
resource = create_resource({
"type": "fake",
})
resource.set_relationship(
"account",
create_resource({
"type": "account",
"id": "someAccountID",
}),
)
self.assertEqual(
resource.get_json_relationships(),
{"account": {
"data": {
"type": "account",
"id": "someAccountID"
}
}},
)
def test_get_json_relationships(self):
resource = create_resource({"type": "fake"})
resource.relationship(
"ignored",
{"data": {
"type": "this",
"id": "that",
}},
)
resource.reset_changed_relations()
resource.relationship("object",
{"data": {
"type": "this",
"id": "that"
}})
resource.relationship("resource", ApiResource("this", "that"))
resource.relationship("resource_identifier",
ApiResourceIdentifier("this", "that"))
resource.relationship("list", [ApiResourceIdentifier("this", "that")])
self.assertEqual(
resource.get_json_changed_relationships(),
{
"object": {
"data": {
"id": "that",
"type": "this"
}
},
"resource": {
"data": {
"id": "that",
"type": "this"
}
},
"resource_identifier": {
"data": {
"id": "that",
"type": "this"
}
},
"list": {
"data": [{
"id": "that",
"type": "this"
}]
},
},
)
def test_init(self):
resource = create_resource({
"type": "fake",
})
resource.attribute("first_name", "John")
resource.attribute("last_name", "Doe")
self.assertEqual(resource.attribute("first_name"), "John")
self.assertEqual(resource.type(), "fake")
self.assertIsNone(resource.id())
self.assertEqual(
resource.attributes(),
{
"first_name": "John",
"last_name": "Doe",
},
)
def test_get_json_changed_relationships(self):
resource = create_resource({"type": "fake"})
resource.relationship(
"messages",
{"data": {
"type": "this",
"id": "that",
}},
)
self.assertEqual(
resource.get_json_changed_relationships(),
{"messages": {
"data": {
"id": "that",
"type": "this"
}
}},
)