diff --git a/tests/test_branch.py b/tests/test_branch.py index 6908566..fcf7db5 100644 --- a/tests/test_branch.py +++ b/tests/test_branch.py @@ -1,4 +1,4 @@ -from speckle.api.models import Branch, Stream +from speckle.api.models import Branch, Commit, Stream import pytest @@ -25,3 +25,28 @@ class TestBranch: ) assert isinstance(branch.id, str) + + def test_branch_get(self, client, mesh, stream, branch): + client.commit.create( + stream_id=stream.id, + branch_name=branch.name, + object_id=mesh.id, + message="a commit for testing branch get", + ) + + fetched_branch = client.branch.get(stream_id=stream.id, name=branch.name) + + assert isinstance(fetched_branch, Branch) + assert fetched_branch.name == branch.name + assert fetched_branch.description == branch.description + assert isinstance(fetched_branch.commits.items, list) + assert isinstance(fetched_branch.commits.items[0], Commit) + + def test_branch_list(self, client, stream, branch): + branches = client.branch.list(stream_id=stream.id) + print(branches) + + assert isinstance(branches, list) + assert len(branches) == 2 + assert isinstance(branches[0], Branch) + assert branches[0].name == branch.name \ No newline at end of file