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")