Files
dolt/go/libraries/doltcore/dtestutils/environment.go
T
2020-09-25 15:35:04 -07:00

54 lines
1.6 KiB
Go

// Copyright 2019 Liquidata, Inc.
//
// 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 dtestutils
import (
"context"
"github.com/dolthub/dolt/go/libraries/doltcore/doltdb"
"github.com/dolthub/dolt/go/libraries/doltcore/env"
"github.com/dolthub/dolt/go/libraries/utils/filesys"
"github.com/dolthub/dolt/go/store/types"
)
const (
TestHomeDir = "/user/bheni"
WorkingDir = "/user/bheni/datasets/states"
)
func testHomeDirFunc() (string, error) {
return TestHomeDir, nil
}
func CreateTestEnv() *env.DoltEnv {
const name = "billy bob"
const email = "bigbillieb@fake.horse"
initialDirs := []string{TestHomeDir, WorkingDir}
fs := filesys.NewInMemFS(initialDirs, nil, WorkingDir)
dEnv := env.Load(context.Background(), testHomeDirFunc, fs, doltdb.InMemDoltDB, "test")
cfg, _ := dEnv.Config.GetConfig(env.GlobalConfig)
cfg.SetStrings(map[string]string{
env.UserNameKey: name,
env.UserEmailKey: email,
})
err := dEnv.InitRepo(context.Background(), types.Format_7_18, name, email)
if err != nil {
panic("Failed to initialize environment:" + err.Error())
}
return dEnv
}