from datasette.app import Datasette
import os
import pytest
import sqlite3
import tempfile


@pytest.fixture(scope='module')
def three_table_app_client():
    with tempfile.TemporaryDirectory() as tmpdir:
        filepath = os.path.join(tmpdir, 'four_tables.db')
        conn = sqlite3.connect(filepath)
        conn.executescript(FOUR_TABLES)
        os.chdir(os.path.dirname(filepath))
        yield Datasette([filepath]).app().test_client


def test_homepage(three_table_app_client):
    _, response = three_table_app_client.get('/')
    assert response.status == 200
    assert 'four_tables' in response.text

    # Now try the JSON
    _, response = three_table_app_client.get('/.json')
    assert response.status == 200
    assert response.json.keys() == {'four_tables': 0}.keys()
    d = response.json['four_tables']
    assert d['name'] == 'four_tables'
    assert d['tables_count'] == 4


def test_database_page(three_table_app_client):
    _, response = three_table_app_client.get('/four_tables', allow_redirects=False)
    assert response.status == 302
    _, response = three_table_app_client.get('/four_tables')
    assert 'four_tables' in response.text
    # Test JSON list of tables
    _, response = three_table_app_client.get('/four_tables.json')
    data = response.json
    assert 'four_tables' == data['database']
    assert [{
        'columns': ['pk', 'content'],
        'name': 'Table With Space In Name',
        'table_rows': 0,
    }, {
        'columns': ['pk1', 'pk2', 'content'],
        'name': 'compound_primary_key',
        'table_rows': 0,
    }, {
        'columns': ['content'],
        'name': 'no_primary_key',
        'table_rows': 0,
    }, {
        'columns': ['pk', 'content'],
        'name': 'simple_primary_key',
        'table_rows': 2,
    }] == data['tables']


def test_custom_sql(three_table_app_client):
    _, response = three_table_app_client.get(
        '/four_tables.jsono?sql=select+content+from+simple_primary_key'
    )
    data = response.json
    assert {
        'sql': 'select content from simple_primary_key',
        'params': {}
    } == data['query']
    assert [
        {'content': 'hello'},
        {'content': 'world'}
    ] == data['rows']
    assert ['content'] == data['columns']
    assert 'four_tables' == data['database']


def test_table_page(three_table_app_client):
    _, response = three_table_app_client.get('/four_tables/simple_primary_key')
    assert response.status == 200
    _, response = three_table_app_client.get('/four_tables/simple_primary_key.jsono')
    assert response.status == 200
    data = response.json
    assert data['query']['sql'] == 'select * from "simple_primary_key" order by pk limit 51'
    assert data['query']['params'] == {}
    assert data['rows'] == [{
        'pk': '1',
        'content': 'hello',
    }, {
        'pk': '2',
        'content': 'world',
    }]


def test_view(three_table_app_client):
    _, response = three_table_app_client.get('/four_tables/simple_view')
    assert response.status == 200
    _, response = three_table_app_client.get('/four_tables/simple_view.jsono')
    assert response.status == 200
    data = response.json
    assert data['rows'] == [{
        'upper_content': 'HELLO',
        'content': 'hello',
    }, {
        'upper_content': 'WORLD',
        'content': 'world',
    }]


FOUR_TABLES = '''
CREATE TABLE simple_primary_key (
  pk varchar(30) primary key,
  content text
);

CREATE TABLE compound_primary_key (
  pk1 varchar(30),
  pk2 varchar(30),
  content text,
  PRIMARY KEY (pk1, pk2)
);

CREATE TABLE no_primary_key (
  content text
);

CREATE TABLE "Table With Space In Name" (
  pk varchar(30) primary key,
  content text
);

INSERT INTO simple_primary_key VALUES (1, 'hello');
INSERT INTO simple_primary_key VALUES (2, 'world');

CREATE VIEW simple_view AS
    SELECT content, upper(content) AS upper_content FROM simple_primary_key;
'''