# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

# XXX Set to be deprecated in favor of socorro/external/postgresql/models.py

import unittest
import socorro.database.database as db
import psycopg2
import psycopg2.extensions
import logging
import threading

from socorro.unittest.testlib.loggerForTest import TestingLogger
from createDBforTest import *
import socorro.lib.util as util

import socorro.lib.ConfigurationManager as cm
import dbTestconfig as testConfig
config = cm.newConfiguration(configurationModule = testConfig, applicationName='Testing Psycopghelper')

"""
Assume that psycopg2 works, then all we need to do is assure ourselves
that our simplistic wrap around a returned array is correct
"""

class TestMultiCursor(psycopg2.extensions.cursor):
  def __init__(self,numCols = 4, numRows=2, **kwargs):
    self.result = []
    for i in range(numRows):
      aRow = []
      for j in range(numCols):
        aRow.append('Row %d, Column %d' %(i,j))
      self.result.append(aRow)
    self.next = self.__next()
  def execute(self,sql, args=None):
    pass
  def fetchall(self):
    return self.result
  def __next(self):
    index = 0
    while True:
      try:
        yield self.result[index]
        index += 1
      except:
        yield None
  def fetchone(self):
    try:
      return self.next.next()
    except:
      return None

class TestEmptyCursor(psycopg2.extensions.cursor):
  def __init__(self):
    self.result = []
  def execute(self,sql, args=None):
    pass
  def fetchall(self):
    return self.result

class TestSingleCursor(psycopg2.extensions.cursor):
  def __init__(self):
    self.result = [['Row 0, Column 0']]
  def execute(self,sql, args=None):
    pass
  def fetchall(self):
    return self.result


