def test_valid_for_table(self):
"""It should update the custom properties.
"""
resource = factories.TableFactory(workspace=self.workspace)
globalid = helpers.to_global_id('TableType', resource.pk)
self.execute_success_test_case(resource, globalid)
def test_query_when_not_authorized(self):
"""Outside users should not be able to access this resource.
"""
table = factories.TableFactory(workspace=self.workspace)
global_id = helpers.to_global_id('TableType', table.pk)
variables = {'objectId': global_id}
results = self.execute(self.statement, variables=variables)
self.assertPermissionDenied(results)
def setUpTestData(cls):
cls.commentable = factories.TableFactory()
cls.workspace = cls.commentable.workspace
cls.user = factories.UserFactory()
cls.comment = cls.commentable.comments.create(
author=cls.user,
html='This is a test',
)
def setUp(self):
super(TestOmnisearch, self).setUp()
self.datastore = Datastore.objects.get(id='s4N8p5g0wjiS')
self.other_datastore = factories.DatastoreFactory(
workspace=self.workspace)
self.other_schema = factories.SchemaFactory(
datastore=self.other_datastore)
self.other_table = factories.TableFactory(schema=self.other_schema,
name='customers')
def setUp(self):
super().setUp()
self.resource = factories.TableFactory(name='accounts', workspace=self.workspace)
self.resource.owners.create(workspace=self.workspace, owner=self.users['READONLY'])
self.resource.owners.create(workspace=self.workspace, owner=self.users['OWNER'])
self.asset_owner = self.resource.owners.create(workspace=self.workspace, owner=self.users['MEMBER'])
self.global_id = helpers.to_global_id('AssetOwnerType', self.asset_owner.id)
self.datastore = self.resource.datastore
def setUp(self):
super(TestGetDatastoreAssets, self).setUp()
self.datastore = self.factory(workspace=self.workspace)
self.schema = factories.SchemaFactory(datastore=self.datastore)
self.assets = []
for i in range(10):
self.assets.append(
factories.TableFactory(name='app%s' % str(i).zfill(3), schema=self.schema)
)
for j in range(10):
self.assets.append(
factories.TableFactory(name='site%s' % str(j).zfill(3), schema=self.schema)
)
self.other_datastore = self.factory(workspace=self.workspace)
self.other_schema = factories.SchemaFactory(datastore=self.other_datastore)
self.other_table = factories.TableFactory(schema=self.other_schema, name='yyz')
def setUp(self):
super(TestGetTableNamesBySchema, self).setUp()
self.datastore = self.factory(workspace=self.workspace)
self.schema = factories.SchemaFactory(datastore=self.datastore)
self.tables = factories.TableFactory.create_batch(10, schema=self.schema)
self.other_schema = factories.SchemaFactory(datastore=self.datastore)
self.other_table = factories.TableFactory(schema=self.other_schema)
self.variables = {
'datastoreId': helpers.to_global_id('DatastoreType', self.datastore.pk),
'schemaName': self.schema.name,
}
def _setup_and_get_variables(self, **overrides):
"""Setup test and return variables for query.
"""
datastore = factories.DatastoreFactory(workspace=self.workspace, **overrides)
schema = factories.SchemaFactory(workspace=self.workspace, datastore=datastore)
table = factories.TableFactory(workspace=self.workspace, schema=schema)
column = self.factory(workspace=self.workspace, table=table, readme=self.readme)
variables = {
'datastoreId': helpers.to_global_id('DatastoreType', datastore.pk),
'schemaName': schema.name,
'tableName': table.name,
'columnName': column.name,
}
return datastore, variables
def test_query(self):
"""It should return only the workspaces that the current User belongs to.
"""
table = factories.TableFactory(workspace=self.workspace)
count = 3
content_type = ContentType.objects.get(model='table')
for co in range(count):
comment = Comment.objects.create(
author=random.choice(self.users_list),
html='Test',
workspace=self.workspace,
object_id=table.pk,
content_type_id=content_type.pk,
parent=None,
)
for c in range(count):
Comment.objects.create(
author=random.choice(self.users_list),
html='Nested',
parent=comment,
object_id=table.pk,
content_type_id=content_type.pk,
workspace=self.workspace,
)
global_id = helpers.to_global_id('TableType', table.pk)
variables = {'objectId': global_id}
results = self.execute(self.statement, variables=variables)
results = results['data'][self.operation]
self.assertEqual(
first=len(results['edges']),
second=count,
msg="Node count should equal number of active comments.")
self.assertEqual(first=len(results['edges']),
second=results['totalCount'],
msg="Node count should equal totalCount field.")
self.assertEqual(
len(results['edges'][0]['node']['childComments']),
count,
)
def test_saves_with_workspace_id(self):
"""It should automatically add the workspace of the ContentObject.
"""
user = factories.UserFactory()
commentable = factories.TableFactory()
properties = {
'content_object': commentable,
'author': user,
'html': helpers.faker.sentence(),
}
comment = self.model_class.objects.create(**properties)
self.assertEqual(
comment.workspace_id,
commentable.workspace_id,
)
def setUpTestData(cls):
cls.commentable = factories.TableFactory()
cls.workspace = cls.commentable.workspace
cls.user = factories.UserFactory()
def setUp(self):
super().setUp()
self.commentable = factories.TableFactory(workspace=self.workspace)
def setUp(self):
super().setUp()
self.resource = factories.TableFactory(workspace=self.workspace)
self.datastore = self.resource.datastore
def setUp(self):
super().setUp()
self.resource = factories.TableFactory(name='accounts', workspace=self.workspace)
self.datastore = self.resource.datastore
self.object_id = helpers.to_global_id('TableType', self.resource.pk)