from django import forms
from django.contrib.auth.forms import UserCreationForm
from django.contrib.auth.models import User
# Define the user_type choices
USER_TYPE_CHOICES = (
    ('landlord', 'Landlord'),
    ('agent', 'Agent'),
    ('prospect', 'Prospect'),
)
# Create the registration form
class RegistrationForm(UserCreationForm):
    username = forms.CharField(widget=forms.TextInput(attrs={'class': 'form-control left-label'}))
    password1 = forms.CharField(widget=forms.PasswordInput(attrs={'class': 'form-control left-label'}))
    password2 = forms.CharField(widget=forms.PasswordInput(attrs={'class': 'form-control left-label'}))
    user_type = forms.ChoiceField(choices=USER_TYPE_CHOICES, widget=forms.Select(attrs={'class': 'form-control left-label'}))
   
    
    class Meta:
        model = User
        fields = ['username', 'password1', 'password2', 'user_type']

def register(request):
    if request.method == 'POST':
        form = RegistrationForm(request.POST)
        if form.is_valid():
            # Create a new user object
            user = form.save()
            # Get the user_type value from the form
            user_type = form.cleaned_data['user_type']
            # Create a new object based on the user_type
            if user_type == 'landlord':
                Landlord.objects.create(user=user)
            elif user_type == 'agent':
                Agent.objects.create(user=user)
            elif user_type == 'prospect':
                Prospect.objects.create(user=user)
            # Log the user in and redirect to the homepage
            login(request, user)
            return redirect('success-account')
    else:
        form = RegistrationForm()
    context =  {
        'form': form,
        'page_title': 'Register',
        }
    return render(request, 'account/register.html', context)

from django.contrib.auth.models import User
from django.test import TestCase, Client
from django.urls import reverse
from account.forms import RegistrationForm
from realestate.models import Landlord, Agent, Prospect
class RegisterViewTestCase(TestCase):
    def setUp(self):
        self.client = Client()
        self.url = reverse('register-account')
    def test_register_view_success(self):
        data = {
            'username': 'testuser',
            'email': 'testuser@example.com',
            'password1': 'testpassword',
            'password2': 'testpassword',
            'user_type': 'landlord'
        }
        form = RegistrationForm(data)
        self.assertTrue(form.is_valid())
        response = self.client.post(self.url, data)
        self.assertEqual(response.status_code, 302)
        self.assertRedirects(response, reverse('success-account'))
        user = User.objects.get(username='testuser')
        self.assertIsNotNone(user)
        self.assertTrue(user.check_password('testpassword'))
        landlord = Landlord.objects.get(user=user)
        self.assertIsNotNone(landlord)
    def test_register_view_invalid_form(self):
        data = {
            'username': 'testuser',
            'email': 'testuser@example.com',
            'password1': 'testpassword',
            'password2': 'wrongpassword',
            'user_type': 'landlord'
        }
        form = RegistrationForm(data)
        self.assertFalse(form.is_valid())
        response = self.client.post(self.url, data)
        self.assertEqual(response.status_code, 200)
        self.assertTemplateUsed(response, 'account/register.html')
        form = response.context['form']
        self.assertTrue(form.has_error('password2'))
        user_exists = User.objects.filter(username='testuser').exists()
        self.assertFalse(user_exists)