Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/**
* Copyright 2005-2016 Red Hat, Inc.
*
* Red Hat licenses this file to you under the Apache License, version
* 2.0 (the "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
* implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
package io.fabric8.kubernetes.assertions;
import io.fabric8.utils.IOHelpers;
import org.assertj.core.api.Fail;
import org.assertj.core.api.MapAssert;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
/**
* Collect the logs from a number of pods so that they can be asserted on
*/
public class PodLogsAssert extends MapAssert {
private static final transient Logger LOG = LoggerFactory.getLogger(PodLogsAssert.class);
private final String containerName;
private final Map logPrefixes;
public PodLogsAssert(Map actual, String containerName) {
this(actual, containerName, new HashMap());
}
public PodLogsAssert(Map actual, String containerName, Map logPrefixes) {
super(actual);
this.containerName = containerName;
this.logPrefixes = logPrefixes;
writeLogs();
}
public PodLogsAssert afterText(String startText) {
Map newLogs = new HashMap<>();
Map prefixes = new HashMap<>();
Set> entries = actual.entrySet();
for (Map.Entry entry : entries) {
String podName = entry.getKey();
String value = entry.getValue();
File file = podLogFileName(podName);
int idx = value.indexOf(startText);
if (idx >= 0) {
int splitIdx = idx + startText.length();
String remaining = value.substring(splitIdx);
String prefix = value.substring(0, splitIdx);
newLogs.put(podName, remaining);
prefixes.put(podName, prefix);
}
}
return new PodLogsAssert(newLogs, containerName, prefixes);
}
public void containsText(String... texts) {
for (String text : texts) {
Set> entries = actual.entrySet();
for (Map.Entry entry : entries) {
String podName = entry.getKey();
String value = entry.getValue();
File file = podLogFileName(podName);
int idx = value.indexOf(text);
if (idx < 0) {
Fail.fail("Log of pod " + podName + " in file: " + file + " does not contains text `" + text + "` last log: " + lastLineOf(file));
}
}
}
}
protected String lastLineOf(File file) {
try {
List lines = IOHelpers.readLines(file);
int size = lines.size();
if (size > 0) {
return lines.get(size - 1);
}
} catch (IOException e) {
LOG.debug("Failed to load: " + file + ". " + e, e );
}
return "";
}
public void doesNotContainText(String... texts) {
for (String text : texts) {
Set> entries = actual.entrySet();
for (Map.Entry entry : entries) {
String podName = entry.getKey();
String value = entry.getValue();
File file = podLogFileName(podName);
int idx = value.indexOf(text);
if (idx >= 0) {
Fail.fail("Log of pod " + podName + " in file: " + file + " contains text `" + text
+ "` at " + logFileCoords(podName, value, idx));
} else {
LOG.debug("does not contain '" + text + "' in Log of pod " + podName + " in file: " + file);
}
}
}
}
public void doesNotContainTextCount(int count, String... texts) {
if (count == 1) {
doesNotContainText(texts);
}
for (String text : texts) {
Set> entries = actual.entrySet();
for (Map.Entry entry : entries) {
String podName = entry.getKey();
String value = entry.getValue();
File file = podLogFileName(podName);
int idx = 0;
for (int i = 0; idx >= 0 && i < count; i++) {
int next = value.indexOf(text, idx);
if (next >= 0) {
idx = next + 1;
} else {
idx = next;
}
}
if (idx >= 0) {
String logText = fullLogText(podName, value.substring(0, idx - 1));
Fail.fail("Log of pod " + podName + " in file: " + file + " contains text `" + text
+ "` " + count + " times with the last at at " + textCoords(logText));
} else {
LOG.debug("does not contain '" + text + "' in Log of pod " + podName + " in file: " + file + " " + count + " times");
}
}
}
}
/**
* Returns the coordinates in the log file of the end of the bit of text
*/
protected String logFileCoords(String podName, String value, int idx) {
String prefix = "";
if (value != null && value.length() > 0 && idx > 0) {
prefix = value.substring(0, idx);
}
String logText = fullLogText(podName, prefix);
return textCoords(logText);
}
protected String fullLogText(String podName, String text) {
String logText = text;
String logPrefix = logPrefixes.get(podName);
if (logPrefix != null) {
logText = logPrefix + logText;
}
return logText;
}
/**
* Returns the line number and column of the end of text
*/
public static String textCoords(String text) {
int line = 1;
int idx = 0;
while (true) {
int next =text.indexOf('\n', idx);
if (next < 0) {
break;
}
idx = next + 1;
line += 1;
}
int column = 1 + text.length() - idx;
return "" + line + ":" + column;
}
protected File podLogFileName(String podName) {
// lets return the file we use to store the pod logs
String basedir = System.getProperty("basedir", ".");
File dir = new File(basedir, "target/fabric8/systest/logs");
String name = podName;
if (containerName != null) {
name += "." + containerName;
}
name += ".log";
File answer = new File(dir, name);
answer.getParentFile().mkdirs();
return answer;
}
private void writeLogs() {
Set> entries = actual.entrySet();
for (Map.Entry entry : entries) {
String podName = entry.getKey();
String value = entry.getValue();
String logText = fullLogText(podName, value);
File file = podLogFileName(podName);
try {
IOHelpers.writeFully(file, logText);
} catch (IOException e) {
LOG.error("Failed to write log of pod " + podName + " container:" + containerName + " to file: "+ file + ". " + e, e);
}
}
}
}
