#!/usr/bin/env python
from api_client import APIconnection

import os
import io
import sys
import time
import numpy as np
import uuid

from PIL import Image

import unittest
from pprint import pprint

import logging, sys
logging.basicConfig(stream=sys.stderr, level=logging.INFO)


class TestCapture(unittest.TestCase):

    def test_capture_config(self):
        connection = APIconnection(host="localhost", port=5000, api_ver="v1")
        config = connection.get_config()
        
        expected_keys = [
            'image_resolution',
            'video_resolution',
            'numpy_resolution',
        ]

        for key in expected_keys:
            self.assertTrue(key in config)

    def test_capture_videoport(self):
        connection = APIconnection(host="localhost", port=5000, api_ver="v1")
        resolution = connection.get_config()['video_resolution']

        capture_array = connection.capture(
            use_video_port=True, 
            keep_on_disk=False, 
            delete_after_use=True)

        self.assertTrue(capture_array.shape == (resolution[1], resolution[0], 3))

    def test_capture_full(self):
        connection = APIconnection(host="localhost", port=5000, api_ver="v1")
        resolution = connection.get_config()['image_resolution']

        capture_array = connection.capture(
            use_video_port=False, 
            keep_on_disk=False, 
            delete_after_use=True)

        self.assertTrue(capture_array.shape == (resolution[1], resolution[0], 3))

class TestStage(unittest.TestCase):

    def test_stage_config(self):
        connection = APIconnection(host="localhost", port=5000, api_ver="v1")
        config = connection.get_config()
        
        expected_keys = [
            'backlash',
        ]

        for key in expected_keys:
            self.assertTrue(key in config)

if __name__ == '__main__':

    suites = [
        unittest.TestLoader().loadTestsFromTestCase(TestCapture),
        unittest.TestLoader().loadTestsFromTestCase(TestStage),
    ]

    alltests = unittest.TestSuite(suites)

    result = unittest.TextTestRunner(verbosity=2).run(alltests)