Fix name issue

Author: stefanDeveloper

src/detector/detector.py

@@ -311,7 +311,7 @@ def detect(self) -> None:  # pragma: no cover
         for message in self.messages:
             # TODO predict all messages
             y_pred = self.model.predict_proba(
-                self.scaler.transform(self._get_features(message["host_domain_name"]))
+                self.scaler.transform(self._get_features(message["domain_name"]))
             )
             logger.info(f"Prediction: {y_pred}")
             if np.argmax(y_pred, axis=1) == 1 and y_pred[0][1] > THRESHOLD:

tests/detector/test_detector.py

@@ -16,7 +16,7 @@
     "response_ip": "252.79.173.222",
     "timestamp": "",
     "status": "NXDOMAIN",
-    "host_domain_name": "IF356gEnJHPdRxnkDId4RDUSgtqxx9I+pZ5n1V53MdghOGQncZWAQgAPRx3kswi.750jnH6iSqmiAAeyDUMX0W6SHGpVsVsKSX8ZkKYDs0GFh/9qU5N9cwl00XSD8ID.NNhBdHZIb7nc0hDQXFPlABDLbRwkJS38LZ8RMX4yUmR2Mb6YqTTJBn+nUcB9P+v.jBQdwdS53XV9W2p1BHjh.16.f.1.6037.tunnel.example.org",
+    "domain_name": "IF356gEnJHPdRxnkDId4RDUSgtqxx9I+pZ5n1V53MdghOGQncZWAQgAPRx3kswi.750jnH6iSqmiAAeyDUMX0W6SHGpVsVsKSX8ZkKYDs0GFh/9qU5N9cwl00XSD8ID.NNhBdHZIb7nc0hDQXFPlABDLbRwkJS38LZ8RMX4yUmR2Mb6YqTTJBn+nUcB9P+v.jBQdwdS53XV9W2p1BHjh.16.f.1.6037.tunnel.example.org",
     "record_type": "A",
     "size": "100b",
 }