#!/usr/bin/python3


import json
import requests
import subprocess
import signal
import socket
import time
import unittest


USER = '_graphite'
ADDR = '127.0.0.1'
PORT = 8080


class SmokeTest(unittest.TestCase):

    @classmethod
    def server_ready(cls):
        try:
            s = socket.socket()
            s.connect((ADDR, PORT, ))
        except socket.error:
            return False
        else:
            return True
        finally:
            s.close()

    @classmethod
    def setUpClass(cls):
        # create db
        subprocess.check_call(['/usr/bin/graphite-manage', 'migrate',
                               '--run-syncdb'],
                              user=USER)

        # launch dev server
        cls.testserver = subprocess.Popen(['/usr/bin/django-admin', 'runserver',
                                           '--settings', 'graphite.settings',
                                           f'{ADDR}:{PORT}'],
                                           user=USER)

        # ensure server is ready
        retries = 10
        while not cls.server_ready() and retries > 0:
            time.sleep(0.5)
            retries = retries - 1
            if retries == 0:
                raise RuntimeError('test server is not available')

    def test_metrics(self):
        with requests.get(f'http://{ADDR}:{PORT}/metrics/index.json') as r:
            mlist = json.loads(r.text)
            self.assertTrue(mlist == [] or mlist[0])


    def test_functions(self):
        with requests.get(f'http://{ADDR}:{PORT}/functions') as r:
            flist = json.loads(r.text)
            self.assertFalse('error' in flist)
            self.assertTrue('asPercent' in flist)

    @classmethod
    def tearDownClass(cls):
        cls.testserver.send_signal(signal.SIGTERM)
        cls.testserver.wait()


if __name__ == '__main__':
    unittest.main()


