from django.test import TestCase
from django.test.client import Client


class Test(TestCase):
    def testLogin(self):
        client = Client()
        headers = {'X-OpenAM-Username': 'user', 'X-OpenAM-Password': 'password', 'Content-Type': 'application/json'}
        data = {}
        response = self.client.post('/login/', headers=headers, data=data, secure=True, follow=True)
        assert (response.status_code == 200)
        # self.assertRedirects(response, '/menu/', status_code=301, target_status_code=200)

def login(request):
    logout(request)
    message = None
    if request.method == "POST":
        form = LoginForm(request.POST)
        if form.is_valid():
            username = request.POST['username']
            password = request.POST['password']
            headers = {'X-OpenAM-Username': username, 'X-OpenAM-Password': password, 'Content-Type': 'application/json'}
            data = {}
            req = requests.post('http://openam.idp.com:8090/openamIDP/json/authenticate', headers=headers, params=data)
            if req.status_code == 200:
                    respJson = json.loads(req.content)
                    tokenIdJson = respJson['tokenId']
                    request.session['tokenId'] = tokenIdJson
                    return render_to_response('menu/menu.html', request)
            elif req.status_code == 401:
                message = "Invalid username and/or password. Please, try again"
    else:
        form = LoginForm()
    return render_to_response('registration/login.html', {'message': message, 'form': form},
                             context_instance=RequestContext(request))