"""Some sanity checking against the REST interface directly"""
import redbiom
import redbiom.admin
import unittest
import biom
import requests
import json
import numpy as np
import pandas as pd
from redbiom.tests import assert_test_env

assert_test_env()


table = biom.load_table('test.biom')
metadata = pd.read_csv('test.txt', sep='\t', dtype=str)
md = pd.read_csv('test.txt', sep='\t', dtype=str).set_index('#SampleID')


def get(cmd, remainder):
    req = requests.get('http://127.0.0.1:7379/%s/%s' % (cmd, remainder))
    if req.status_code != 200:
        raise requests.HTTPError()
    return req.json()[cmd]


# database integrity tests
class RESTTests(unittest.TestCase):
    def setUp(self):
        host = redbiom.get_config()['hostname']
        req = requests.get(host + '/FLUSHALL')
        assert req.status_code == 200

    def test_feature_sample_associations(self):
        context = 'test'
        redbiom.admin.create_context(context, 'foo')
        redbiom.admin.load_sample_metadata(metadata)
        redbiom.admin.ScriptManager.load_scripts(read_only=False)
        redbiom.admin.load_sample_data(table, context, tag=None)

        sample_ids = np.array(['UNTAGGED_%s' % i for i in table.ids()])
        inv_index = get('HGETALL', 'test:sample-index-inverted')
        for values, id_, _ in table.iter(axis='observation'):
            exp = set(sample_ids[values > 0])
            obs = get('LRANGE', 'test:feature:%s/%s/%s' %
                      (id_, '0', '-1'))
            obs = {inv_index[i] for i in obs[::2]}
            self.assertEqual(obs, exp)

    def test_sample_data(self):
        context = 'test'
        redbiom.admin.create_context(context, 'foo')
        redbiom.admin.load_sample_metadata(metadata)
        redbiom.admin.ScriptManager.load_scripts(read_only=False)
        redbiom.admin.load_sample_data(table, context, tag=None)

        feature_ids = table.ids(axis='observation')

        # get the inverted mapping of index -> ID
        inv_index = get('HGETALL', 'test:feature-index-inverted')

        for values, id_, _ in table.iter():
            exp_data = values[values > 0]
            exp_ids = feature_ids[values > 0]
            exp_dict = {i: v for i, v in zip(exp_ids, exp_data)}

            obs = get('LRANGE',
                      'test:sample:UNTAGGED_%s/%s/%s' %
                      (id_, "0", "-1"))
            obs_dict = {inv_index[i]: float(v) for i, v in zip(obs[::2],
                                                               obs[1::2])}

            self.assertEqual(obs_dict, exp_dict)

    def test_metadata_categories(self):
        context = 'test'
        redbiom.admin.create_context(context, 'foo')
        redbiom.admin.load_sample_metadata(metadata)

        null_values = {'Not applicable', 'Unknown', 'Unspecified',
                       'Missing: Not collected', np.nan,
                       'Missing: Not provided',
                       'Missing: Restricted access',
                       'null', 'NULL', 'no_data', 'None', 'nan'}

        for idx, row in md.iterrows():
            exp = [c for c, v in zip(md.columns, row.values)
                   if v not in null_values]
            obs = json.loads(get('GET', 'metadata:categories:%s' % idx))

            self.assertEqual(obs, exp)

    def test_category_values(self):
        context = 'test'
        redbiom.admin.create_context(context, 'foo')
        redbiom.admin.load_sample_metadata(metadata)

        null_values = {'Not applicable', 'Unknown', 'Unspecified',
                       'Missing: Not collected', np.nan,
                       'Missing: Not provided',
                       'Missing: Restricted access',
                       'null', 'NULL', 'no_data', 'None', 'nan'}

        for idx, row in md.iterrows():
            cats = [c for c, v in zip(md.columns, row.values)
                    if v not in null_values and '/' not in str(v)]
            for c in cats:
                exp = row[c]
                obs = get('HGET', 'metadata:category:%s/%s' % (c, idx))
                self.assertEqual(obs, str(exp))


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