builder_context_test.go 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. package tarsum
  2. import (
  3. "io"
  4. "io/ioutil"
  5. "os"
  6. "testing"
  7. )
  8. // Try to remove tarsum (in the BuilderContext) that do not exists, won't change a thing
  9. func TestTarSumRemoveNonExistent(t *testing.T) {
  10. filename := "testdata/46af0962ab5afeb5ce6740d4d91652e69206fc991fd5328c1a94d364ad00e457/layer.tar"
  11. reader, err := os.Open(filename)
  12. if err != nil {
  13. t.Fatal(err)
  14. }
  15. defer reader.Close()
  16. ts, err := NewTarSum(reader, false, Version0)
  17. if err != nil {
  18. t.Fatal(err)
  19. }
  20. // Read and discard bytes so that it populates sums
  21. _, err = io.Copy(ioutil.Discard, ts)
  22. if err != nil {
  23. t.Errorf("failed to read from %s: %s", filename, err)
  24. }
  25. expected := len(ts.GetSums())
  26. ts.(BuilderContext).Remove("")
  27. ts.(BuilderContext).Remove("Anything")
  28. if len(ts.GetSums()) != expected {
  29. t.Fatalf("Expected %v sums, go %v.", expected, ts.GetSums())
  30. }
  31. }
  32. // Remove a tarsum (in the BuilderContext)
  33. func TestTarSumRemove(t *testing.T) {
  34. filename := "testdata/46af0962ab5afeb5ce6740d4d91652e69206fc991fd5328c1a94d364ad00e457/layer.tar"
  35. reader, err := os.Open(filename)
  36. if err != nil {
  37. t.Fatal(err)
  38. }
  39. defer reader.Close()
  40. ts, err := NewTarSum(reader, false, Version0)
  41. if err != nil {
  42. t.Fatal(err)
  43. }
  44. // Read and discard bytes so that it populates sums
  45. _, err = io.Copy(ioutil.Discard, ts)
  46. if err != nil {
  47. t.Errorf("failed to read from %s: %s", filename, err)
  48. }
  49. expected := len(ts.GetSums()) - 1
  50. ts.(BuilderContext).Remove("etc/sudoers")
  51. if len(ts.GetSums()) != expected {
  52. t.Fatalf("Expected %v sums, go %v.", expected, len(ts.GetSums()))
  53. }
  54. }