package repo import ( "database/sql" "strings" "testing" "github.com/drone/drone/server/database" "github.com/drone/drone/server/database/testdata" _ "github.com/mattn/go-sqlite3" ) // in-memory database instance for unit testing var db *sql.DB // setup the test database and test fixtures func setup() { db, _ = sql.Open("sqlite3", ":memory:") database.Load(db) testdata.Load(db) } // teardown the test database func teardown() { db.Close() } func TestFind(t *testing.T) { setup() defer teardown() repos := NewManager(db) repo, err := repos.Find(1) if err != nil { t.Errorf("Want Repo from ID, got %s", err) } testRepo(t, repo) } func TestName(t *testing.T) { setup() defer teardown() repos := NewManager(db) user, err := repos.FindName("github.com", "lhofstadter", "lenwoloppali") if err != nil { t.Errorf("Want Repo by Name, got %s", err) } testRepo(t, user) } func TestList(t *testing.T) { setup() defer teardown() repos := NewManager(db) all, err := repos.List(1) if err != nil { t.Errorf("Want Repos, got %s", err) } var got, want = len(all), 2 if got != want { t.Errorf("Want %v Repos, got %v", want, got) } testRepo(t, all[0]) } func TestInsert(t *testing.T) { setup() defer teardown() repo, _ := New("github.com", "mrwolowitz", "lenwoloppali") repos := NewManager(db) if err := repos.Insert(repo); err != nil { t.Errorf("Want Repo created, got %s", err) } // verify unique remote + owner + name login constraint var err = repos.Insert(&Repo{Host: repo.Host, Owner: repo.Owner, Name: repo.Name}) if err == nil || !strings.Contains(err.Error(), "repo_host, repo_owner, repo_name are not unique") { t.Errorf("Want unique constraint violated, got %s", err) } } func TestUpdate(t *testing.T) { setup() defer teardown() repos := NewManager(db) repo, err := repos.Find(1) if err != nil { t.Errorf("Want Repo from ID, got %s", err) } // update the repo's access token repo.Active = false repo.Private = false repo.Privileged = false repo.PostCommit = false repo.PullRequest = false if err := repos.Update(repo); err != nil { t.Errorf("Want Repo updated, got %s", err) } updated, _ := repos.Find(1) var got, want = updated.Active, repo.Active if got != want { t.Errorf("Want updated Active %v, got %v", want, got) } got, want = updated.Private, repo.Private if got != want { t.Errorf("Want updated Private %v, got %v", want, got) } got, want = updated.Privileged, repo.Privileged if got != want { t.Errorf("Want updated Privileged %v, got %v", want, got) } got, want = updated.PostCommit, repo.PostCommit if got != want { t.Errorf("Want updated PostCommit %v, got %v", want, got) } got, want = updated.PullRequest, repo.PullRequest if got != want { t.Errorf("Want updated PullRequest %v, got %v", want, got) } } func TestDelete(t *testing.T) { setup() defer teardown() repos := NewManager(db) repo, err := repos.Find(1) if err != nil { t.Errorf("Want Repo from ID, got %s", err) } // delete the repo if err := repos.Delete(repo); err != nil { t.Errorf("Want Repo deleted, got %s", err) } // check to see if the deleted repo is actually gone if _, err := repos.Find(1); err != sql.ErrNoRows { t.Errorf("Want ErrNoRows, got %s", err) } } // testRepo is a helper function that compares the repo // to an expected set of fixed field values. func testRepo(t *testing.T, repo *Repo) { var got, want = repo.Remote, "github.com" if got != want { t.Errorf("Want Remote %v, got %v", want, got) } got, want = repo.Host, "github.com" if got != want { t.Errorf("Want Host %v, got %v", want, got) } got, want = repo.Owner, "lhofstadter" if got != want { t.Errorf("Want Owner %v, got %v", want, got) } got, want = repo.Name, "lenwoloppali" if got != want { t.Errorf("Want Name %v, got %v", want, got) } got, want = repo.CloneURL, "git://github.com/lhofstadter/lenwoloppali.git" if got != want { t.Errorf("Want URL %v, got %v", want, got) } got, want = repo.PublicKey, "publickey" if got != want { t.Errorf("Want PublicKey %v, got %v", want, got) } got, want = repo.PrivateKey, "privatekey" if got != want { t.Errorf("Want PrivateKey %v, got %v", want, got) } got, want = repo.Params, "params" if got != want { t.Errorf("Want Params %v, got %v", want, got) } var gotBool, wantBool = repo.Active, true if gotBool != wantBool { t.Errorf("Want Active %v, got %v", wantBool, gotBool) } gotBool, wantBool = repo.Private, true if gotBool != wantBool { t.Errorf("Want Private %v, got %v", wantBool, gotBool) } gotBool, wantBool = repo.Privileged, true if gotBool != wantBool { t.Errorf("Want Privileged %v, got %v", wantBool, gotBool) } gotBool, wantBool = repo.PostCommit, true if gotBool != wantBool { t.Errorf("Want PostCommit %v, got %v", wantBool, gotBool) } gotBool, wantBool = repo.PullRequest, true if gotBool != wantBool { t.Errorf("Want PullRequest %v, got %v", wantBool, gotBool) } var gotInt64, wantInt64 = repo.ID, int64(1) if gotInt64 != wantInt64 { t.Errorf("Want ID %v, got %v", wantInt64, gotInt64) } gotInt64, wantInt64 = repo.Created, int64(1398065343) if gotInt64 != wantInt64 { t.Errorf("Want Created %v, got %v", wantInt64, gotInt64) } gotInt64, wantInt64 = repo.Updated, int64(1398065344) if gotInt64 != wantInt64 { t.Errorf("Want Updated %v, got %v", wantInt64, gotInt64) } gotInt64, wantInt64 = repo.Timeout, int64(900) if gotInt64 != wantInt64 { t.Errorf("Want Timeout %v, got %v", wantInt64, gotInt64) } }