Add user_repo_unit to yaml generation

Author: guillep2k

contrib/fixtures/fixture_generation.go

@@ -23,6 +23,7 @@ var (
 	}{
 		{
 			models.GetYamlFixturesAccess, "access",
+			models.GetYamlFixturesUserRepoUnit, "user_repo_unit",
 		},
 	}
 	fixturesDir string

models/fixture_generation.go

@@ -43,3 +43,30 @@ func GetYamlFixturesAccess() (string, error) {
 
 	return b.String(), nil
 }
+
+// GetYamlFixturesUserRepoUnit returns a string containing the contents for the
+// user_repo_unit table, as recalculated using repo.RebuildAllUserRepoUnits()
+func GetYamlFixturesUserRepoUnit() (string, error) {
+
+	if err := RebuildAllUserRepoUnits(x); err != nil {
+		return "", err
+	}
+
+	var b strings.Builder
+
+	units := make([]*UserRepoUnit, 0, 200)
+	if err := x.OrderBy("user_id, repo_id").Find(&units); err != nil {
+		return "", err
+	}
+
+	for _, u := range units {
+		fmt.Fprintf(&b, "-\n")
+		fmt.Fprintf(&b, "  user_id: %d\n", u.UserID)
+		fmt.Fprintf(&b, "  repo_id: %d\n", u.RepoID)
+		fmt.Fprintf(&b, "  type: %d\n", u.Type)
+		fmt.Fprintf(&b, "  mode: %d\n", u.Mode)
+		fmt.Fprintf(&b, "\n")
+	}
+
+	return b.String(), nil
+}

models/fixture_test.go

@@ -31,4 +31,5 @@ func TestFixtureGeneration(t *testing.T) {
 	}
 
 	test(GetYamlFixturesAccess, "access")
+	test(GetYamlFixturesUserRepoUnit, "user_repo_unit")
 }