From 97cdce9bac4f7e1d859d015499b15a2bb36c4c3c Mon Sep 17 00:00:00 2001
From: Martin Burchell <martinb@aptivate.org>
Date: Wed, 29 Jul 2015 16:38:26 +0100
Subject: [PATCH] Refactored item_delete_tests to use the REST API
---
.../rest_api/tests/item_delete_tests.py | 25 ++++++++++++-------
1 file changed, 16 insertions(+), 9 deletions(-)
diff --git a/django/website/rest_api/tests/item_delete_tests.py b/django/website/rest_api/tests/item_delete_tests.py
index d2d9bc8d..701e8189 100644
--- a/django/website/rest_api/tests/item_delete_tests.py
+++ b/django/website/rest_api/tests/item_delete_tests.py
@@ -2,12 +2,13 @@ from __future__ import unicode_literals, absolute_import
import pytest
-from data_layer.tests.factories import ItemFactory
-from data_layer.models import Item
from rest_framework.test import APIRequestFactory
from rest_framework import status
from ..views import ItemViewSet
+from .item_create_view_tests import create_item
+from .item_list_view_tests import get as list_items
+
def delete_item(id):
request = APIRequestFactory().delete('/')
@@ -15,14 +16,20 @@ def delete_item(id):
return view(request, pk=id)
+def count_items():
+ items = list_items().data
+ return len(items)
+
+
@pytest.mark.django_db
def test_delete_item():
- ItemFactory(body="test1")
- item = ItemFactory()
- assert Item.objects.count() == 2
-
- response = delete_item(item.id)
+ create_item(body="test1")
+ item = create_item(body="test2")
+ assert count_items() == 2
+ response = delete_item(item.data['id'])
assert status.is_success(response.status_code)
- assert Item.objects.count() == 1
- assert Item.objects.get().body == "test1"
+ assert count_items() == 1
+
+ [item] = list_items().data
+ assert item['body'] == "test1"
--
GitLab