eregs/regulations-core

import json
from unittest import TestCase

from django.test.client import Client
from mock import patch


class ViewsDiffTest(TestCase):

    def test_add_not_json(self):
        url = '/diff/lablab/oldold/newnew'

        response = Client().put(url, content_type='application/json',
                                data='{Invalid}')
        self.assertEqual(400, response.status_code)

    @patch('regcore_write.views.diff.storage')
    def test_add_label_success(self, storage):
        url = '/diff/lablab/oldold/newnew'

        Client().put(url, content_type='application/json',
                     data=json.dumps({'some': 'struct'}))
        args = storage.for_diffs.insert.call_args[0]
        self.assertEqual(('lablab', 'oldold', 'newnew', {'some': 'struct'}),
                         args)