Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(153)

Side by Side Diff: golden/go/expstorage/expstorage_test.go

Issue 813443002: Overhaul database package (Closed) Base URL: https://skia.googlesource.com/buildbot@master
Patch Set: Created 6 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 package expstorage 1 package expstorage
2 2
3 import ( 3 import (
4 "testing" 4 "testing"
5 ) 5 )
6 6
7 import ( 7 import (
8 // Using 'require' which is like using 'assert' but causes tests to fail . 8 // Using 'require' which is like using 'assert' but causes tests to fail .
9 assert "github.com/stretchr/testify/require" 9 assert "github.com/stretchr/testify/require"
10 10
11 "skia.googlesource.com/buildbot.git/go/database" 11 "skia.googlesource.com/buildbot.git/go/database"
12 "skia.googlesource.com/buildbot.git/go/database/testutil" 12 "skia.googlesource.com/buildbot.git/go/database/testutil"
13 "skia.googlesource.com/buildbot.git/golden/go/db" 13 "skia.googlesource.com/buildbot.git/golden/go/db"
14 "skia.googlesource.com/buildbot.git/golden/go/types" 14 "skia.googlesource.com/buildbot.git/golden/go/types"
15 ) 15 )
16 16
17 func TestExpectationStores(t *testing.T) { 17 func TestMySQLExpectationsStore(t *testing.T) {
18 » // Test the memory store.
19 » memStore := NewMemExpectationsStore()
20 » testExpectationStore(t, memStore)
21
22 » // Initialize the DB to use a local SQLite instances.
23 » vdb := database.NewVersionedDB(&database.DatabaseConfig{
24 » » SQLiteFilePath: "correctness.db",
25 » » MigrationSteps: db.MigrationSteps(),
26 » })
27 » sqlStore := NewSQLExpectationStore(vdb)
28 » testExpectationStore(t, sqlStore)
29 }
30
31 func TestMySQLExpecatationsStore(t *testing.T) {
32 // Set up the database and make sure it's at the right version. 18 // Set up the database and make sure it's at the right version.
33 conf := &database.DatabaseConfig{ 19 conf := &database.DatabaseConfig{
34 MySQLString: testutil.GetTestMySQLConnStr(t, "root", "correct ness"), 20 MySQLString: testutil.GetTestMySQLConnStr(t, "root", "correct ness"),
35 MigrationSteps: db.MigrationSteps(), 21 MigrationSteps: db.MigrationSteps(),
36 } 22 }
37 23
38 // Lock to serialize DB tests 24 // Lock to serialize DB tests
39 lockVdb := testutil.GetMySQlLock(t, conf) 25 lockVdb := testutil.GetMySQlLock(t, conf)
40 defer func() { 26 defer func() {
41 testutil.ReleaseMySQLLock(t, lockVdb) 27 testutil.ReleaseMySQLLock(t, lockVdb)
(...skipping 63 matching lines...) Expand 10 before | Expand all | Expand 10 after
105 assert.Equal(t, types.TestClassification(map[string]types.Label{DIGEST_2 1: types.POSITIVE}), foundExps.Tests[TEST_2]) 91 assert.Equal(t, types.TestClassification(map[string]types.Label{DIGEST_2 1: types.POSITIVE}), foundExps.Tests[TEST_2])
106 92
107 changeExps.RemoveDigests([]string{DIGEST_12}) 93 changeExps.RemoveDigests([]string{DIGEST_12})
108 err = store.Put(changeExps, "user-3") 94 err = store.Put(changeExps, "user-3")
109 assert.Nil(t, err) 95 assert.Nil(t, err)
110 96
111 foundExps, err = store.Get(false) 97 foundExps, err = store.Get(false)
112 assert.Nil(t, err) 98 assert.Nil(t, err)
113 assert.Equal(t, 1, len(foundExps.Tests)) 99 assert.Equal(t, 1, len(foundExps.Tests))
114 } 100 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698