1 files changed, 145 insertions, 0 deletions
diff --git a/internal/gtsmodel/instance_test.go b/internal/gtsmodel/instance_test.go
new file mode 100644
index 000000000..5c685bb25
--- /dev/null
+++ b/internal/gtsmodel/instance_test.go
@@ -0,0 +1,145 @@
+/*
+   GoToSocial
+   Copyright (C) 2021 GoToSocial Authors admin@gotosocial.org
+
+   This program is free software: you can redistribute it and/or modify
+   it under the terms of the GNU Affero General Public License 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 License for more details.
+
+   You should have received a copy of the GNU Affero General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package gtsmodel_test
+
+import (
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/suite"
+	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
+)
+
+func happyInstance() *gtsmodel.Instance {
+	return &gtsmodel.Instance{
+		ID:                     "01FE91RJR88PSEEE30EV35QR8N",
+		CreatedAt:              time.Now(),
+		UpdatedAt:              time.Now(),
+		Domain:                 "example.org",
+		Title:                  "Example Instance",
+		URI:                    "https://example.org",
+		SuspendedAt:            time.Time{},
+		DomainBlockID:          "",
+		DomainBlock:            nil,
+		ShortDescription:       "This is a description for the example/testing instance.",
+		Description:            "This is a way longer description for the example/testing instance!",
+		Terms:                  "Don't be a knobhead.",
+		ContactEmail:           "admin@example.org",
+		ContactAccountUsername: "admin",
+		ContactAccountID:       "01FEE20H5QWHJDEXAEE9G96PR0",
+		ContactAccount:         nil,
+		Reputation:             420,
+		Version:                "gotosocial 0.1.0",
+	}
+}
+
+type InstanceValidateTestSuite struct {
+	suite.Suite
+}
+
+func (suite *InstanceValidateTestSuite) TestValidateInstanceHappyPath() {
+	// no problem here
+	m := happyInstance()
+	err := gtsmodel.ValidateStruct(*m)
+	suite.NoError(err)
+}
+
+func (suite *InstanceValidateTestSuite) TestValidateInstanceBadID() {
+	m := happyInstance()
+
+	m.ID = ""
+	err := gtsmodel.ValidateStruct(*m)
+	suite.EqualError(err, "Key: 'Instance.ID' Error:Field validation for 'ID' failed on the 'required' tag")
+
+	m.ID = "01FE96W293ZPRG9FQQP48HK8N001FE96W32AT24VYBGM12WN3GKB"
+	err = gtsmodel.ValidateStruct(*m)
+	suite.EqualError(err, "Key: 'Instance.ID' Error:Field validation for 'ID' failed on the 'ulid' tag")
+}
+
+func (suite *InstanceValidateTestSuite) TestValidateInstanceAccountURI() {
+	i := happyInstance()
+
+	i.URI = ""
+	err := gtsmodel.ValidateStruct(*i)
+	suite.EqualError(err, "Key: 'Instance.URI' Error:Field validation for 'URI' failed on the 'required' tag")
+
+	i.URI = "---------------------------"
+	err = gtsmodel.ValidateStruct(*i)
+	suite.EqualError(err, "Key: 'Instance.URI' Error:Field validation for 'URI' failed on the 'url' tag")
+}
+
+func (suite *InstanceValidateTestSuite) TestValidateInstanceDodgyAccountID() {
+	i := happyInstance()
+
+	i.ContactAccountID = "9HZJ76B6VXSKF"
+	err := gtsmodel.ValidateStruct(*i)
+	suite.EqualError(err, "Key: 'Instance.ContactAccountID' Error:Field validation for 'ContactAccountID' failed on the 'ulid' tag")
+
+	i.ContactAccountID = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa!!!!!!!!!!!!"
+	err = gtsmodel.ValidateStruct(*i)
+	suite.EqualError(err, "Key: 'Instance.ContactAccountID' Error:Field validation for 'ContactAccountID' failed on the 'ulid' tag")
+
+	i.ContactAccountID = ""
+	err = gtsmodel.ValidateStruct(*i)
+	suite.EqualError(err, "Key: 'Instance.ContactAccountID' Error:Field validation for 'ContactAccountID' failed on the 'required_with' tag")
+
+	i.ContactAccountUsername = ""
+	err = gtsmodel.ValidateStruct(*i)
+	suite.NoError(err)
+}
+
+func (suite *InstanceValidateTestSuite) TestValidateInstanceDomain() {
+	i := happyInstance()
+
+	i.Domain = "poopoo"
+	err := gtsmodel.ValidateStruct(*i)
+	suite.EqualError(err, "Key: 'Instance.Domain' Error:Field validation for 'Domain' failed on the 'fqdn' tag")
+
+	i.Domain = ""
+	err = gtsmodel.ValidateStruct(*i)
+	suite.EqualError(err, "Key: 'Instance.Domain' Error:Field validation for 'Domain' failed on the 'required' tag")
+
+	i.Domain = "https://aaaaaaaaaaaaah.org"
+	err = gtsmodel.ValidateStruct(*i)
+	suite.EqualError(err, "Key: 'Instance.Domain' Error:Field validation for 'Domain' failed on the 'fqdn' tag")
+}
+
+func (suite *InstanceValidateTestSuite) TestValidateInstanceContactEmail() {
+	i := happyInstance()
+
+	i.ContactEmail = "poopoo"
+	err := gtsmodel.ValidateStruct(*i)
+	suite.EqualError(err, "Key: 'Instance.ContactEmail' Error:Field validation for 'ContactEmail' failed on the 'email' tag")
+
+	i.ContactEmail = ""
+	err = gtsmodel.ValidateStruct(*i)
+	suite.NoError(err)
+}
+
+func (suite *InstanceValidateTestSuite) TestValidateInstanceNoCreatedAt() {
+	i := happyInstance()
+
+	i.CreatedAt = time.Time{}
+	err := gtsmodel.ValidateStruct(*i)
+	suite.NoError(err)
+}
+
+func TestInstanceValidateTestSuite(t *testing.T) {
+	suite.Run(t, new(InstanceValidateTestSuite))
+}