Пример #1
0
 def setUp(self):
     super(TestPostgresListenStore, self).setUp()
     self.log = logging.getLogger(__name__)
     self.logstore = init_postgres_connection(self.config.SQLALCHEMY_DATABASE_URI)
     # TODO: Does this use the normal or test DB??
     self.testuser_id = db.user.create("test")
     user = db.user.get(self.testuser_id)
     print(user)
     self.testuser_name = db.user.get(self.testuser_id)['musicbrainz_id']
    def setUp(self):
        super(TestAPICompatUserClass, self).setUp()
        self.log = logging.getLogger(__name__)
        self.logstore = init_postgres_connection(self.config.TEST_SQLALCHEMY_DATABASE_URI)

        # Create a user
        uid = db.user.create("test")
        self.assertIsNotNone(db.user.get(uid))
        with db.engine.connect() as connection:
            result = connection.execute(text('SELECT * FROM "user" WHERE id = :id'),
                                        {"id": uid})
            self.user = User(result.fetchone())

        # Insert some listens
        date = datetime(2015, 9, 3, 0, 0, 0)
        self.log.info("Inserting test data...")
        test_data = generate_data(date, 100)
        self.logstore.insert(test_data)
        self.log.info("Test data inserted")
Пример #3
0
    def setUp(self):
        super(TestAPICompatUserClass, self).setUp()
        self.log = logging.getLogger(__name__)
        self.logstore = init_postgres_connection(
            self.config.SQLALCHEMY_DATABASE_URI)

        # Create a user
        uid = db.user.create("test")
        self.assertIsNotNone(db.user.get(uid))
        with db.engine.connect() as connection:
            result = connection.execute(
                text('SELECT * FROM "user" WHERE id = :id'), {"id": uid})
            row = result.fetchone()
            self.user = User(row['id'], row['created'], row['musicbrainz_id'],
                             row['auth_token'])

        # Insert some listens
        date = datetime(2015, 9, 3, 0, 0, 0)
        self.log.info("Inserting test data...")
        test_data = generate_data(date, 100)
        self.logstore.insert(test_data)
        self.log.info("Test data inserted")
 def setUp(self):
     super(TestListenStore, self).setUp()
     self.log = logging.getLogger(__name__)
     self.logstore = init_postgres_connection(self.config.TEST_SQLALCHEMY_DATABASE_URI)
     self._create_test_data()
 def setUp(self):
     super(TestPostgresListenStore, self).setUp()
     self.log = logging.getLogger(__name__)
     self.logstore = init_postgres_connection(self.config.TEST_SQLALCHEMY_DATABASE_URI)
     self.testuser_id = db.user.create("test")