// Vikunja is a to-do list application to facilitate your life. // Copyright 2018-2021 Vikunja and contributors. All rights reserved. // // This program is free software: you can redistribute it and/or modify // it under the terms of the GNU Affero General Public Licensee as published by // the Free Software Foundation, either version 3 of the License, or // (at your option) any later version. // // This program 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 Affero General Public Licensee for more details. // // You should have received a copy of the GNU Affero General Public Licensee // along with this program. If not, see . package models import ( "testing" "code.vikunja.io/api/pkg/db" "code.vikunja.io/api/pkg/user" "github.com/stretchr/testify/assert" ) func TestTeamMember_Create(t *testing.T) { doer := &user.User{ ID: 1, } t.Run("normal", func(t *testing.T) { db.LoadAndAssertFixtures(t) s := db.NewSession() defer s.Close() tm := &TeamMember{ TeamID: 1, Username: "user3", } err := tm.Create(s, doer) assert.NoError(t, err) err = s.Commit() assert.NoError(t, err) db.AssertExists(t, "team_members", map[string]interface{}{ "id": tm.ID, "team_id": 1, "user_id": 3, }, false) }) t.Run("already existing", func(t *testing.T) { db.LoadAndAssertFixtures(t) s := db.NewSession() defer s.Close() tm := &TeamMember{ TeamID: 1, Username: "user1", } err := tm.Create(s, doer) assert.Error(t, err) assert.True(t, IsErrUserIsMemberOfTeam(err)) }) t.Run("nonexisting user", func(t *testing.T) { db.LoadAndAssertFixtures(t) s := db.NewSession() defer s.Close() tm := &TeamMember{ TeamID: 1, Username: "nonexistinguser", } err := tm.Create(s, doer) assert.Error(t, err) assert.True(t, user.IsErrUserDoesNotExist(err)) }) t.Run("nonexisting team", func(t *testing.T) { db.LoadAndAssertFixtures(t) s := db.NewSession() defer s.Close() tm := &TeamMember{ TeamID: 9999999, Username: "user1", } err := tm.Create(s, doer) assert.Error(t, err) assert.True(t, IsErrTeamDoesNotExist(err)) }) } func TestTeamMember_Delete(t *testing.T) { t.Run("normal", func(t *testing.T) { db.LoadAndAssertFixtures(t) s := db.NewSession() defer s.Close() tm := &TeamMember{ TeamID: 1, Username: "user1", } err := tm.Delete(s) assert.NoError(t, err) err = s.Commit() assert.NoError(t, err) db.AssertMissing(t, "team_members", map[string]interface{}{ "team_id": 1, "user_id": 1, }) }) } func TestTeamMember_Update(t *testing.T) { t.Run("normal", func(t *testing.T) { db.LoadAndAssertFixtures(t) s := db.NewSession() defer s.Close() tm := &TeamMember{ TeamID: 1, Username: "user1", Admin: true, } err := tm.Update(s) assert.NoError(t, err) assert.False(t, tm.Admin) // Since this endpoint toggles the right, we should get a false for admin back. err = s.Commit() assert.NoError(t, err) db.AssertExists(t, "team_members", map[string]interface{}{ "team_id": 1, "user_id": 1, "admin": false, }, false) }) // This should have the same result as the normal run as the update function // should ignore what was passed. t.Run("explicitly false in payload", func(t *testing.T) { db.LoadAndAssertFixtures(t) s := db.NewSession() defer s.Close() tm := &TeamMember{ TeamID: 1, Username: "user1", Admin: true, } err := tm.Update(s) assert.NoError(t, err) assert.False(t, tm.Admin) err = s.Commit() assert.NoError(t, err) db.AssertExists(t, "team_members", map[string]interface{}{ "team_id": 1, "user_id": 1, "admin": false, }, false) }) }