mirror of
https://github.com/chanzuckerberg/cellxgene.git
synced 2026-09-25 19:38:12 +08:00
config refactor (#1854)
* split out config * add tests for base and app config, refactor client config out of app config * refactor default config retrieval * create config test class and helper functions * move default_config into server to fix import issue
This commit is contained in:
@@ -16,45 +16,48 @@ class DatabaseTest(unittest.TestCase):
|
||||
del cls.db
|
||||
|
||||
def test_user_creation(self):
|
||||
one_user = self.db.get(table=CellxGeneUser, entity_id='test_user_id')
|
||||
self.assertEqual(one_user.id, 'test_user_id')
|
||||
one_user = self.db.get(table=CellxGeneUser, entity_id="test_user_id")
|
||||
self.assertEqual(one_user.id, "test_user_id")
|
||||
user_count = self.db.session.query(CellxGeneUser).count()
|
||||
self.assertGreater(user_count, 10)
|
||||
|
||||
def test_dataset_creation(self):
|
||||
one_dataset = self.db.query(table_args=[CellxGeneDataset],
|
||||
filter_args=[CellxGeneDataset.name == 'test_dataset'])
|
||||
self.assertEqual(one_dataset[0].name, 'test_dataset')
|
||||
one_dataset = self.db.query(
|
||||
table_args=[CellxGeneDataset], filter_args=[CellxGeneDataset.name == "test_dataset"]
|
||||
)
|
||||
self.assertEqual(one_dataset[0].name, "test_dataset")
|
||||
dataset_count = self.db.session.query(CellxGeneDataset).count()
|
||||
self.assertGreater(dataset_count, 10)
|
||||
|
||||
def test_annotation_creation(self):
|
||||
one_annotation = self.db.query(table_args=[Annotation], filter_args=[Annotation.tiledb_uri == 'tiledb_uri'])[0]
|
||||
self.assertEqual(one_annotation.tiledb_uri, 'tiledb_uri')
|
||||
one_annotation = self.db.query(table_args=[Annotation], filter_args=[Annotation.tiledb_uri == "tiledb_uri"])[0]
|
||||
self.assertEqual(one_annotation.tiledb_uri, "tiledb_uri")
|
||||
annotation_count = self.db.session.query(Annotation).count()
|
||||
self.assertGreater(annotation_count, 10)
|
||||
|
||||
def test_get_most_recent_annotation_for_user_dataset(self):
|
||||
dataset_id = str(self.db.query(table_args=[CellxGeneDataset],
|
||||
filter_args=[CellxGeneDataset.name == 'test_dataset'])[0].id)
|
||||
dataset_id = str(
|
||||
self.db.query(table_args=[CellxGeneDataset], filter_args=[CellxGeneDataset.name == "test_dataset"])[0].id
|
||||
)
|
||||
|
||||
# have to commit separately because created_at time written on the db server
|
||||
self.db.session.add(Annotation(dataset_id=dataset_id, user_id='test_user_id', tiledb_uri='tiledb_uri_0'))
|
||||
self.db.session.add(Annotation(dataset_id=dataset_id, user_id="test_user_id", tiledb_uri="tiledb_uri_0"))
|
||||
self.db.session.commit()
|
||||
|
||||
self.db.session.add(Annotation(dataset_id=dataset_id, user_id='test_user_id', tiledb_uri='tiledb_uri_1'))
|
||||
self.db.session.add(Annotation(dataset_id=dataset_id, user_id="test_user_id", tiledb_uri="tiledb_uri_1"))
|
||||
self.db.session.commit()
|
||||
|
||||
self.db.session.add(Annotation(dataset_id=dataset_id, user_id='test_user_id', tiledb_uri='tiledb_uri_2'))
|
||||
self.db.session.add(Annotation(dataset_id=dataset_id, user_id="test_user_id", tiledb_uri="tiledb_uri_2"))
|
||||
self.db.session.commit()
|
||||
|
||||
self.db.session.add(Annotation(dataset_id=dataset_id, user_id='test_user_id', tiledb_uri='tiledb_uri_3'))
|
||||
self.db.session.add(Annotation(dataset_id=dataset_id, user_id="test_user_id", tiledb_uri="tiledb_uri_3"))
|
||||
self.db.session.commit()
|
||||
|
||||
self.db.session.add(Annotation(dataset_id=dataset_id, user_id='test_user_id', tiledb_uri='tiledb_uri_4'))
|
||||
self.db.session.add(Annotation(dataset_id=dataset_id, user_id="test_user_id", tiledb_uri="tiledb_uri_4"))
|
||||
self.db.session.commit()
|
||||
|
||||
most_recent_annotation = self.db.query_for_most_recent(Annotation, [Annotation.dataset_id == dataset_id,
|
||||
Annotation.user_id == 'test_user_id'])
|
||||
most_recent_annotation = self.db.query_for_most_recent(
|
||||
Annotation, [Annotation.dataset_id == dataset_id, Annotation.user_id == "test_user_id"]
|
||||
)
|
||||
|
||||
self.assertEqual(most_recent_annotation.tiledb_uri, 'tiledb_uri_4')
|
||||
self.assertEqual(most_recent_annotation.tiledb_uri, "tiledb_uri_4")
|
||||
|
||||
Reference in New Issue
Block a user