Function TestGetResourcesTiDB

backend/component/export/export_test.go:323–386

(t *testing.T)

}

func TestGetResourcesTiDB(t *testing.T) {
	a := assert.New(t)
	ctx := context.Background()

	instance := &store.InstanceMessage{
		ResourceID: "test-tidb-instance",
		Metadata: &storepb.Instance{
			Engine: storepb.Engine_TIDB,
		},
	}

	getDatabaseMetadataFunc := func(_ context.Context, _ string, _ string) (string, *model.DatabaseMetadata, error) {
		dbMeta := model.NewDatabaseMetadata(
			&storepb.DatabaseSchemaMetadata{
				Name: "",
				Schemas: []*storepb.SchemaMetadata{
					{
						Name: "",
						Tables: []*storepb.TableMetadata{
							{
								Name: "cbt_plans",
								Columns: []*storepb.ColumnMetadata{
									{Name: "id", Type: "bigint"},
								},
							},
						},
					},
				},
			},
			nil,
			nil,
			storepb.Engine_TIDB,
			false,
		)
		return "", dbMeta, nil
	}

	listDatabaseNamesFunc := func(_ context.Context, _ string) ([]string, error) {
		return nil, nil
	}

	getLinkedDatabaseMetadataFunc := func(_ context.Context, _, _, _ string) (string, string, *model.DatabaseMetadata, error) {
		return "", "", nil, nil
	}

	// With databaseName = "":
	// - If TiDB hits default case: returns error "database must be specified"
	// - If TiDB hits MySQL case: calls GetQuerySpan and returns resources
	resources, err := GetResources(
		ctx,
		nil,
		storepb.Engine_TIDB,
		"",
		"SELECT id FROM cbt_plans;",
		instance,
		getDatabaseMetadataFunc,
		listDatabaseNamesFunc,
		getLinkedDatabaseMetadataFunc,