import base64
from django.core.management import call_command
from recognizer.models import CameraUser, LivePhoto, LivePersonPass
from rest_framework.test import APIClient, force_authenticate, APIRequestFactory
from recognizer.views import search_for_similar_view


def livephoto():
    user = CameraUser.objects.get(username='camerauser')
    request_factory = APIRequestFactory()
    with open('recognizer/tests/sherlock.jpeg', 'rb') as image_file:
        image_file = base64.standard_b64encode(image_file.read())
        form_data = {'data': image_file,
                     'cadr': image_file,
                     'identification': 1}
        request = request_factory.post(path='/camera/search/',
                                       data=form_data,
                                       format='json')
        force_authenticate(request, user=user)
        response = search_for_similar_view(request)
        return response


@pytest.mark.django_db(transaction=True)
def test_createlivepersonpass1(load_fixtures):
    # Тест на схлопывание фоток в проходку, ждем период схлопывания и жбахаем еще разок.
    def checklivephotopass():
        livephoto()
        livepersonpass = LivePersonPass.objects.all().values_list('id', flat=True)
        return len(livepersonpass)

    assert checklivephotopass() == 1
    time.sleep(5)
    assert checklivephotopass() == 1
    time.sleep(20)
    assert checklivephotopass() == 2