from select_test_case import TagBasedQuery
import pytest
import sqlite3 as db
import argparse

@pytest.fixture(scope="module")
def cur():
    tag_based_query = TagBasedQuery()
    conn = db.connect('C:/dbbot/robot_results.db')
    if conn:
        pass
    else:
       print "could not make DB connection"
    cur = conn.cursor()
    yield cur
    cur.close()
    conn.close()

def test_read_from_database(cur):
    data = cur.fetchone()
    assert(data,"first_path testname1 project1")


def test_read_from_commandline(cur):
    parser = argparse.ArgumentParser()
    parser = test_read_from_commandline(['query_opration','select'])
    assert(parser.query_operation,'select')