From 568f1dc0577bcc7699e5a025c049880610d50202 Mon Sep 17 00:00:00 2001
From: Keyur Shah <keyurs@google.com>
Date: Mon, 3 Jun 2024 17:01:12 -0700
Subject: [PATCH] Use lookup_key instead of key. (#324)

---
 simple/stats/db.py                                            | 2 +-
 .../runner/expected/config_driven/key_value_store.db.csv      | 2 +-
 .../expected/config_with_wildcards/key_value_store.db.csv     | 2 +-
 .../expected/generate_svg_hierarchy/key_value_store.db.csv    | 2 +-
 .../runner/expected/input_dir_driven/key_value_store.db.csv   | 2 +-
 .../runner/expected/sv_nl_sentences/key_value_store.db.csv    | 2 +-
 .../runner/expected/topic_nl_sentences/key_value_store.db.csv | 2 +-
 simple/tests/stats/test_util.py                               | 4 ++--
 8 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/simple/stats/db.py b/simple/stats/db.py
index aed23c4f..4d5fbf30 100644
--- a/simple/stats/db.py
+++ b/simple/stats/db.py
@@ -85,7 +85,7 @@
 
 _CREATE_KEY_VALUE_STORE_TABLE = """
 create table if not exists key_value_store (
-    key varchar(255),
+    lookup_key varchar(255),
     value TEXT
 );
 """
diff --git a/simple/tests/stats/test_data/runner/expected/config_driven/key_value_store.db.csv b/simple/tests/stats/test_data/runner/expected/config_driven/key_value_store.db.csv
index f3bde496..23bb062c 100644
--- a/simple/tests/stats/test_data/runner/expected/config_driven/key_value_store.db.csv
+++ b/simple/tests/stats/test_data/runner/expected/config_driven/key_value_store.db.csv
@@ -1,2 +1,2 @@
-key,value
+lookup_key,value
 StatVarGroups,H4sIAAAAAAAC/+OS5OJMSdZP1w/Kzy8R4pHi4uKA8bjcEGwhKy4B59LikvxchbDEoszEpJzUYiEhLpayxCJDKTCpBCahYkZgMSOwmBEAlEss8mMAAAA=
diff --git a/simple/tests/stats/test_data/runner/expected/config_with_wildcards/key_value_store.db.csv b/simple/tests/stats/test_data/runner/expected/config_with_wildcards/key_value_store.db.csv
index f3bde496..23bb062c 100644
--- a/simple/tests/stats/test_data/runner/expected/config_with_wildcards/key_value_store.db.csv
+++ b/simple/tests/stats/test_data/runner/expected/config_with_wildcards/key_value_store.db.csv
@@ -1,2 +1,2 @@
-key,value
+lookup_key,value
 StatVarGroups,H4sIAAAAAAAC/+OS5OJMSdZP1w/Kzy8R4pHi4uKA8bjcEGwhKy4B59LikvxchbDEoszEpJzUYiEhLpayxCJDKTCpBCahYkZgMSOwmBEAlEss8mMAAAA=
diff --git a/simple/tests/stats/test_data/runner/expected/generate_svg_hierarchy/key_value_store.db.csv b/simple/tests/stats/test_data/runner/expected/generate_svg_hierarchy/key_value_store.db.csv
index 73ea78a9..537236c5 100644
--- a/simple/tests/stats/test_data/runner/expected/generate_svg_hierarchy/key_value_store.db.csv
+++ b/simple/tests/stats/test_data/runner/expected/generate_svg_hierarchy/key_value_store.db.csv
@@ -1,2 +1,2 @@
-key,value
+lookup_key,value
 StatVarGroups,H4sIAAAAAAAC/32RT0vDQBDFKUiwjwp1UWs2KEvBo5RG8CIepFJPLVKleivbdKyBJBs3W7+EB7+yCdFG88fLzryZYXfeb3GFnjdYD+5JJyoaqU1k5qSN78mACXSnm3BJWqgXEZOKA+IdoBjHDLuZmill2Bjd0SYxKhRzqX25DCjhPRwW0zdr+rma242v4rOF46K3uKNoRfp8TKEMiCVw8rp48s2ryJviWny3H9FOVEiLd6mH/OAhTbfLiKGYypD69VWnXL2lxNN+bPzU5WWDC3aCvVQ20rn4rdgZrDzjDvYrBpmVR9hor7ZMOxwFYXy0cFQlM8mMv8Gu5TIpUXHLVNxaKm4tFfcPlbDOxTNYdQ/e/+dHrTzy00ZvO9n5BeXclHGoAgAA
diff --git a/simple/tests/stats/test_data/runner/expected/input_dir_driven/key_value_store.db.csv b/simple/tests/stats/test_data/runner/expected/input_dir_driven/key_value_store.db.csv
index f3bde496..23bb062c 100644
--- a/simple/tests/stats/test_data/runner/expected/input_dir_driven/key_value_store.db.csv
+++ b/simple/tests/stats/test_data/runner/expected/input_dir_driven/key_value_store.db.csv
@@ -1,2 +1,2 @@
-key,value
+lookup_key,value
 StatVarGroups,H4sIAAAAAAAC/+OS5OJMSdZP1w/Kzy8R4pHi4uKA8bjcEGwhKy4B59LikvxchbDEoszEpJzUYiEhLpayxCJDKTCpBCahYkZgMSOwmBEAlEss8mMAAAA=
diff --git a/simple/tests/stats/test_data/runner/expected/sv_nl_sentences/key_value_store.db.csv b/simple/tests/stats/test_data/runner/expected/sv_nl_sentences/key_value_store.db.csv
index 857881f5..d23e2a9c 100644
--- a/simple/tests/stats/test_data/runner/expected/sv_nl_sentences/key_value_store.db.csv
+++ b/simple/tests/stats/test_data/runner/expected/sv_nl_sentences/key_value_store.db.csv
@@ -1,2 +1,2 @@
-key,value
+lookup_key,value
 StatVarGroups,H4sIAAAAAAAC/wMAAAAAAAAAAAA=
diff --git a/simple/tests/stats/test_data/runner/expected/topic_nl_sentences/key_value_store.db.csv b/simple/tests/stats/test_data/runner/expected/topic_nl_sentences/key_value_store.db.csv
index 857881f5..d23e2a9c 100644
--- a/simple/tests/stats/test_data/runner/expected/topic_nl_sentences/key_value_store.db.csv
+++ b/simple/tests/stats/test_data/runner/expected/topic_nl_sentences/key_value_store.db.csv
@@ -1,2 +1,2 @@
-key,value
+lookup_key,value
 StatVarGroups,H4sIAAAAAAAC/wMAAAAAAAAAAAA=
diff --git a/simple/tests/stats/test_util.py b/simple/tests/stats/test_util.py
index 7fca08ca..9a90e425 100644
--- a/simple/tests/stats/test_util.py
+++ b/simple/tests/stats/test_util.py
@@ -88,8 +88,8 @@ def write_key_values(db_path: str, output_path: str):
   """
   with sqlite3.connect(db_path) as db:
     rows = db.execute("select * from key_value_store").fetchall()
-    pd.DataFrame(rows, columns=["key", "value"]).to_csv(output_path,
-                                                        index=False)
+    pd.DataFrame(rows, columns=["lookup_key", "value"]).to_csv(output_path,
+                                                               index=False)
 
 
 class FakeGzipTime: