From d70a7209f7acec43c9636a5506feac857764f79f Mon Sep 17 00:00:00 2001
From: Dmitry S <dsagal+git@gmail.com>
Date: Sat, 11 Feb 2023 14:57:56 -0500
Subject: [PATCH] (core) Attempt to fix test_formula_prompt test broken by a
 merge

Test Plan: Tests should pass

Reviewers: jarek

Reviewed By: jarek

Differential Revision: https://phab.getgrist.com/D3797
---
 sandbox/grist/test_formula_prompt.py | 28 ++++++++++++++++++----------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/sandbox/grist/test_formula_prompt.py b/sandbox/grist/test_formula_prompt.py
index 6cceb1db..f26ad3b9 100644
--- a/sandbox/grist/test_formula_prompt.py
+++ b/sandbox/grist/test_formula_prompt.py
@@ -8,18 +8,26 @@ from formula_prompt import (
   values_type, column_type, referenced_tables, get_formula_prompt,
 )
 from objtypes import RaisedException
-from records import Record, RecordSet
+from records import Record as BaseRecord, RecordSet as BaseRecordSet
 
 
 class FakeTable(object):
 
   def __init__(self):
-    pass
+    class Record(BaseRecord):
+      _table = self
+    class RecordSet(BaseRecordSet):
+      _table = self
+    self.Record = Record
+    self.RecordSet = RecordSet
 
   table_id = "Table1"
   _identity_relation = None
 
 
+fake_table = FakeTable()
+
+
 @unittest.skipUnless(six.PY3, "Python 3 only")
 class TestFormulaPrompt(test_engine.EngineTestCase):
   def test_values_type(self):
@@ -38,22 +46,22 @@ class TestFormulaPrompt(test_engine.EngineTestCase):
     self.assertEqual(values_type([1, 2, "3", None]), "Any")
 
     self.assertEqual(values_type([
-      Record(FakeTable(), None),
-      Record(FakeTable(), None),
+      fake_table.Record(None),
+      fake_table.Record(None),
     ]), "Table1")
     self.assertEqual(values_type([
-      Record(FakeTable(), None),
-      Record(FakeTable(), None),
+      fake_table.Record(None),
+      fake_table.Record(None),
       None,
     ]), "Optional[Table1]")
 
     self.assertEqual(values_type([
-      RecordSet(FakeTable(), None),
-      RecordSet(FakeTable(), None),
+      fake_table.RecordSet(None),
+      fake_table.RecordSet(None),
     ]), "List[Table1]")
     self.assertEqual(values_type([
-      RecordSet(FakeTable(), None),
-      RecordSet(FakeTable(), None),
+      fake_table.RecordSet(None),
+      fake_table.RecordSet(None),
       None,
     ]), "Optional[List[Table1]]")