from django.contrib import auth
from django.urls import reverse

from cookbook.models import Storage
from cookbook.tests.views.test_views import TestViews


class TestEditsRecipe(TestViews):

    def setUp(self):
        super(TestEditsRecipe, self).setUp()

        self.storage = Storage.objects.create(
            name='TestStorage',
            method=Storage.DROPBOX,
            created_by=auth.get_user(self.admin_client_1),
            token='test',
            username='test',
            password='test',
        )
        self.url = reverse('edit_storage', args=[self.storage.pk])

    def test_edit_storage(self):
        r = self.admin_client_1.post(self.url, {'name': 'NewStorage', 'password': '1234_pw', 'token': '1234_token', 'method': Storage.DROPBOX})
        self.storage.refresh_from_db()
        self.assertEqual(self.storage.password, '1234_pw')
        self.assertEqual(self.storage.token, '1234_token')

        r = self.admin_client_1.post(self.url, {'name': 'NewStorage', 'password': '1234_pw', 'token': '1234_token', 'method': 'not_a_valid_method'})
        self.assertFormError(r, 'form', 'method', ['Select a valid choice. not_a_valid_method is not one of the available choices.'])

    def test_edit_storage_permissions(self):
        r = self.anonymous_client.get(self.url)
        self.assertEqual(r.status_code, 302)

        r = self.guest_client_1.get(self.url)
        self.assertEqual(r.status_code, 302)

        r = self.user_client_1.get(self.url)
        self.assertEqual(r.status_code, 302)

        r = self.admin_client_1.get(self.url)
        self.assertEqual(r.status_code, 200)

        r = self.superuser_client.get(self.url)
        self.assertEqual(r.status_code, 200)
