def testUnicodeValues(self):
foo_artifact_source = rdf_artifacts.ArtifactSource(
type=rdf_artifacts.ArtifactSource.SourceType.GRR_CLIENT_ACTION,
attributes={"client_action": "FooAction"})
foo_artifact = rdf_artifacts.Artifact(name="Foo",
doc="Foo bar baz.",
sources=[foo_artifact_source],
provides=["os"],
labels=["System"],
supported_os=["Linux"])
with artifact_test_lib.PatchCleanArtifactRegistry():
artifact_registry.REGISTRY.RegisterArtifact(foo_artifact)
with test_lib.ConfigOverrider(
{"Artifacts.knowledge_base": ["Foo"]}):
session_id = flow_test_lib.TestFlowHelper(
artifact.KnowledgeBaseInitializationFlow.__name__,
client_mock=action_mocks.ActionMock(FooAction),
client_id=test_lib.TEST_CLIENT_ID,
token=self.token)
results = flow_test_lib.GetFlowResults(test_lib.TEST_CLIENT_ID,
session_id)
self.assertLen(results, 1)
self.assertEqual(results[0].os, "zażółć gęślą jaźń 🎮")
def Run(self):
with artifact_test_lib.PatchCleanArtifactRegistry():
test_artifacts_file = os.path.join(config.CONFIG["Test.data_dir"],
"artifacts", "test_artifact.json")
artifact_registry.REGISTRY.AddFileSource(test_artifacts_file)
self.Check("ListArtifacts")
