select * from foo where a == 2; should return all columns not just 'a'

src/parser/query_api.go

@@ -61,7 +61,7 @@ func (self *Query) GetReferencedColumns() map[*Value][]string {
 		name := value.Name
 		returnedMapping[value] = uniq(append(mapping[name], notPrefixedColumns...))
 		if len(returnedMapping[value]) > 1 && returnedMapping[value][0] == "*" {
-			returnedMapping[value] = returnedMapping[value][1:]
+			returnedMapping[value] = returnedMapping[value][:1]
 		}
 
 		delete(mapping, name)

src/parser/query_api_test.go

@@ -103,15 +103,15 @@ func (self *QueryApiSuite) TestGetReferencedColumnsWithARegexTable(c *C) {
 	}
 }
 
-func (self *QueryApiSuite) TestGetReferencedColumnsReturnsGroupByColumn(c *C) {
-	queryStr := "select count(*), region from events group by time(1h), region;"
+func (self *QueryApiSuite) TestGetReferencedColumnsWithWhereClause(c *C) {
+	queryStr := "select * from foo where a == 5;"
 	query, err := ParseQuery(queryStr)
 	c.Assert(err, IsNil)
 	columns := query.GetReferencedColumns()
 	c.Assert(columns, HasLen, 1)
 	for v, columns := range columns {
-		c.Assert(v.Name, Equals, "events")
-		c.Assert(columns, DeepEquals, []string{"region"})
+		c.Assert(v.Name, Equals, "foo")
+		c.Assert(columns, DeepEquals, []string{"*"})
 	}
 }