예제 #1
0
파일: test_coverage.py 프로젝트: ooici/pyon
    def test_spans(self):
        datastore = DatastoreFactory.get_datastore(datastore_name='coverage_spans', config=CFG)
        table_name = datastore._get_datastore_name()

        delete_statement = ''.join(['DELETE FROM ', table_name, " WHERE span_address='test'"])
        with datastore.pool.cursor(**datastore.cursor_args) as cur:
            cur.execute(delete_statement)

        statement = ''.join(['INSERT into ', table_name, " (span_address, coverage_id, vertical_range) ",
                             "VALUES ('test', 'cov_1', '[1.0, 2.2]')"])
        with datastore.pool.cursor(**datastore.cursor_args) as cur:
            cur.execute(statement)

        statement = ''.join(['SELECT span_address, coverage_id, lower(vertical_range), upper(vertical_range) FROM ',
                             table_name, " WHERE span_address='test'"])
        with datastore.pool.cursor(**datastore.cursor_args) as cur:
            cur.execute(statement)
            self.assertGreater(cur.rowcount, 0)
            row = cur.fetchone()
            self.assertEqual(row[0], 'test')
            self.assertEqual(row[1], 'cov_1')
            self.assertEqual(float(row[2]), float(1.0))
            self.assertEqual(float(row[3]), float(2.2))

        with datastore.pool.cursor(**datastore.cursor_args) as cur:
            cur.execute(delete_statement)
예제 #2
0
    def test_coverage(self):
        datastore = DatastoreFactory.get_datastore(datastore_name="coverage", config=CFG)
        table_name = datastore._get_datastore_name()

        delete_statement = "".join(["DELETE FROM ", table_name, " WHERE id='test'"])
        with datastore.pool.cursor(**datastore.cursor_args) as cur:
            cur.execute(delete_statement)

        statement = "".join(["INSERT into ", table_name, " (id, name) VALUES ('test', 'insert')"])
        with datastore.pool.cursor(**datastore.cursor_args) as cur:
            cur.execute(statement)

        statement = "".join(["SELECT id, name FROM ", table_name, " WHERE id='test'"])
        with datastore.pool.cursor(**datastore.cursor_args) as cur:
            cur.execute(statement)
            self.assertGreater(cur.rowcount, 0)
            row = cur.fetchone()
            self.assertEqual(row[0], "test")
            self.assertEqual(row[1], "insert")

        with datastore.pool.cursor(**datastore.cursor_args) as cur:
            cur.execute(delete_statement)