Add Workspace.roles

This commit is contained in:
richard-dds 2018-08-21 10:26:12 -04:00
parent 43263f35cc
commit ee17ca6633
3 changed files with 8 additions and 3 deletions

View File

@ -11,10 +11,10 @@ class Workspace(Base, TimestampsMixin):
id = Id() id = Id()
name = Column(String, unique=True) name = Column(String, unique=True)
request_id = Column(ForeignKey("requests.id"), nullable=False) request_id = Column(ForeignKey("requests.id"), nullable=False)
request = relationship("Request") request = relationship("Request")
projects = relationship("Project", back_populates="workspace") projects = relationship("Project", back_populates="workspace")
roles = relationship("WorkspaceRole")
@property @property
def owner(self): def owner(self):

View File

@ -11,7 +11,7 @@ class WorkspaceRole(Base):
id = Id() id = Id()
workspace_id = Column(UUID(as_uuid=True), ForeignKey("workspaces.id"), index=True) workspace_id = Column(UUID(as_uuid=True), ForeignKey("workspaces.id"), index=True)
workspace = relationship("Workspace") workspace = relationship("Workspace", back_populates="roles")
role_id = Column(UUID(as_uuid=True), ForeignKey("roles.id")) role_id = Column(UUID(as_uuid=True), ForeignKey("roles.id"))
role = relationship("Role") role = relationship("Role")

View File

@ -18,7 +18,6 @@ def test_can_create_workspace():
def test_default_workspace_name_is_request_id(): def test_default_workspace_name_is_request_id():
request = RequestFactory.create() request = RequestFactory.create()
workspace = Workspaces.create(request) workspace = Workspaces.create(request)
assert workspace.request == request
assert workspace.name == str(request.id) assert workspace.name == str(request.id)
@ -51,3 +50,9 @@ def test_workspace_has_timestamps():
request = RequestFactory.create() request = RequestFactory.create()
workspace = Workspaces.create(request) workspace = Workspaces.create(request)
assert workspace.time_created == workspace.time_updated assert workspace.time_created == workspace.time_updated
def test_workspace_has_roles():
request = RequestFactory.create()
workspace = Workspaces.create(request)
assert workspace.roles[0].user == request.creator