package dbstore

import (
  "testing"
)

func TestGetQuery(t *testing.T) {
  tests := []struct {
    name string
    val string
    exp bool
  } {
    { "pass", "init", true },
    { "fail", "invalid", false },
  }

  for _, test := range(tests) {
    t.Run(test.name, func(t *testing.T) {
      got, err := getQuery(test.val)
      if err != nil && test.exp {
        t.Error(err)
      } else if err == nil && !test.exp {
        t.Errorf("got %v, exp error", got)
      }
    })
  }
}

func TestGetQueries(t *testing.T) {
  tests := []struct {
    name string
    vals []string
    exp bool
  } {
    { "pass", []string { "init" }, true },
    { "fail", []string { "invalid" }, false },
  }

  for _, test := range(tests) {
    t.Run(test.name, func(t *testing.T) {
      got, err := getQueries(test.vals)
      if err != nil && test.exp {
        t.Error(err)
      } else if err == nil && !test.exp {
        t.Errorf("got %v, exp error", got)
      }
    })
  }
}