ast_test.go

// Copyright 2017 Pilosa Corp.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package pql_test

import (
	"testing"

	"github.com/pilosa/pilosa/pql"
)

// Ensure call can be converted into a string.
func TestCall_String(t *testing.T) {
	t.Run("Empty", func(t *testing.T) {
		c := &pql.Call{Name: "Bitmap"}
		if s := c.String(); s != `Bitmap()` {
			t.Fatalf("unexpected string: %s", s)
		}
	})
}

// Ensure call can be converted into a string.
func TestCall_SupportsInverse(t *testing.T) {
	t.Run("Bitmap", func(t *testing.T) {
		q, err := pql.ParseString(`Bitmap()`)
		if err != nil {
			t.Fatal(err)
		} else if q.Calls[0].SupportsInverse() != true {
			t.Fatalf("call should support inverse: %s", q.Calls[0])
		}
	})
	t.Run("Count Bitmap", func(t *testing.T) {
		q, err := pql.ParseString(`Count(Bitmap())`)
		if err != nil {
			t.Fatal(err)
		} else if q.Calls[0].SupportsInverse() == true {
			t.Fatalf("call should not support inverse: %s", q.Calls[0])
		}
	})
	t.Run("Union Bitmaps", func(t *testing.T) {
		q, err := pql.ParseString(`Union(Bitmap(), Bitmap())`)
		if err != nil {
			t.Fatal(err)
		} else if q.Calls[0].SupportsInverse() == true {
			t.Fatalf("call should not support inverse: %s", q.Calls[0])
		}
	})

}

// Ensure call is correctly determined to be against an inverse view.
func TestCall_IsInverse(t *testing.T) {
	t.Run("Bitmap Row", func(t *testing.T) {
		q, err := pql.ParseString(`Bitmap(frame="f", row=1)`)
		if err != nil {
			t.Fatal(err)
		} else if q.Calls[0].IsInverse("row", "col") != false {
			t.Fatalf("incorrect call inverse: %s", q.Calls[0])
		}
	})
	t.Run("Bitmap Column", func(t *testing.T) {
		q, err := pql.ParseString(`Bitmap(frame="f", col=1)`)
		if err != nil {
			t.Fatal(err)
		} else if q.Calls[0].IsInverse("row", "col") != true {
			t.Fatalf("incorrect call inverse: %s", q.Calls[0])
		}
	})
	t.Run("Bitmap Column No Label", func(t *testing.T) {
		q, err := pql.ParseString(`Bitmap(frame="f", col=1)`)
		if err != nil {
			t.Fatal(err)
		} else if q.Calls[0].IsInverse("rowX", "colX") != false {
			t.Fatalf("incorrect call inverse: %s", q.Calls[0])
		}
	})
	t.Run("Count", func(t *testing.T) {
		q, err := pql.ParseString(`Count(Bitmap(frame="f", col=1))`)
		if err != nil {
			t.Fatal(err)
		} else if q.Calls[0].IsInverse("row", "col") != false {
			t.Fatalf("incorrect call inverse: %s", q.Calls[0])
		}
	})

}