func TestCreateTable(t *testing.T) {
testTableName := "test1_" + getTimestampString()
t.Log("testTableName=" + testTableName)
ac := gohbase.NewAdminClient(*host)
crt := hrpc.NewCreateTable(context.Background(), []byte(testTableName), cFamilies)
if err := ac.CreateTable(crt); err != nil {
t.Errorf("CreateTable returned an error: %v", err)
}
// check in hbase:meta if there's a region for the table
c := gohbase.NewClient(*host)
metaKey := testTableName + ",,"
keyFilter := filter.NewPrefixFilter([]byte(metaKey))
scan, err := hrpc.NewScanStr(context.Background(), metaTableName, hrpc.Filters(keyFilter))
if err != nil {
t.Fatalf("Failed to create Scan request: %s", err)
}
rsp, err := c.Scan(scan)
if err != nil {
t.Errorf("Scan returned an error: %v", err)
}
if len(rsp) != 1 {
t.Errorf("Meta returned %s rows for prefix '%s' , want 1", len(rsp), metaKey)
}
}
func TestEnableTable(t *testing.T) {
testTableName := "test1_" + getTimestampString()
t.Log("testTableName=" + testTableName)
ac := gohbase.NewAdminClient(*host)
crt := hrpc.NewCreateTable(context.Background(), []byte(testTableName), cFamilies)
if err := ac.CreateTable(crt); err != nil {
t.Errorf("CreateTable returned an error: %v", err)
}
// disable
dit := hrpc.NewDisableTable(context.Background(), []byte(testTableName))
err := ac.DisableTable(dit)
if err != nil {
t.Errorf("DisableTable returned an error: %v", err)
}
et := hrpc.NewEnableTable(context.Background(), []byte(testTableName))
err = ac.EnableTable(et)
if err != nil {
t.Errorf("EnableTable returned an error: %v", err)
}
delt := hrpc.NewDeleteTable(context.Background(), []byte(testTableName))
err = ac.DeleteTable(delt)
if err == nil || !strings.Contains(err.Error(), "TableNotDisabledException") {
t.Errorf("DeleteTable should error with TableNotDisabledException, got %s", err)
}
}
func TestMain(m *testing.M) {
if host == nil {
panic("Host is not set!")
}
ac := gohbase.NewAdminClient(*host)
err := test.CreateTable(ac, table, []string{"cf", "cf2"})
if err != nil {
panic(err)
}
res := m.Run()
err = test.DeleteTable(ac, table)
if err != nil {
panic(err)
}
os.Exit(res)
}