g2h: Add unit tests and restructure class layout

Instead of having a large generator class use individual pages

1 files changed, 13 insertions, 0 deletions

diff --git a/g2h/page_factory.py b/g2h/page_factory.py
new file mode 100644
index 0000000..dae9675
--- /dev/null
+++ b/g2h/page_factory.py
@@ -0,0 +1,13 @@
+from .overview_page import OverviewPage
+from .log_page import LogPage
+
+class PageFactory:
+    def __init__(self, out_dir, name):
+        self.out_dir = out_dir
+        self.name = name
+
+    def new_overview(self, branches, tags):
+        return OverviewPage(self.out_dir, self.name, branches, tags)
+
+    def new_log(self, commits, branches, tags):
+        return LogPage(self.out_dir, self.name, commits, branches, tags)