// Package persistence provides the database interface for Flamenco Manager. package persistence /* ***** BEGIN GPL LICENSE BLOCK ***** * * Original Code Copyright (C) 2022 Blender Foundation. * * This file is part of Flamenco. * * Flamenco is free software: you can redistribute it and/or modify it under * the terms of the GNU General Public License as published by the Free Software * Foundation, either version 3 of the License, or (at your option) any later * version. * * Flamenco is distributed in the hope that it will be useful, but WITHOUT ANY * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR * A PARTICULAR PURPOSE. See the GNU General Public License for more details. * * You should have received a copy of the GNU General Public License along with * Flamenco. If not, see . * * ***** END GPL LICENSE BLOCK ***** */ import ( "os" "testing" "time" "github.com/stretchr/testify/assert" "golang.org/x/net/context" ) const TestDSN = "flamenco-test.sqlite" func CreateTestDB(t *testing.T) *DB { // Creating a new database should be fast. ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second) defer cancel() if _, err := os.Stat(TestDSN); err == nil { // File exists. if err := os.Remove(TestDSN); err != nil { t.Fatalf("unable to remove %s: %v", TestDSN, err) } } db, err := openDB(ctx, TestDSN) assert.NoError(t, err) err = db.migrate() assert.NoError(t, err) return db }