class TestDatabase(unittest.TestCase):
  def setUp(self):
    self.logger = TestingLogger()
    self.connectionData0 = (config.databaseHost,config.databaseName,config.databaseUserName,config.databasePassword)
    self.connectionDataL = (config.databaseHost,config.databaseName,config.databaseUserName,config.databasePassword,self.logger)
    self.dsn = "host=%s dbname=%s user=%s password=%s" % self.connectionData0
    self.connection = psycopg2.connect(self.dsn)
    createDB(self.connection)

  def tearDown(self):
    dropDB(self.connection)
    self.connection.close()

  def testExecute(self):
    aCursor = TestMultiCursor(numCols=1,numRows=3)
    f = db.execute(aCursor,"")
    vals = [x for x in f]
    assert 3 == len(vals)
    assert 'Row 0, Column 0' == vals[0][0]
    assert 'Row 2, Column 0' == vals[-1][0]
    aCursor = TestMultiCursor(numCols=1,numRows=1)

  def testSingleValueEmpty(self):
    try:
      cur = TestEmptyCursor()
      db.singleValueSql(cur,"")
      assert False, "must raise SQLDidNotReturnSingleValue"
    except db.SQLDidNotReturnSingleValue,e:
      pass

  def testSingleValueSingle(self):
    try:
      cur = TestSingleCursor()
      assert "Row 0, Column 0" == db.singleValueSql(cur,"")
    except Exception, e:
      assert False, "must not raise an exception for this %s" %e

  def testSingleValueMulti(self):
    try:
      cur = TestMultiCursor(numRows=5)
      assert "Row 0, Column 0" == db.singleValueSql(cur,"")
    except Exception, e:
      assert False, "must not raise an exception for this "+e

  def testSingleRowEmpty(self):
    try:
      cur = TestEmptyCursor()
      db.singleRowSql(cur,"")
      assert False, "must raise SQLDidNotReturnSingleRow"
    except db.SQLDidNotReturnSingleRow,e:
      pass

  def testSingleRowSingle(self):
    try:
      cur = TestSingleCursor()
      assert ["Row 0, Column 0"] == db.singleRowSql(cur,"")
    except Exception, e:
      assert False, "must not raise this exception"

  def testSingleRowMulti(self):
    try:
      cur = TestMultiCursor(numRows=5, numCols=1)
      assert ["Row 0, Column 0"] == db.singleRowSql(cur,"")
    except Exception, e:
      assert False, "must not raise this exception"

  def testDatabaseInstantiation(self):
    sample1 = {'databaseHost': 'A','databasePort': 'B','databaseName': 'C','databaseUserName': 'D','databasePassword': 'E',}
    d = db.Database(sample1)
    assert d.dsn == 'host=A port=B dbname=C user=D password=E', 'dsn not created correctly'
    assert type(d.logger) == type(util.FakeLogger()), 'should have a %s but got %s instead' % (type(util.FakeLogger()), type(d.logger))
    d = db.Database(sample1, 1)
    assert d.logger == 1, 'logger pass as a parameter was not saved, got %s instead' % d.logger
    sample1 = {'databaseHost': 'A','databasePort': 'B','databaseName': 'C','databaseUserName': 'D','databasePassword': 'E', 'logger':2}
    d = db.Database(sample1)
    assert d.dsn == 'host=A port=B dbname=C user=D password=E', 'dsn not created correctly'
    assert d.logger == 2, 'logger passed with dictionary was not saved, got %s instead' % d.logger
    d = db.Database(sample1, 1)
    assert d.dsn == 'host=A port=B dbname=C user=D password=E', 'dsn not created correctly'
    assert d.logger == 1, 'logger passed with dictionary was not overridden by logger passed as a parameter, got %s instead' % d.logger


  def testConnectionPoolConstructor(self):
    # just test some variations on constructor calls
    logger = self.logger
    logger.clear()
    try:
      cp = db.DatabaseConnectionPool()
      assert False, 'expected a raised TypeError, not to get here'
    except TypeError,x:
      pass
    except Exception,x:
      assert False, 'expected a TypeError, not %s: %s'%(type(x),x)
    try:
      cp = db.DatabaseConnectionPool(config)
    except Exception,x:
      assert False, 'expected the non-logger constructor to succeed, got %s: %s'%(type(x),x)
    try:
      cp = db.DatabaseConnectionPool(config, self.logger)
    except Exception,x:
      assert False, 'expected the with-logger constructor to succeed, got %s: %s'%(type(x),x)

  def testConnectionPoolConnectToDatabase(self):
    logger = self.logger
    logger.clear()
    cp = db.DatabaseConnectionPool(config, logger)
    logger.clear()
    try:
      connection,cursor = cp.connectionCursorPair()
      assert connection
      assert cursor
    except Exception,x:
      assert False, 'expected no exceptions, got %s: %s'%(type(x),x)

  def testConnectionPoolConnectionCursorPair(self):
    logger = self.logger
    logger.clear()
    cp = db.DatabaseConnectionPool(config, logger)
    connection0 = cursor0 = None
    try:
      connection0, cursor0 = cp.connectionCursorPair()
      assert connection0
      assert cursor0
    except Exception,x:
      assert False, 'expected nothing, got %s: %s'%(type(x),x)
    connection1, cursor1 = cp.connectionCursorPair()
    assert connection0 == connection1
    assert cursor0 != cursor1

    logger.clear()
    cp = db.DatabaseConnectionPool(config, logger)
    connection0 = cursor0 = None
    try:
      connection0,cursor0 = cp.connectionCursorPair()
    except Exception,x:
      assert False, 'Expected OperationalError above, got %s: %s' %(type(x),x)

  def testConnectionPoolCleanup(self):
    class FakeLogger(object):
      def __init__(self):
        self.logs = []
      def debug(self, *args):
        self.logs.append(args)
    logger = FakeLogger()
    cp = db.DatabaseConnectionPool(config, logger)
    conn = cp.connection()
    cp.cleanup()
    conn = cp.connection()
    conn = cp.connection('fred')
    cp.cleanup()
    expected = [('%s - killing database connections', 'MainThread'),
                ('%s - connection %s closed', 'MainThread', 'MainThread'),
                ('%s - killing database connections', 'MainThread'),
                ('%s - connection %s closed', 'MainThread', 'MainThread'),
                ('%s - connection %s closed', 'MainThread', 'fred')]
    assert len(expected) == len(logger.logs)
    for e, a in zip(expected, logger.logs):
      assert e == a

  def test_connection_attempt_count(self):
    logger = self.logger
    logger.clear()
    class ConnectionCountingFakeDatabase(object):
      def __init__(self, config, logger=None):
        self.connect_counter = 0
      def connection(self, database_module=None):
        self.connect_counter += 1
        return 17
      logger = self.logger
    temp_Database = db.Database
    db.Database = ConnectionCountingFakeDatabase
    try:
      db_pool = db.DatabaseConnectionPool(config, logger)
      c1 = db_pool.connection()
      assert db_pool.database.connect_counter == 1
      c1 = db_pool.connection()
      assert db_pool.database.connect_counter == 1
      c1 = db_pool.connection('fred')
      assert db_pool.database.connect_counter == 2
      c1 = db_pool.connection()
      assert db_pool.database.connect_counter == 2
      c1 = db_pool.connection('fred')
      assert db_pool.database.connect_counter == 2
    finally:
      db.Database = temp_Database


  def testLoggingCursorExecute(self):
    logCursor = self.connection.cursor(cursor_factory=db.LoggingCursor)
    logCursor.setLogger(self.logger)
    self.logger.clear()
    logCursor.execute('select 4;')
    assert logging.INFO == self.logger.levels[0], "Expect level %s, go %s"%(logging.INFO,self.logger.levels[0])
    assert self.logger.buffer[0] == 'select 4;','... but got %s'%(self.logger.buffer[0])
    params = {'id':3}
    logCursor.execute("select id from gringo where id=%(id)s;",params)
    assert logging.INFO == self.logger.levels[1]
    expected = "select id from gringo where id=%(id)s;"%(params)
    got = self.logger.buffer[1]
    assert expected == got, "Expected [%s] but got [%s]"%(expected,got)
    params = [3]
    logCursor.execute("select id from gringo where id=%s;",params)
    expected = "select id from gringo where id=%s;"%(params[0])
    got = self.logger.buffer[2]
    assert expected == got, "Expected [%s] but got [%s]"%(expected,got)

  def testLoggingCursorExecutemany(self):
    logCursor = self.connection.cursor(cursor_factory=db.LoggingCursor)
    logCursor.setLogger(self.logger)
    self.logger.clear()
    def chargen():
      for i in 'abcdef':
        yield (i,)

    logCursor.executemany("insert into chartable values (%s)",chargen())
    assert self.logger.buffer[0] == 'insert into chartable values (%s) ...'
    assert self.logger.levels[0] == logging.INFO
    data = ('g','h')
    logCursor.executemany("insert into chartable values (%s)",data)
    assert self.logger.buffer[1].startswith("insert into chartable values")
    assert "'g'" in self.logger.buffer[1]
    assert "..." in self.logger.buffer[1]
    assert self.logger.levels[1] == logging.INFO

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

