move

BUILD.bazel

@@ -1,4 +1,4 @@
-load("@io_bazel_rules_go//go:def.bzl", "go_binary", "go_library", "go_test")
+load("@io_bazel_rules_go//go:def.bzl", "go_binary", "go_library")
 load("@bazel_gazelle//:def.bzl", "gazelle")
 # load(":rules.bzl", "foo_binary")
 
@@ -103,10 +103,3 @@ go_binary(
     goos = "linux",
     visibility = ["//visibility:public"],
 )
-
-go_test(
-    name = "osquery-extension_test",
-    srcs = ["main_test.go"],
-    embed = [":osquery-extension_lib"],
-    deps = ["@com_github_stretchr_testify//assert"],
-)

main.go

@@ -44,7 +44,7 @@ func main() {
 		panic("Version not set")
 	}
 
-	useragent := buildUserAgent(Version)
+	useragent := sofa.BuildUserAgent(Version)
 	sofaOpts := []sofa.Option{
 		sofa.WithUserAgent(useragent),
 	}
@@ -113,7 +113,3 @@ func main() {
 		log.Fatalln(err)
 	}
 }
-
-func buildUserAgent(version string) string {
-	return "macadmins-osquery-extension/" + version
-}

tables/sofa/client.go

@@ -296,3 +296,7 @@ func (s *SofaClient) downloadSofaJSON() (Root, error) {
 
 	return s.loadCachedData()
 }
+
+func BuildUserAgent(version string) string {
+	return "macadmins-osquery-extension/" + version
+}

tables/sofa/client_test.go

@@ -369,3 +369,11 @@ func TestCreateCacheDir(t *testing.T) {
 
 	assert.DirExists(t, client.cacheDir)
 }
+
+func TestBuildUserAgent(t *testing.T) {
+	version := "1.0.0"
+	expectedUserAgent := "macadmins-osquery-extension/1.0.0"
+	userAgent := BuildUserAgent(version)
+
+	assert.Equal(t, expectedUserAgent, userAgent)
+}