Merge pull request #32 from tristangrebot/master

Fix missing headercolor setting in Table dbml generation

pydbml/classes/table.py

@@ -245,6 +245,8 @@ def dbml(self):
         result += f'Table {name} '
         if self.alias:
             result += f'as "{self.alias}" '
+        if self.header_color:
+            result += f'[headercolor: {self.header_color}] '
         result += '{\n'
         columns_str = '\n'.join(c.dbml for c in self.columns)
         result += indent(columns_str) + '\n'

test/test_classes/test_table.py

@@ -422,6 +422,24 @@ def test_dbml_simple(self):
 }'''
         self.assertEqual(t.dbml, expected)
 
+    def test_header_color_dbml(self):
+        t = Table('products')
+        t.header_color = '#C84432'
+        c1 = Column('id', 'integer')
+        c2 = Column('name', 'varchar2')
+        t.add_column(c1)
+        t.add_column(c2)
+        s = Database()
+        s.add(t)
+
+        expected = \
+'''Table "products" [headercolor: #C84432] {
+    "id" integer
+    "name" varchar2
+}'''
+        self.assertEqual(t.dbml, expected)
+
+
     def test_schema_dbml(self):
         t = Table('products', schema="myschema")
         c1 = Column('id', 'integer')