Ejemplo n.º 1
0
  def testQuerySingleColumn(self):
    with temp.AutoTempFilePath(suffix=".sqlite") as db_filepath:
      with contextlib.closing(sqlite3.connect(db_filepath)) as conn:
        with contextlib.closing(conn.cursor()) as cursor:
          cursor.execute("CREATE TABLE foo (bar TEXT)")
          cursor.execute("INSERT INTO foo VALUES ('quux'), ('norf'), ('thud')")

        conn.commit()

        context = sqlite.ConnectionContext(conn)
        results = list(context.Query("SELECT * FROM foo"))
        self.assertEqual(results, [("quux",), ("norf",), ("thud",)])
Ejemplo n.º 2
0
  def testQueryMultipleColumns(self):
    with temp.AutoTempFilePath(suffix=".sqlite") as db_filepath:
      with contextlib.closing(sqlite3.connect(db_filepath)) as conn:
        with contextlib.closing(conn.cursor()) as cursor:
          cursor.execute("CREATE TABLE foo (bar INTEGER, baz TEXT)")
          cursor.execute("INSERT INTO foo(bar, baz) VALUES (42, 'quux')")
          cursor.execute("INSERT INTO foo(bar, baz) VALUES (108, 'norf')")

        conn.commit()

        context = sqlite.ConnectionContext(conn)
        results = list(context.Query("SELECT bar, baz FROM foo"))
        self.assertEqual(results, [(42, "quux"), (108, "norf")])