Fix unit tests
This commit is contained in:
parent
e5641b7f93
commit
0d7b948e44
|
|
@ -1,5 +1,4 @@
|
|||
import graphene
|
||||
from django.db.models import Q
|
||||
from graphene import relay
|
||||
|
||||
from api.types import FailureNode, SuccessNode
|
||||
|
|
@ -93,7 +92,7 @@ class DeleteSnapshot(relay.ClientIDMutation):
|
|||
return cls(result=NotOwnerFailure)
|
||||
|
||||
snapshot.delete()
|
||||
return cls(result=DeleteSnapshotResult)
|
||||
return cls(result=DeleteSnapshotSuccess)
|
||||
|
||||
|
||||
class ApplySnapshot(relay.ClientIDMutation):
|
||||
|
|
|
|||
|
|
@ -154,13 +154,16 @@ mutation UpdateSnapshot($input: UpdateSnapshotInput!) {
|
|||
"""
|
||||
|
||||
DELETE_SNAPSHOT_MUTATION = """
|
||||
mutation DeleteSnapshot($id: ID!) {
|
||||
deleteSnapshot(id: $id) {
|
||||
mutation DeleteSnapshot($input: DeleteSnapshotInput!) {
|
||||
deleteSnapshot(input: $input) {
|
||||
result {
|
||||
success
|
||||
__typename
|
||||
...on NotOwner {
|
||||
reason
|
||||
}
|
||||
...on Success {
|
||||
message
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -451,11 +451,21 @@ class SnapshotTestCase(SkillboxTestCase):
|
|||
def test_delete_snapshot(self):
|
||||
result = self.get_client().execute(DELETE_SNAPSHOT_MUTATION, variables={
|
||||
'input': {
|
||||
'id': self.snapshot
|
||||
'id': to_global_id('Snapshot', self.snapshot.id),
|
||||
}
|
||||
})
|
||||
self.assertIsNone(result.errors)
|
||||
self.assertEqual(result.data.get('deleteSnapshot').get('result').get('__typename'), 'SuccessNode')
|
||||
self.assertEqual(result.data.get('deleteSnapshot').get('result').get('__typename'), 'Success')
|
||||
|
||||
def test_delete_snapshot_not_owner_fails(self):
|
||||
self.assertTrue(False)
|
||||
self.snapshot.creator = self.teacher2
|
||||
self.snapshot.save()
|
||||
result = self.get_client().execute(DELETE_SNAPSHOT_MUTATION, variables={
|
||||
'input': {
|
||||
'id': to_global_id('Snapshot', self.snapshot.id),
|
||||
}
|
||||
})
|
||||
self.assertIsNone(result.errors)
|
||||
result = result.data.get('deleteSnapshot').get('result')
|
||||
self.assertEqual(result.get('__typename'), 'NotOwner')
|
||||
self.assertEqual(result.get('reason'), 'Not the owner')
|
||||
|
|
|
|||
Loading…
Reference in New Issue