From 5e1e359e5e012dc18ac24ddbe011ccfc5a8f339e Mon Sep 17 00:00:00 2001
From: radao <a.radice@lumicks.com>
Date: Mon, 17 Jul 2023 15:33:09 +0200
Subject: [PATCH] Fix unit tests by updating `Branch` type

---
 doltcli/types.py    | 2 ++
 tests/test_types.py | 4 +++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/doltcli/types.py b/doltcli/types.py
index d531c3f..bac632f 100644
--- a/doltcli/types.py
+++ b/doltcli/types.py
@@ -26,6 +26,8 @@ class BranchT(BaseDataclass):
     latest_committer_email: Optional[str] = None
     latest_commit_date: Optional[datetime.datetime] = None
     latest_commit_message: Optional[str] = None
+    remote: Optional[str] = None
+    branch: Optional[str] = None
 
 
 @dataclass
diff --git a/tests/test_types.py b/tests/test_types.py
index 7b47775..3fb5df0 100644
--- a/tests/test_types.py
+++ b/tests/test_types.py
@@ -13,12 +13,14 @@ def test_datetime_serialize():
         latest_commit_date=dt,
         latest_committer_email=None,
         latest_commit_message=None,
+        remote=None,
+        branch=None,
     )
     br = Branch(**cmp)
     assert br.dict() == cmp
     assert (
         br.json()
         == """
-            {"name": "test", "hash": "23", "latest_committer": null, "latest_committer_email": null, "latest_commit_date": "2018-06-29 00:00:00", "latest_commit_message": null}
+            {"name": "test", "hash": "23", "latest_committer": null, "latest_committer_email": null, "latest_commit_date": "2018-06-29 00:00:00", "latest_commit_message": null, "remote": null, "branch": null}
             """.strip()
     )