feat: add Mbql

Signed-off-by: Charles Larivier <charles@dribbble.com>

Author: Charles Larivier

src/metabase/mbql/__init__.py

@@ -1 +0,0 @@
-from metabase.mbql.query import Query

src/metabase/mbql/base.py

@@ -1,18 +1,17 @@
 from typing import List
 
 
-class Mbql:
-    def compile(self):
-        raise NotImplementedError()
-
-    def __repr__(self):
-        return str(self.compile())
+class Option:
+    pass
 
 
-class Field(Mbql):
-    def __init__(self, id: int, option=None):
+class Mbql:
+    def __init__(self, id: int, option: Option = None):
         self.id = id
         self.option = option
 
     def compile(self) -> List:
         return ["field", self.id, self.option]
+
+    def __repr__(self):
+        return str(self.compile())

tests/mbql/__init__.py

@@ -0,0 +1,21 @@
+from unittest import TestCase
+
+from metabase.mbql.base import Mbql
+
+
+class MbqlTests(TestCase):
+    def test_compile(self):
+        """Ensure Mbql.compile() returns a list formatted as ["field", self.id, self.option]."""
+        mbql = Mbql(id=5, option={"foo": "bar"})
+        self.assertEqual(["field", 5, {"foo": "bar"}], mbql.compile())
+
+        mbql = Mbql(id=10, option=None)
+        self.assertEqual(["field", 10, None], mbql.compile())
+
+    def test_repr(self):
+        """Ensure Mbql.__repr__() returns the compiled Mbql as a string."""
+        mbql = Mbql(id=5, option={"foo": "bar"})
+        self.assertEqual("['field', 5, {'foo': 'bar'}]", mbql.__repr__())
+
+        mbql = Mbql(id=10, option=None)
+        self.assertEqual("['field', 10, None]", mbql.__repr__())