From 92d70005710156a6c2e0d3d9f32fa58821f38892 Mon Sep 17 00:00:00 2001
From: Yuce Tekol <yucetekol@gmail.com>
Date: Thu, 14 Sep 2017 11:03:05 +0300
Subject: [PATCH] SetColumnAttrs excludes frame attribute; resolves #792

---
 executor.go      |  1 +
 executor_test.go | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 48 insertions(+)

diff --git a/executor.go b/executor.go
index 018728f53..03ce03aaa 100644
--- a/executor.go
+++ b/executor.go
@@ -1252,6 +1252,7 @@ func (e *Executor) executeSetColumnAttrs(ctx context.Context, index string, c *p
 	// Copy args and remove reserved fields.
 	attrs := pql.CopyArgs(c.Args)
 	delete(attrs, colName)
+	delete(attrs, "frame")
 
 	// Set attributes.
 	if err := idx.ColumnAttrStore().SetAttrs(id, attrs); err != nil {
diff --git a/executor_test.go b/executor_test.go
index b426b3875..482903bc2 100644
--- a/executor_test.go
+++ b/executor_test.go
@@ -1062,3 +1062,50 @@ func TestExecutor_Execute_ErrMaxWritesPerRequest(t *testing.T) {
 		t.Fatalf("unexpected error: %s", err)
 	}
 }
+
+// Ensure SetColumnAttrs doesn't save `frame` as an attribute
+func TestExectutor_SetColumnAttrs_ExcludeFrame(t *testing.T) {
+	hldr := test.MustOpenHolder()
+	defer hldr.Close()
+	index := hldr.MustCreateIndexIfNotExists("i", pilosa.IndexOptions{})
+	index.CreateFrame("f", pilosa.FrameOptions{InverseEnabled: true})
+	targetAttrs := map[string]interface{}{
+		"foo": "bar",
+	}
+	e := test.NewExecutor(hldr.Holder, test.NewCluster(1))
+
+	// SetColumnAttrs call should exclude the frame attribute
+	_, err := e.Execute(context.Background(), "i", test.MustParse("SetBit(frame='f', rowID=1, columnID=10)"), nil, nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+	_, err = e.Execute(context.Background(), "i", test.MustParse("SetColumnAttrs(frame='f', columnID=10, foo='bar')"), nil, nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+	attrs, err := index.ColumnAttrStore().Attrs(10)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !reflect.DeepEqual(attrs, targetAttrs) {
+		t.Fatalf("%#v != %#v", targetAttrs, attrs)
+	}
+
+	// SetColumnAttrs call should not break if frame is not specified
+	_, err = e.Execute(context.Background(), "i", test.MustParse("SetBit(frame='f', rowID=1, columnID=20)"), nil, nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+	_, err = e.Execute(context.Background(), "i", test.MustParse("SetColumnAttrs(columnID=20, foo='bar')"), nil, nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+	attrs, err = index.ColumnAttrStore().Attrs(20)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !reflect.DeepEqual(attrs, targetAttrs) {
+		t.Fatalf("%#v != %#v", targetAttrs, attrs)
+	}
+
+}