"""
System Name: Vasyworks
Project Name: vacancy_mgr
Encoding: UTF-8
Copyright (C) 2020 Yasuhiro Yamamoto
"""
from unittest import TestCase
from django.test import Client
from django.urls import reverse
from django.db import transaction
import warnings


class CreateStaffViewTest(TestCase):
    """
    スタッフ作成ビューのテスト
    """
    def setUp(self):
        warnings.simplefilter('ignore')
        self.client = Client()
        if transaction.get_autocommit():
            transaction.set_autocommit(False)

        response = self.client.post(
            reverse('login'),
            {'username': 't-kanri', 'password': 'guest1234', },
            follow=True
        )
        self.assertEqual(response.status_code, 200)

    def tearDown(self):
        transaction.rollback()

    def test_post_create_staff_view(self):
        url = reverse('company_create_staff')
        response = self.client.post(
            url,
            {
                'last_name': 'テスト',
                'first_name': 'スタッフ生成',
            },
            follow=True,
        )
        self.assertEqual(response.status_code, 200)
        messages = list(response.context['messages'])
        self.assertEqual(str(messages[0]), '追加しました。')

    def test_post_create_staff_view_with_department(self):
        url = reverse(
            'company_create_staff',
            args=['Mg'],  # 賃貸管理部
        )
        response = self.client.post(
            url,
            {
                'last_name': 'テスト',
                'first_name': 'スタッフ生成',
            },
            follow=True,
        )
        self.assertEqual(response.status_code, 200)
        messages = list(response.context['messages'])
        self.assertEqual(str(messages[0]), '追加しました。')
